


import unittest

from transformers import ErnieMTokenizer
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin


_UpperCAmelCase      :	Tuple  =get_tests_dir("""fixtures/spiece.model""")







@require_sentencepiece
@require_tokenizers
class 					snake_case__(				UpperCAmelCase__,    unittest.TestCase				):







						'''simple docstring'''




						SCREAMING_SNAKE_CASE__   :	Any       =				ErnieMTokenizer
						SCREAMING_SNAKE_CASE__   :	str       =				False
						SCREAMING_SNAKE_CASE__   :	str       =				True
						SCREAMING_SNAKE_CASE__   :	Optional[int]       =				False
						SCREAMING_SNAKE_CASE__   :	Any       =				False




						def     lowercase_			(					self				)       ->     Optional[int]:
										super().setUp()

										# We have a SentencePiece fixture for testing
										lowerCAmelCase_			:   Tuple     					=					ErnieMTokenizer(__lowercase ,    unk_token='''<unk>''' ,    pad_token='''<pad>'''				)
										tokenizer.save_pretrained(self.tmpdirname				)




						def     lowercase_			(					self ,    __lowercase				)       ->     Optional[int]:
										lowerCAmelCase_			:   str     					=					'''this is a test'''
										lowerCAmelCase_			:   str     					=					'''this is a test'''
										return input_text, output_text




						def     lowercase_			(					self				)       ->     Optional[Any]:
										lowerCAmelCase_			:   int     					=					'''<pad>'''
										lowerCAmelCase_			:   Optional[Any]     					=					0

										self.assertEqual(self.get_tokenizer()._convert_token_to_id(__lowercase				) ,    __lowercase				)
										self.assertEqual(self.get_tokenizer()._convert_id_to_token(__lowercase				) ,    __lowercase				)




						def     lowercase_			(					self				)       ->     Optional[Any]:
										lowerCAmelCase_			:   int     					=					list(self.get_tokenizer().get_vocab().keys()				)

										self.assertEqual(vocab_keys[0] ,    '''<pad>'''				)
										self.assertEqual(vocab_keys[1] ,    '''<unk>'''				)
										self.assertEqual(vocab_keys[-1] ,    '''▁eloquent'''				)
										self.assertEqual(len(__lowercase				) ,    3_0_0_0_0				)




						def     lowercase_			(					self				)       ->     Optional[Any]:
										self.assertEqual(self.get_tokenizer().vocab_size ,    3_0_0_0_0				)




						def     lowercase_			(					self				)       ->     List[Any]:
										if not self.test_rust_tokenizer:
														return

										lowerCAmelCase_			:   str     					=					self.get_tokenizer()
										lowerCAmelCase_			:   Dict     					=					self.get_rust_tokenizer()

										lowerCAmelCase_			:   Tuple     					=					'''I was born in 92000, and this is falsé.'''

										lowerCAmelCase_			:   Any     					=					tokenizer.tokenize(__lowercase				)
										lowerCAmelCase_			:   Union[str, Any]     					=					rust_tokenizer.tokenize(__lowercase				)
										self.assertListEqual(__lowercase ,    __lowercase				)

										lowerCAmelCase_			:   Union[str, Any]     					=					tokenizer.encode(__lowercase ,    add_special_tokens=__lowercase				)
										lowerCAmelCase_			:   Optional[int]     					=					rust_tokenizer.encode(__lowercase ,    add_special_tokens=__lowercase				)
										self.assertListEqual(__lowercase ,    __lowercase				)

										lowerCAmelCase_			:   Tuple     					=					self.get_rust_tokenizer()
										lowerCAmelCase_			:   Optional[Any]     					=					tokenizer.encode(__lowercase				)
										lowerCAmelCase_			:   List[str]     					=					rust_tokenizer.encode(__lowercase				)
										self.assertListEqual(__lowercase ,    __lowercase				)




						def     lowercase_			(					self				)       ->     Optional[int]:
										lowerCAmelCase_			:   Tuple     					=					ErnieMTokenizer(__lowercase ,    do_lower_case=__lowercase ,    unk_token='''<unk>''' ,    pad_token='''<pad>'''				)

										lowerCAmelCase_			:   List[Any]     					=					tokenizer.tokenize('''This is a test'''				)
										self.assertListEqual(__lowercase ,    ['''▁this''', '''▁is''', '''▁a''', '''▁test''']				)

										self.assertListEqual(tokenizer.convert_tokens_to_ids(__lowercase				) ,    [4_8, 2_5, 2_1, 1_2_8_9]				)

										lowerCAmelCase_			:   Tuple     					=					tokenizer.tokenize('''I was born in 92000, and this is falsé.'''				)
										# ErnieMTokenizer(paddlenlp implementation) outputs '9' instead of '_9' so to mimic that '_9' is changed to '9'
										self.assertListEqual(
										    __lowercase ,    ['''▁i''', '''▁was''', '''▁born''', '''▁in''', '''9''', '''2000''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁fal''', '''s''', '''é''', '''.''']				)
										lowerCAmelCase_			:   List[str]     					=					tokenizer.convert_tokens_to_ids(__lowercase				)
										self.assertListEqual(__lowercase ,    [3_1, 2_3, 3_8_6, 1_9, 5_1_8, 3_0_5_0, 1_5, 1_7, 4_8, 2_5, 8_2_5_6, 1_8, 1, 9]				)

										lowerCAmelCase_			:   str     					=					tokenizer.convert_ids_to_tokens(__lowercase				)
										self.assertListEqual(
										    __lowercase ,    ['''▁i''', '''▁was''', '''▁born''', '''▁in''', '''9''', '''2000''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁fal''', '''s''', '''<unk>''', '''.'''] ,    )




						def     lowercase_			(					self				)       ->     Optional[Any]:
										lowerCAmelCase_			:   Optional[int]     					=					ErnieMTokenizer(__lowercase ,    unk_token='''<unk>''' ,    pad_token='''<pad>'''				)

										lowerCAmelCase_			:   Optional[Any]     					=					tokenizer.encode('''sequence builders'''				)
										lowerCAmelCase_			:   List[Any]     					=					tokenizer.encode('''multi-sequence build'''				)

										lowerCAmelCase_			:   Union[str, Any]     					=					tokenizer.build_inputs_with_special_tokens(__lowercase				)
										lowerCAmelCase_			:   str     					=					tokenizer.build_inputs_with_special_tokens(__lowercase ,    __lowercase				)

										assert encoded_sentence == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id]
										assert encoded_pair == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id] + [
										    tokenizer.sep_token_id
										] + text_a + [tokenizer.sep_token_id]




						@slow
						def     lowercase_			(					self				)       ->     Dict:
										# fmt: off
										lowerCAmelCase_			:   int     					=					{'''input_ids''': [[0, 1_1_0_6_2, 8_2_7_7_2, 7, 1_5, 8_2_7_7_2, 5_3_8, 5_1_5_2_9, 2_3_7, 1_7_1_9_8, 1_2_9_0, 2_0_6, 9, 2_1_5_1_7_5, 1_3_1_4, 1_3_6, 1_7_1_9_8, 1_2_9_0, 2_0_6, 9, 5_6_3_5_9, 4_2, 1_2_2_0_0_9, 9, 1_6_4_6_6, 1_6, 8_7_3_4_4, 4_5_3_7, 9, 4_7_1_7, 7_8_3_8_1, 6, 1_5_9_9_5_8, 7, 1_5, 2_4_4_8_0, 6_1_8, 4, 5_2_7, 2_2_6_9_3, 9, 3_0_4, 4, 2_7_7_7, 2_4_4_8_0, 9_8_7_4, 4, 4_3_5_2_3, 5_9_4, 4, 8_0_3, 1_8_3_9_2, 3_3_1_8_9, 1_8, 4, 4_3_5_2_3, 2_4_4_4_7, 5, 5, 5, 1_6, 1_0_0, 2_4_9_5_5, 8_3_6_5_8, 9_6_2_6, 1_4_4_0_5_7, 1_5, 8_3_9, 2_2_3_3_5, 1_6, 1_3_6, 2_4_9_5_5, 8_3_6_5_8, 8_3_4_7_9, 1_5, 3_9_1_0_2, 7_2_4, 1_6, 6_7_8, 6_4_5, 6_4_6_0, 1_3_2_8, 4_5_8_9, 4_2, 1_2_2_0_0_9, 1_1_5_7_7_4, 2_3, 3_5_5_9, 1_3_2_8, 4_6_8_7_6, 7, 1_3_6, 5_3_8_9_4, 1_9_4_0, 4_2_2_2_7, 4_1_1_5_9, 1_7_7_2_1, 8_2_3, 4_2_5, 4, 2_7_5_1_2, 9_8_7_2_2, 2_0_6, 1_3_6, 5_5_3_1, 4_9_7_0, 9_1_9, 1_7_3_3_6, 5, 2], [0, 2_0_0_8_0, 6_1_8, 8_3, 8_2_7_7_5, 4_7, 4_7_9, 9, 1_5_1_7, 7_3, 5_3_8_9_4, 3_3_3, 8_0_5_8_1, 1_1_0_1_1_7, 1_8_8_1_1, 5_2_5_6, 1_2_9_5, 5_1, 1_5_2_5_2_6, 2_9_7, 7_9_8_6, 3_9_0, 1_2_4_4_1_6, 5_3_8, 3_5_4_3_1, 2_1_4, 9_8, 1_5_0_4_4, 2_5_7_3_7, 1_3_6, 7_1_0_8, 4_3_7_0_1, 2_3, 7_5_6, 1_3_5_3_5_5, 7, 5, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1], [0, 5_8_1, 6_3_7_7_3, 1_1_9_4_5_5, 6, 1_4_7_7_9_7, 8_8_2_0_3, 7, 6_4_5, 7_0, 2_1, 3_2_8_5, 1_0_2_6_9, 5, 2, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]]}
										# fmt: on

										self.tokenizer_integration_test_util(
										    expected_encoding=__lowercase ,    model_name='''susnato/ernie-m-base_pytorch''' ,    sequences=[
										        '''Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides '''
										        '''general-purpose architectures (BERT, GPT-2, RoBERTa, XLM, DistilBert, XLNet...) for Natural '''
										        '''Language Understanding (NLU) and Natural Language Generation (NLG) with over32+ pretrained '''
										        '''models in100+ languages and deep interoperability between Jax, PyTorch and TensorFlow.''',
										        '''BERT is designed to pre-train deep bidirectional representations from unlabeled text by jointly '''
										        '''conditioning on both left and right context in all layers.''',
										        '''The quick brown fox jumps over the lazy dog.''',
										    ] ,    )
