
import json
import os
import re
import unittest

from transformers import CodeGenTokenizer, CodeGenTokenizerFast
from transformers.models.codegen.tokenization_codegen import VOCAB_FILES_NAMES
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin







@require_tokenizers
class      A__							(     __magic_name__						,   unittest.TestCase	):
				lowercase			    =  CodeGenTokenizer
				lowercase			    =  CodeGenTokenizerFast
				lowercase			    =  True
				lowercase			    =  {"add_prefix_space": True}
				lowercase			    =  False






				def 							_lowerCamelCase		(						self					:      Union[str, Any]   ):






							'''simple docstring'''







							super().setUp()

							# Adapted from Sennrich et al. 2015 and https://github.com/rsennrich/subword-nmt
							lowerCAmelCase__      :      Optional[Any]         =		[
							    "l",
							    "o",
							    "w",
							    "e",
							    "r",
							    "s",
							    "t",
							    "i",
							    "d",
							    "n",
							    "\u0120",
							    "\u0120l",
							    "\u0120n",
							    "\u0120lo",
							    "\u0120low",
							    "er",
							    "\u0120lowest",
							    "\u0120newer",
							    "\u0120wider",
							    "<unk>",
							    "<|endoftext|>",
							]
							lowerCAmelCase__      :      Tuple         =		dict(zip(snake_case__    ,     range(len(snake_case__   )   )   )   )
							lowerCAmelCase__      :      Dict         =		["#version: 0.2", "\u0120 l", "\u0120l o", "\u0120lo w", "e r", ""]
							lowerCAmelCase__      :      Optional[int]         =		{"unk_token": "<unk>"}

							lowerCAmelCase__      :      Optional[Any]         =		os.path.join(self.tmpdirname    ,     VOCAB_FILES_NAMES['vocab_file']   )
							lowerCAmelCase__      :      Dict         =		os.path.join(self.tmpdirname    ,     VOCAB_FILES_NAMES['merges_file']   )
							with open(self.vocab_file    ,     'w'    ,     encoding='utf-8'   ) as fp:
										fp.write(json.dumps(snake_case__   ) + '\n'   )
							with open(self.merges_file    ,     'w'    ,     encoding='utf-8'   ) as fp:
										fp.write('\n'.join(snake_case__   )   )






				def 							_lowerCamelCase		(						self					:      List[Any]    ,     **a					:      List[Any]   ):






							'''simple docstring'''







							kwargs.update(self.special_tokens_map   )
							return CodeGenTokenizer.from_pretrained(self.tmpdirname    ,     **snake_case__   )






				def 							_lowerCamelCase		(						self					:      List[Any]    ,     **a					:      Optional[Any]   ):






							'''simple docstring'''







							kwargs.update(self.special_tokens_map   )
							return CodeGenTokenizerFast.from_pretrained(self.tmpdirname    ,     **snake_case__   )






				def 							_lowerCamelCase		(						self					:      Dict    ,     a					:      Dict   ):






							'''simple docstring'''







							lowerCAmelCase__      :      List[str]         =		"lower newer"
							lowerCAmelCase__      :      Tuple         =		"lower newer"
							return input_text, output_text






				def 							_lowerCamelCase		(						self					:      Optional[int]   ):






							'''simple docstring'''







							lowerCAmelCase__      :      int         =		CodeGenTokenizer(self.vocab_file    ,     self.merges_file    ,     **self.special_tokens_map   )
							lowerCAmelCase__      :      Union[str, Any]         =		"lower newer"
							lowerCAmelCase__      :      Optional[int]         =		["\u0120low", "er", "\u0120", "n", "e", "w", "er"]
							lowerCAmelCase__      :      List[str]         =		tokenizer.tokenize(snake_case__    ,     add_prefix_space=snake_case__   )
							self.assertListEqual(snake_case__    ,     snake_case__   )

							lowerCAmelCase__      :      int         =		tokens + [tokenizer.unk_token]
							lowerCAmelCase__      :      Union[str, Any]         =		[14, 15, 10, 9, 3, 2, 15, 19]
							self.assertListEqual(tokenizer.convert_tokens_to_ids(snake_case__   )    ,     snake_case__   )






				def 							_lowerCamelCase		(						self					:      Optional[int]   ):






							'''simple docstring'''







							if not self.test_rust_tokenizer:
										return

							lowerCAmelCase__      :      Any         =		self.get_tokenizer()
							lowerCAmelCase__      :      Union[str, Any]         =		self.get_rust_tokenizer(add_prefix_space=snake_case__   )

							lowerCAmelCase__      :      Union[str, Any]         =		"lower newer"

							# Testing tokenization
							lowerCAmelCase__      :      Optional[Any]         =		tokenizer.tokenize(snake_case__    ,     add_prefix_space=snake_case__   )
							lowerCAmelCase__      :      List[str]         =		rust_tokenizer.tokenize(snake_case__   )
							self.assertListEqual(snake_case__    ,     snake_case__   )

							# Testing conversion to ids without special tokens
							lowerCAmelCase__      :      Union[str, Any]         =		tokenizer.encode(snake_case__    ,     add_special_tokens=snake_case__    ,     add_prefix_space=snake_case__   )
							lowerCAmelCase__      :      Optional[int]         =		rust_tokenizer.encode(snake_case__    ,     add_special_tokens=snake_case__   )
							self.assertListEqual(snake_case__    ,     snake_case__   )

							# Testing conversion to ids with special tokens
							lowerCAmelCase__      :      Tuple         =		self.get_rust_tokenizer(add_prefix_space=snake_case__   )
							lowerCAmelCase__      :      List[Any]         =		tokenizer.encode(snake_case__    ,     add_prefix_space=snake_case__   )
							lowerCAmelCase__      :      int         =		rust_tokenizer.encode(snake_case__   )
							self.assertListEqual(snake_case__    ,     snake_case__   )

							# Testing the unknown token
							lowerCAmelCase__      :      str         =		tokens + [rust_tokenizer.unk_token]
							lowerCAmelCase__      :      List[str]         =		[14, 15, 10, 9, 3, 2, 15, 19]
							self.assertListEqual(rust_tokenizer.convert_tokens_to_ids(snake_case__   )    ,     snake_case__   )






				def 							_lowerCamelCase		(						self					:      Union[str, Any]    ,     *a					:      Tuple    ,     **a					:      List[str]   ):






							'''simple docstring'''







							pass






				def 							_lowerCamelCase		(						self					:      Tuple    ,     a					:      Optional[Any]=15   ):






							'''simple docstring'''







							for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
										with self.subTest(f'''{tokenizer.__class__.__name__} ({pretrained_name})'''   ):
													lowerCAmelCase__      :      List[Any]         =		self.rust_tokenizer_class.from_pretrained(snake_case__    ,     **snake_case__   )

													# Simple input
													lowerCAmelCase__      :      Any         =		"This is a simple input"
													lowerCAmelCase__      :      Tuple         =		["This is a simple input 1", "This is a simple input 2"]
													lowerCAmelCase__      :      Dict         =		("This is a simple input", "This is a pair")
													lowerCAmelCase__      :      Union[str, Any]         =		[
													    ("This is a simple input 1", "This is a simple input 2"),
													    ("This is a simple pair 1", "This is a simple pair 2"),
													]

													# Simple input tests
													self.assertRaises(snake_case__    ,     tokenizer_r.encode    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'   )

													# Simple input
													self.assertRaises(snake_case__    ,     tokenizer_r.encode_plus    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'   )

													# Simple input
													self.assertRaises(
													    snake_case__    ,     tokenizer_r.batch_encode_plus    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'    ,     )

													# Pair input
													self.assertRaises(snake_case__    ,     tokenizer_r.encode    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'   )

													# Pair input
													self.assertRaises(snake_case__    ,     tokenizer_r.encode_plus    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'   )

													# Pair input
													self.assertRaises(
													    snake_case__    ,     tokenizer_r.batch_encode_plus    ,     snake_case__    ,     max_length=snake_case__    ,     padding='max_length'    ,     )






				def 							_lowerCamelCase		(						self					:      Any   ):






							'''simple docstring'''







							lowerCAmelCase__      :      int         =		CodeGenTokenizer.from_pretrained(self.tmpdirname    ,     pad_token='<pad>'   )

							# Simple input
							lowerCAmelCase__      :      str         =		"This is a simple input"
							lowerCAmelCase__      :      List[Any]         =		["This is a simple input looooooooong", "This is a simple input"]
							lowerCAmelCase__      :      Any         =		("This is a simple input", "This is a pair")
							lowerCAmelCase__      :      Tuple         =		[
							    ("This is a simple input loooooong", "This is a simple input"),
							    ("This is a simple pair loooooong", "This is a simple pair"),
							]

							lowerCAmelCase__      :      Optional[int]         =		tokenizer.pad_token_id

							lowerCAmelCase__      :      Optional[Any]         =		tokenizer(snake_case__    ,     padding='max_length'    ,     max_length=30    ,     return_tensors='np'   )
							lowerCAmelCase__      :      Optional[Any]         =		tokenizer(snake_case__    ,     padding=snake_case__    ,     truncate=snake_case__    ,     return_tensors='np'   )
							lowerCAmelCase__      :      List[Any]         =		tokenizer(*snake_case__    ,     padding='max_length'    ,     max_length=60    ,     return_tensors='np'   )
							lowerCAmelCase__      :      Dict         =		tokenizer(snake_case__    ,     padding=snake_case__    ,     truncate=snake_case__    ,     return_tensors='np'   )

							# s
							# test single string max_length padding
							self.assertEqual(out_s['input_ids'].shape[-1]    ,     30   )
							self.assertTrue(pad_token_id in out_s['input_ids']   )
							self.assertTrue(0 in out_s['attention_mask']   )

							# s2
							# test automatic padding
							self.assertEqual(out_sa['input_ids'].shape[-1]    ,     33   )
							# long slice doesn't have padding
							self.assertFalse(pad_token_id in out_sa['input_ids'][0]   )
							self.assertFalse(0 in out_sa['attention_mask'][0]   )
							# short slice does have padding
							self.assertTrue(pad_token_id in out_sa['input_ids'][1]   )
							self.assertTrue(0 in out_sa['attention_mask'][1]   )

							# p
							# test single pair max_length padding
							self.assertEqual(out_p['input_ids'].shape[-1]    ,     60   )
							self.assertTrue(pad_token_id in out_p['input_ids']   )
							self.assertTrue(0 in out_p['attention_mask']   )

							# p2
							# test automatic padding pair
							self.assertEqual(out_pa['input_ids'].shape[-1]    ,     52   )
							# long slice pair doesn't have padding
							self.assertFalse(pad_token_id in out_pa['input_ids'][0]   )
							self.assertFalse(0 in out_pa['attention_mask'][0]   )
							# short slice pair does have padding
							self.assertTrue(pad_token_id in out_pa['input_ids'][1]   )
							self.assertTrue(0 in out_pa['attention_mask'][1]   )






				def 							_lowerCamelCase		(						self					:      List[Any]   ):






							'''simple docstring'''







							lowerCAmelCase__      :      Optional[Any]         =		"$$$"
							lowerCAmelCase__      :      int         =		CodeGenTokenizer.from_pretrained(self.tmpdirname    ,     bos_token=snake_case__    ,     add_bos_token=snake_case__   )

							lowerCAmelCase__      :      int         =		"This is a simple input"
							lowerCAmelCase__      :      int         =		["This is a simple input 1", "This is a simple input 2"]

							lowerCAmelCase__      :      int         =		tokenizer.bos_token_id

							lowerCAmelCase__      :      Optional[int]         =		tokenizer(snake_case__   )
							lowerCAmelCase__      :      int         =		tokenizer(snake_case__   )

							self.assertEqual(out_s.input_ids[0]    ,     snake_case__   )
							self.assertTrue(all(o[0] == bos_token_id for o in out_sa.input_ids   )   )

							lowerCAmelCase__      :      int         =		tokenizer.decode(out_s.input_ids   )
							lowerCAmelCase__      :      List[Any]         =		tokenizer.batch_decode(out_sa.input_ids   )

							self.assertEqual(decode_s.split()[0]    ,     snake_case__   )
							self.assertTrue(all(d.split()[0] == bos_token for d in decode_sa   )   )






				@slow
				def 							_lowerCamelCase		(						self					:      List[str]   ):






							'''simple docstring'''







							lowerCAmelCase__      :      List[str]         =		CodeGenTokenizer.from_pretrained('Salesforce/codegen-350M-mono'   )

							lowerCAmelCase__      :      Any         =		"\nif len_a > len_b:\n    result = a\nelse:\n    result = b\n\n\n\n#"
							lowerCAmelCase__      :      List[Any]         =		"\nif len_a > len_b:      result = a\nelse:      result = b"

							lowerCAmelCase__      :      List[str]         =		tokenizer.encode(snake_case__   )
							lowerCAmelCase__      :      Any         =		["^#", re.escape('<|endoftext|>'   ), "^'''", "^\"\"\"", "\n\n\n"]
							lowerCAmelCase__      :      int         =		tokenizer.decode(snake_case__    ,     truncate_before_pattern=snake_case__   )
							self.assertEqual(snake_case__    ,     snake_case__   )






				def 							_lowerCamelCase		(						self					:      int   ):






							'''simple docstring'''







							pass



"""simple docstring"""







from ...configuration_utils import PretrainedConfig
from ...utils import logging


lowerCAmelCase__      				=   logging.get_logger(__name__)

lowerCAmelCase__      				=   {}


class       SCREAMING_SNAKE_CASE__				(		lowercase				):
       """simple docstring"""


       a  :				str   ="llama"
       a  :				List[str]   =["past_key_values"]

       def __init__(						self		,     snake_case__=32_000		,     snake_case__=4_096		,     snake_case__=11_008		,     snake_case__=32		,     snake_case__=32		,     snake_case__=None		,     snake_case__="silu"		,     snake_case__=2_048		,     snake_case__=0.02		,     snake_case__=1e-6		,     snake_case__=True		,     snake_case__=0		,     snake_case__=1		,     snake_case__=2		,     snake_case__=1		,     snake_case__=False		,     snake_case__=None		,     **snake_case__		,     ):






         """simple docstring"""





         lowerCAmelCase     :     Optional[Any]								=							vocab_size
         lowerCAmelCase     :     str								=							max_position_embeddings
         lowerCAmelCase     :     str								=							hidden_size
         lowerCAmelCase     :     Optional[int]								=							intermediate_size
         lowerCAmelCase     :     Any								=							num_hidden_layers
         lowerCAmelCase     :     List[str]								=							num_attention_heads

         # for backward compatibility
         if num_key_value_heads is None:
           lowerCAmelCase     :     Tuple								=							num_attention_heads

         lowerCAmelCase     :     Dict								=							num_key_value_heads
         lowerCAmelCase     :     Optional[Any]								=							hidden_act
         lowerCAmelCase     :     Optional[Any]								=							initializer_range
         lowerCAmelCase     :     Any								=							rms_norm_eps
         lowerCAmelCase     :     List[Any]								=							pretraining_tp
         lowerCAmelCase     :     int								=							use_cache
         lowerCAmelCase     :     List[str]								=							rope_scaling
         self._rope_scaling_validation()

         super().__init__(
             pad_token_id=snake_case__		,     bos_token_id=snake_case__		,     eos_token_id=snake_case__		,     tie_word_embeddings=snake_case__		,     **snake_case__		,     )







       def        lowercase__							(						self						):






         """simple docstring"""





         if self.rope_scaling is None:
           return

         if not isinstance(self.rope_scaling		,     snake_case__						) or len(self.rope_scaling						) != 2:
           raise ValueError(
               "`rope_scaling` must be a dictionary with with two fields, `name` and `factor`, "
               f"""got {self.rope_scaling}"""						)
         lowerCAmelCase     :     Optional[Any]								=							self.rope_scaling.get("type"		,     snake_case__						)
         lowerCAmelCase     :     int								=							self.rope_scaling.get("factor"		,     snake_case__						)
         if rope_scaling_type is None or rope_scaling_type not in ["linear", "dynamic"]:
           raise ValueError(
               f"""`rope_scaling`'s name field must be one of ['linear', 'dynamic'], got {rope_scaling_type}"""						)
         if rope_scaling_factor is None or not isinstance(snake_case__		,     snake_case__						) or rope_scaling_factor <= 1.0:
           raise ValueError(f"""`rope_scaling`'s factor field must be an float > 1, got {rope_scaling_factor}"""						)





