





import unittest

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

from ...test_tokenization_common import TokenizerTesterMixin


a__      :     Optional[Any]         =		get_tests_dir('fixtures/test_sentencepiece_with_bytefallback.model')





@require_sentencepiece
@require_tokenizers
class UpperCAmelCase_				( SCREAMING_SNAKE_CASE_     ,		unittest.TestCase ):
  __lowerCAmelCase   :						Optional[int]							=   GPTSwaTokenizer
  __lowerCAmelCase   :						List[str]							=   False
  __lowerCAmelCase   :						List[str]							=   True
  __lowerCAmelCase   :						Tuple							=   False


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       super().setUp()

       # We have a SentencePiece fixture for testing
       A_          =       GPTSwaTokenizer(snake_case__ ,eos_token='''<unk>''' ,bos_token='''<unk>''' ,pad_token='''<unk>''' )

       tokenizer.save_pretrained(self.tmpdirname )


  def 						__UpperCAmelCase     (							self ,__snake_case ):



       """simple docstring"""





       A_          =       'This is a test'
       A_          =       'This is a test'
       return input_text, output_text


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       A_          =       '<s>'
       A_          =       1

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


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       A_          =       list(self.get_tokenizer().get_vocab().keys() )

       self.assertEqual(vocab_keys[0] ,'''<unk>''' )
       self.assertEqual(vocab_keys[1] ,'''<s>''' )
       self.assertEqual(vocab_keys[-1] ,'''j''' )
       self.assertEqual(len(snake_case__ ) ,2_0_0_0 )


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       self.assertEqual(self.get_tokenizer().vocab_size ,2_0_0_0 )


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       A_          =       GPTSwaTokenizer(snake_case__ )

       A_          =       tokenizer.tokenize('''This is a test''' )
       self.assertListEqual(snake_case__ ,['''▁This''', '''▁is''', '''▁a''', '''▁t''', '''est'''] )

       self.assertListEqual(tokenizer.convert_tokens_to_ids(snake_case__ ) ,[4_6_5, 2_8_7, 2_6_5, 6_3_1, 8_4_2] )

       A_          =       tokenizer.tokenize('''I was born in 92000, and this is falsé.''' )
       # fmt: off
       self.assertListEqual(
           snake_case__ ,['''▁I''', '''▁was''', '''▁bor''', '''n''', '''▁in''', '''▁''', '''<0x39>''', '''2''', '''0''', '''0''', '''0''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁f''', '''al''', '''s''', '''<0xC3>''', '''<0xA9>''', '''.'''] ,)
       # fmt: on

       A_          =       tokenizer.convert_tokens_to_ids(snake_case__ )
       self.assertListEqual(
           snake_case__ ,[2_6_2, 2_7_2, 1_5_2_5, 2_8_6, 2_7_1, 2_6_8, 6_0, 9_1_6, 6_3_3, 6_3_3, 6_3_3, 2_5_9, 2_6_6, 3_0_1, 2_8_7, 3_8_4, 3_6_7, 2_6_3, 1_9_8, 1_7_2, 2_6_0] ,)

       A_          =       tokenizer.convert_ids_to_tokens(snake_case__ )
       # fmt: off
       self.assertListEqual(
           snake_case__ ,['''▁I''', '''▁was''', '''▁bor''', '''n''', '''▁in''', '''▁''', '''<0x39>''', '''2''', '''0''', '''0''', '''0''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁f''', '''al''', '''s''', '''<0xC3>''', '''<0xA9>''', '''.'''] )
       # fmt: on


  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       A_          =       GPTSwaTokenizer(snake_case__ )
       A_          =       ['This is a test', 'I was born in 92000, and this is falsé.']
       A_          =       [
           [4_6_5, 2_8_7, 2_6_5, 6_3_1, 8_4_2],
           [2_6_2, 2_7_2, 1_5_2_5, 2_8_6, 2_7_1, 2_6_8, 6_0, 9_1_6, 6_3_3, 6_3_3, 6_3_3, 2_5_9, 2_6_6, 3_0_1, 2_8_7, 3_8_4, 3_6_7, 2_6_3, 1_9_8, 1_7_2, 2_6_0],
       ]

       # Test that encode_fast returns the same as tokenize + convert_tokens_to_ids
       for text, expected_ids in zip(snake_case__ ,snake_case__ ):
            self.assertListEqual(tokenizer.encode_fast(snake_case__ ) ,snake_case__ )

       # Test that decode_fast returns the input text
       for text, token_ids in zip(snake_case__ ,snake_case__ ):
            self.assertEqual(tokenizer.decode_fast(snake_case__ ) ,snake_case__ )




  @slow
  def 						__UpperCAmelCase     (							self ):



       """simple docstring"""





       A_          =       [
           '<|python|>def fibonacci(n)\n    if n < 0:\n        print(\'Incorrect input\')',
           'Hey there, how are you doing this fine day?',
           'This is a text with a trailing spaces followed by a dot     .',
           'Häj sväjs lillebrör! =)',
           'Det är inget fel på Mr. Cool',
       ]

       # fmt: off
       A_          =       {'input_ids': [[6_3_4_2_3, 5, 6_8_1_1, 1_4_9_5_4, 2_8_2, 8_1_6, 3_8_2_1, 6_3_4_6_6, 6_3_4_2_5, 6_3_4_6_2, 1_8, 6_3_9_7_8, 6_7_8, 3_0_1, 1_3_2_0, 6_3_4_2_3, 6_3_4_5_5, 6_3_4_5_8, 1_8, 6_3_9_8_2, 4_2_4_6, 3_9_4_0, 1_9_0_1, 4_7_7_8_9, 5_5_4_7, 1_8_9_9_4], [1_9_6_3_0, 1_1_0_0, 6_3_4_4_6, 1_3_4_2, 6_3_3, 5_4_4, 4_4_8_8, 5_9_3, 5_1_0_2, 2_4_1_6, 6_3_4_9_5, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1_6_5_2, 4_2_8, 2_6_8, 1_9_3_6, 5_1_5, 2_6_8, 5_8_5_9_3, 2_2_4_1_3, 9_1_0_6, 5_4_6, 2_6_8, 3_3_2_1_3, 6_3_9_7_9, 6_9_8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [5_5_1_3_0, 6_3_4_5_0, 9_2_4, 6_3_4_4_9, 2_2_4_9, 4_0_6_2, 1_5_5_8, 3_1_8, 6_3_5_0_4, 2_1_4_9_8, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [5_0_9, 3_7_7, 2_8_2_7, 2_5_5_9, 3_3_2, 6_5_7_5, 6_3_4_4_3, 2_6_8_0_1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], 'token_type_ids': [[0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]], 'attention_mask': [[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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]]}
       # fmt: on
       self.tokenizer_integration_test_util(
           expected_encoding=snake_case__ ,model_name='''AI-Sweden/gpt-sw3-126m''' ,sequences=snake_case__ ,)







'''simple docstring'''







import unittest

from transformers import RoFormerTokenizer, RoFormerTokenizerFast
from transformers.testing_utils import require_rjieba, require_tokenizers

from ...test_tokenization_common import TokenizerTesterMixin
@require_rjieba
@require_tokenizers
class 						UpperCamelCase__       (			SCREAMING_SNAKE_CASE_     ,       unittest.TestCase						):



     """simple docstring"""






     __magic_name__           =     RoFormerTokenizer
     __magic_name__           =     RoFormerTokenizerFast
     __magic_name__           =     True
     __magic_name__           =     True







     def a						(			self ):

            '''simple docstring'''

            super().setUp()







     def a						(			self   ,						**snake_case__ ):

            '''simple docstring'''

            return self.tokenizer_class.from_pretrained('junnyu/roformer_chinese_base'   ,						**snake_case__ )







     def a						(			self   ,						**snake_case__ ):

            '''simple docstring'''

            return self.rust_tokenizer_class.from_pretrained('junnyu/roformer_chinese_base'   ,						**snake_case__ )







     def a						(			self ):

            '''simple docstring'''

            _lowerCAmelCase     :				List[Any]       =							'永和服装饰品有限公司,今天天气非常好'
            _lowerCAmelCase     :				str       =							'永和 服装 饰品 有限公司 , 今 天 天 气 非常 好'
            return input_text, output_text







     def a						(			self ):

            '''simple docstring'''

            _lowerCAmelCase     :				Optional[int]       =							self.get_tokenizer()
            _lowerCAmelCase      ,  _lowerCAmelCase     :				int       =							self.get_chinese_input_output_texts()
            _lowerCAmelCase     :				List[Any]       =							tokenizer.tokenize(snake_case__ )

            self.assertListEqual(snake_case__   ,						output_text.split() )

            _lowerCAmelCase     :				Optional[Any]       =							tokens + [tokenizer.unk_token]
            _lowerCAmelCase     :				List[str]       =							[2_2943, 2_1332, 3_4431, 4_5904, 117, 306, 1231, 1231, 2653, 3_3994, 1266, 100]
            self.assertListEqual(tokenizer.convert_tokens_to_ids(snake_case__ )   ,						snake_case__ )







     def a						(			self ):

            '''simple docstring'''

            _lowerCAmelCase     :				List[Any]       =							self.get_rust_tokenizer()
            _lowerCAmelCase      ,  _lowerCAmelCase     :				Any       =							self.get_chinese_input_output_texts()
            _lowerCAmelCase     :				Optional[int]       =							tokenizer.tokenize(snake_case__ )
            self.assertListEqual(snake_case__   ,						output_text.split() )
            _lowerCAmelCase     :				Tuple       =							tokens + [tokenizer.unk_token]
            _lowerCAmelCase     :				List[Any]       =							[2_2943, 2_1332, 3_4431, 4_5904, 117, 306, 1231, 1231, 2653, 3_3994, 1266, 100]
            self.assertListEqual(tokenizer.convert_tokens_to_ids(snake_case__ )   ,						snake_case__ )







     def a						(			self ):

            '''simple docstring'''

            pass







     def a						(			self ):

            '''simple docstring'''

            pass







     def a						(			self ):

            '''simple docstring'''

            pass