'''simple docstring'''
def 				__UpperCamelCase (					__UpperCAmelCase  :					float ,   __UpperCAmelCase  :					float ,   __UpperCAmelCase  :					float			)							->	list[int]:
  if hue < 0 or hue > 360:
    raise Exception('''hue should be between 0 and 360'''			)

  if saturation < 0 or saturation > 1:
    raise Exception('''saturation should be between 0 and 1'''			)

  if value < 0 or value > 1:
    raise Exception('''value should be between 0 and 1'''			)

  lowercase				:							List[str]									=   value * saturation
  lowercase				:							Optional[Any]									=   hue / 60
  lowercase				:							Dict									=   chroma * (1 - abs(hue_section % 2 - 1			))
  lowercase				:							List[Any]									=   value - chroma

  if hue_section >= 0 and hue_section <= 1:
    lowercase				:							Optional[int]									=   round(255 * (chroma + match_value)			)
    lowercase				:							List[Any]									=   round(255 * (second_largest_component + match_value)			)
    lowercase				:							int									=   round(255 * (match_value)			)
  elif hue_section > 1 and hue_section <= 2:
    lowercase				:							Optional[Any]									=   round(255 * (second_largest_component + match_value)			)
    lowercase				:							Any									=   round(255 * (chroma + match_value)			)
    lowercase				:							Any									=   round(255 * (match_value)			)
  elif hue_section > 2 and hue_section <= 3:
    lowercase				:							Dict									=   round(255 * (match_value)			)
    lowercase				:							Optional[int]									=   round(255 * (chroma + match_value)			)
    lowercase				:							List[str]									=   round(255 * (second_largest_component + match_value)			)
  elif hue_section > 3 and hue_section <= 4:
    lowercase				:							List[str]									=   round(255 * (match_value)			)
    lowercase				:							List[str]									=   round(255 * (second_largest_component + match_value)			)
    lowercase				:							Tuple									=   round(255 * (chroma + match_value)			)
  elif hue_section > 4 and hue_section <= 5:
    lowercase				:							Optional[int]									=   round(255 * (second_largest_component + match_value)			)
    lowercase				:							Tuple									=   round(255 * (match_value)			)
    lowercase				:							Optional[int]									=   round(255 * (chroma + match_value)			)
  else:
    lowercase				:							int									=   round(255 * (chroma + match_value)			)
    lowercase				:							Optional[Any]									=   round(255 * (match_value)			)
    lowercase				:							Optional[int]									=   round(255 * (second_largest_component + match_value)			)

  return [red, green, blue]







def 				__UpperCamelCase (					__UpperCAmelCase  :					int ,   __UpperCAmelCase  :					int ,   __UpperCAmelCase  :					int			)							->	list[float]:
  if red < 0 or red > 255:
    raise Exception('''red should be between 0 and 255'''			)

  if green < 0 or green > 255:
    raise Exception('''green should be between 0 and 255'''			)

  if blue < 0 or blue > 255:
    raise Exception('''blue should be between 0 and 255'''			)

  lowercase				:							str									=   red / 255
  lowercase				:							Union[str, Any]									=   green / 255
  lowercase				:							Union[str, Any]									=   blue / 255
  lowercase				:							str									=   max(__UpperCAmelCase ,   __UpperCAmelCase ,   __UpperCAmelCase			)
  lowercase				:							int									=   value - min(__UpperCAmelCase ,   __UpperCAmelCase ,   __UpperCAmelCase			)
  lowercase				:							Optional[int]									=   0 if value == 0 else chroma / value

  if chroma == 0:
    lowercase				:							Dict									=   0.0
  elif value == float_red:
    lowercase				:							Tuple									=   60 * (0 + (float_green - float_blue) / chroma)
  elif value == float_green:
    lowercase				:							Optional[int]									=   60 * (2 + (float_blue - float_red) / chroma)
  else:
    lowercase				:							Dict									=   60 * (4 + (float_red - float_green) / chroma)

  lowercase				:							List[str]									=   (hue + 360) % 360

  return [hue, saturation, value]







def 				__UpperCamelCase (					__UpperCAmelCase  :					list[float] ,   __UpperCAmelCase  :					list[float]			)							->	bool:
  lowercase				:							int									=   abs(hsv_a[0] - hsv_a[0]			) < 0.2
  lowercase				:							Optional[int]									=   abs(hsv_a[1] - hsv_a[1]			) < 0.002
  lowercase				:							Dict									=   abs(hsv_a[2] - hsv_a[2]			) < 0.002

  return check_hue and check_saturation and check_value