




import json
import os
import re
import shutil
import tempfile
import unittest
from typing import Tuple

from transformers import AddedToken, BatchEncoding, ByTaTokenizer
from transformers.utils import cached_property, is_tf_available, is_torch_available

from ...test_tokenization_common import TokenizerTesterMixin


if is_torch_available():
    snake_case_ : Optional[Any]    		=       "pt"
elif is_tf_available():
    snake_case_ : Union[str, Any]    		=       "tf"
else:
    snake_case_ : str    		=       "jax"







class __snake_case			(			a ,  unittest.TestCase		):
   UpperCAmelCase__  :    List[Any]          =  ByTaTokenizer
   UpperCAmelCase__  :    int          =  False







   def lowerCamelCase						(      self		:						Optional[int]):

        """simple docstring"""


        super().setUp()
        UpperCAmelCase_          =							ByTaTokenizer()
        tokenizer.save_pretrained(self.tmpdirname)







   @cached_property
   def lowerCamelCase						(      self		:						Tuple):

        """simple docstring"""


        return ByTaTokenizer.from_pretrained('''google/byt5-small''')







   def lowerCamelCase						(      self		:						List[str]					,						**_snake_case		:						Union[str, Any]):

        """simple docstring"""


        return self.tokenizer_class.from_pretrained(self.tmpdirname					,						**_snake_case)







   def lowerCamelCase						(      self		:						Dict					,						_snake_case		:						int					,						_snake_case		:						Tuple=False					,						_snake_case		:						Dict=20					,						_snake_case		:						Optional[Any]=5):

        """simple docstring"""


        UpperCAmelCase_          =							[]
        for i in range(len(_snake_case)):
             try:
                  UpperCAmelCase_          =							tokenizer.decode([i]					,						clean_up_tokenization_spaces=_snake_case)
             except UnicodeDecodeError:
                  pass
             toks.append((i, tok))

        UpperCAmelCase_          =							list(filter(lambda _snake_case: re.match(r'''^[ a-zA-Z]+$'''					,						t[1])					,						_snake_case))
        UpperCAmelCase_          =							list(filter(lambda _snake_case: [t[0]] == tokenizer.encode(t[1]					,						add_special_tokens=_snake_case)					,						_snake_case))
        if max_length is not None and len(_snake_case) > max_length:
             UpperCAmelCase_          =							toks[:max_length]
        if min_length is not None and len(_snake_case) < min_length and len(_snake_case) > 0:
             while len(_snake_case) < min_length:
                  UpperCAmelCase_          =							toks + toks
        # toks_str = [t[1] for t in toks]
        UpperCAmelCase_          =							[t[0] for t in toks]

        # Ensure consistency
        UpperCAmelCase_          =							tokenizer.decode(_snake_case					,						clean_up_tokenization_spaces=_snake_case)
        if " " not in output_txt and len(_snake_case) > 1:
             UpperCAmelCase_          =							(
                 tokenizer.decode([toks_ids[0]]					,						clean_up_tokenization_spaces=_snake_case)
                 + ''' '''
                 + tokenizer.decode(toks_ids[1:]					,						clean_up_tokenization_spaces=_snake_case)
             )
        if with_prefix_space:
             UpperCAmelCase_          =							''' ''' + output_txt
        UpperCAmelCase_          =							tokenizer.encode(_snake_case					,						add_special_tokens=_snake_case)
        return output_txt, output_ids







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

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							tokenizer(['''hi</s>''', '''I went to the gym</s>''', '''</s>'''])
        UpperCAmelCase_          =							tokenizer(['''hi''', '''I went to the gym''', ''''''])
        self.assertListEqual(batch_with_eos_added['''input_ids''']					,						batch_without_eos_added['''input_ids'''])







   def lowerCamelCase						(      self		:						str):

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							'''Unicode €.'''
        UpperCAmelCase_          =							tokenizer(_snake_case)
        UpperCAmelCase_          =							[88, 113, 108, 102, 114, 103, 104, 35, 229, 133, 175, 49, 1]
        self.assertEqual(encoded['''input_ids''']					,						_snake_case)

        # decoding
        UpperCAmelCase_          =							tokenizer.decode(_snake_case)
        self.assertEqual(_snake_case					,						'''Unicode €.</s>''')

        UpperCAmelCase_          =							tokenizer('''e è é ê ë''')
        UpperCAmelCase_          =							[104, 35, 198, 171, 35, 198, 172, 35, 198, 173, 35, 198, 174, 1]
        self.assertEqual(encoded['''input_ids''']					,						_snake_case)
        # decoding
        UpperCAmelCase_          =							tokenizer.decode(_snake_case)
        self.assertEqual(_snake_case					,						'''e è é ê ë</s>''')

        # encode/decode, but with `encode` instead of `__call__`
        self.assertEqual(tokenizer.decode(tokenizer.encode('''e è é ê ë'''))					,						'''e è é ê ë</s>''')







   def lowerCamelCase						(      self		:						Any):

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							['''A long paragraph for summarization.''', '''Another paragraph for summarization.''']
        # fmt: off
        UpperCAmelCase_          =							[68, 35, 111, 114, 113, 106, 35, 115, 100, 117, 100, 106, 117, 100, 115, 107, 35, 105, 114, 117, 35, 118, 120, 112, 112, 100, 117, 108, 125, 100, 119, 108, 114, 113, 49, 1, 0]
        # fmt: on
        UpperCAmelCase_          =							tokenizer(_snake_case					,						padding=_snake_case					,						return_tensors=_snake_case)
        self.assertIsInstance(_snake_case					,						_snake_case)

        if FRAMEWORK != "jax":
             UpperCAmelCase_          =							list(batch.input_ids.numpy()[0])
        else:
             UpperCAmelCase_          =							list(batch.input_ids.tolist()[0])

        self.assertListEqual(_snake_case					,						_snake_case)

        self.assertEqual((2, 37)					,						batch.input_ids.shape)
        self.assertEqual((2, 37)					,						batch.attention_mask.shape)







   def lowerCamelCase						(      self		:						Optional[Any]):

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							['''A long paragraph for summarization.''', '''Another paragraph for summarization.''']
        UpperCAmelCase_          =							tokenizer(_snake_case					,						padding=_snake_case					,						return_tensors=_snake_case)
        # check if input_ids are returned and no decoder_input_ids
        self.assertIn('''input_ids'''					,						_snake_case)
        self.assertIn('''attention_mask'''					,						_snake_case)
        self.assertNotIn('''decoder_input_ids'''					,						_snake_case)
        self.assertNotIn('''decoder_attention_mask'''					,						_snake_case)







   def lowerCamelCase						(      self		:						Tuple):

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							[
            '''Summary of the text.''',
            '''Another summary.''',
        ]
        UpperCAmelCase_          =							tokenizer(
            text_target=_snake_case					,						max_length=32					,						padding='''max_length'''					,						truncation=_snake_case					,						return_tensors=_snake_case)
        self.assertEqual(32					,						targets['''input_ids'''].shape[1])







   def lowerCamelCase						(      self		:						int):

        """simple docstring"""


        UpperCAmelCase_          =							self.ta_base_tokenizer
        UpperCAmelCase_          =							['''A long paragraph for summarization. </s>''']
        UpperCAmelCase_          =							['''Summary of the text. </s>''']
        # fmt: off
        UpperCAmelCase_          =							[68, 35, 111, 114, 113, 106, 35, 115, 100, 117, 100, 106, 117, 100, 115, 107, 35, 105, 114, 117, 35, 118, 120, 112, 112, 100, 117, 108, 125, 100, 119, 108, 114, 113, 49, 35, 1]
        UpperCAmelCase_          =							[86, 120, 112, 112, 100, 117, 124, 35, 114, 105, 35, 119, 107, 104, 35, 119, 104, 123, 119, 49, 35, 1]
        # fmt: on

        UpperCAmelCase_          =							tokenizer(_snake_case					,						text_target=_snake_case)

        self.assertEqual(_snake_case					,						batch['''input_ids'''][0])
        self.assertEqual(_snake_case					,						batch['''labels'''][0])







   def lowerCamelCase						(      self		:						Tuple):

        """simple docstring"""


        UpperCAmelCase_          =							self.get_tokenizers()
        for tokenizer in tokenizers:
             with self.subTest(F"""{tokenizer.__class__.__name__}"""):
                  self.assertNotEqual(tokenizer.model_max_length					,						42)

        # Now let's start the test
        UpperCAmelCase_          =							self.get_tokenizers()
        for tokenizer in tokenizers:
             with self.subTest(F"""{tokenizer.__class__.__name__}"""):
                  # Isolate this from the other tests because we save additional tokens/etc
                  UpperCAmelCase_          =							tempfile.mkdtemp()

                  UpperCAmelCase_          =							''' He is very happy, UNwant\u00E9d,running'''
                  UpperCAmelCase_          =							tokenizer.encode(_snake_case					,						add_special_tokens=_snake_case)
                  tokenizer.save_pretrained(_snake_case)

                  UpperCAmelCase_          =							tokenizer.__class__.from_pretrained(_snake_case)
                  UpperCAmelCase_          =							after_tokenizer.encode(_snake_case					,						add_special_tokens=_snake_case)
                  self.assertListEqual(_snake_case					,						_snake_case)

                  shutil.rmtree(_snake_case)

        UpperCAmelCase_          =							self.get_tokenizers(model_max_length=42)
        for tokenizer in tokenizers:
             with self.subTest(F"""{tokenizer.__class__.__name__}"""):
                  # Isolate this from the other tests because we save additional tokens/etc
                  UpperCAmelCase_          =							tempfile.mkdtemp()

                  UpperCAmelCase_          =							''' He is very happy, UNwant\u00E9d,running'''
                  tokenizer.add_tokens(['''bim''', '''bambam'''])
                  UpperCAmelCase_          =							tokenizer.additional_special_tokens
                  additional_special_tokens.append('''new_additional_special_token''')
                  tokenizer.add_special_tokens({'''additional_special_tokens''': additional_special_tokens})
                  UpperCAmelCase_          =							tokenizer.encode(_snake_case					,						add_special_tokens=_snake_case)
                  tokenizer.save_pretrained(_snake_case)

                  UpperCAmelCase_          =							tokenizer.__class__.from_pretrained(_snake_case)
                  UpperCAmelCase_          =							after_tokenizer.encode(_snake_case					,						add_special_tokens=_snake_case)
                  self.assertListEqual(_snake_case					,						_snake_case)
                  self.assertIn('''new_additional_special_token'''					,						after_tokenizer.additional_special_tokens)
                  self.assertEqual(after_tokenizer.model_max_length					,						42)

                  UpperCAmelCase_          =							tokenizer.__class__.from_pretrained(_snake_case					,						model_max_length=43)
                  self.assertEqual(tokenizer.model_max_length					,						43)

                  shutil.rmtree(_snake_case)







   def lowerCamelCase						(      self		:						List[Any]):

        """simple docstring"""


        UpperCAmelCase_          =							[]
        if self.test_slow_tokenizer:
             tokenizer_list.append((self.tokenizer_class, self.get_tokenizer()))

        if self.test_rust_tokenizer:
             tokenizer_list.append((self.rust_tokenizer_class, self.get_rust_tokenizer()))

        for tokenizer_class, tokenizer_utils in tokenizer_list:
             with tempfile.TemporaryDirectory() as tmp_dir:
                  tokenizer_utils.save_pretrained(_snake_case)

                  with open(os.path.join(_snake_case					,						'''special_tokens_map.json''')					,						encoding='''utf-8''') as json_file:
                       UpperCAmelCase_          =							json.load(_snake_case)

                  with open(os.path.join(_snake_case					,						'''tokenizer_config.json''')					,						encoding='''utf-8''') as json_file:
                       UpperCAmelCase_          =							json.load(_snake_case)

                  UpperCAmelCase_          =							[F"""<extra_id_{i}>""" for i in range(125)]

                  UpperCAmelCase_          =							added_tokens_extra_ids + [
                      '''an_additional_special_token'''
                  ]
                  UpperCAmelCase_          =							added_tokens_extra_ids + [
                      '''an_additional_special_token'''
                  ]

                  with open(os.path.join(_snake_case					,						'''special_tokens_map.json''')					,						'''w'''					,						encoding='''utf-8''') as outfile:
                       json.dump(_snake_case					,						_snake_case)
                  with open(os.path.join(_snake_case					,						'''tokenizer_config.json''')					,						'''w'''					,						encoding='''utf-8''') as outfile:
                       json.dump(_snake_case					,						_snake_case)

                  # the following checks allow us to verify that our test works as expected, i.e. that the tokenizer takes
                  # into account the new value of additional_special_tokens given in the "tokenizer_config.json" and
                  # "special_tokens_map.json" files
                  UpperCAmelCase_          =							tokenizer_class.from_pretrained(
                      _snake_case					,						)
                  self.assertIn(
                      '''an_additional_special_token'''					,						tokenizer_without_change_in_init.additional_special_tokens)
                  # self.assertIn("an_additional_special_token",tokenizer_without_change_in_init.get_vocab()) # ByT5Tokenization no vocab
                  self.assertEqual(
                      ['''an_additional_special_token''']					,						tokenizer_without_change_in_init.convert_ids_to_tokens(
                          tokenizer_without_change_in_init.convert_tokens_to_ids(['''an_additional_special_token''']))					,						)

                  # Now we test that we can change the value of additional_special_tokens in the from_pretrained
                  UpperCAmelCase_          =							added_tokens_extra_ids + [AddedToken('''a_new_additional_special_token'''					,						lstrip=_snake_case)]
                  UpperCAmelCase_          =							tokenizer_class.from_pretrained(
                      _snake_case					,						additional_special_tokens=_snake_case					,						)

                  self.assertIn('''a_new_additional_special_token'''					,						tokenizer.additional_special_tokens)
                  self.assertEqual(
                      ['''a_new_additional_special_token''']					,						tokenizer.convert_ids_to_tokens(
                          tokenizer.convert_tokens_to_ids(['''a_new_additional_special_token''']))					,						)







   def lowerCamelCase						(      self		:						Any):

        """simple docstring"""


        UpperCAmelCase_          =							[]
        if self.test_slow_tokenizer:
             tokenizer_list.append((self.tokenizer_class, self.get_tokenizer()))

        if self.test_rust_tokenizer:
             tokenizer_list.append((self.rust_tokenizer_class, self.get_rust_tokenizer()))

        for tokenizer_class, tokenizer_utils in tokenizer_list:
             with tempfile.TemporaryDirectory() as tmp_dir:
                  tokenizer_utils.save_pretrained(_snake_case)

                  UpperCAmelCase_          =							tokenizer_class.from_pretrained(_snake_case)

                  self.assertTrue(tokenizer.decode([255]) == '''''')







   def lowerCamelCase						(      self		:						int):

        """simple docstring"""


        pass







   def lowerCamelCase						(      self		:						Optional[int]):

        """simple docstring"""


        pass







   def lowerCamelCase						(      self		:						Dict):

        """simple docstring"""


        pass







   def lowerCamelCase						(      self		:						List[Any]):

        """simple docstring"""


        pass







   def lowerCamelCase						(      self		:						Tuple):

        """simple docstring"""


        UpperCAmelCase_          =							self.get_tokenizers(fast=_snake_case					,						do_lower_case=_snake_case)
        for tokenizer in tokenizers:
             with self.subTest(F"""{tokenizer.__class__.__name__}"""):
                  UpperCAmelCase_          =							['''t''', '''h''', '''i''', '''s''', ''' ''', '''i''', '''s''', ''' ''', '''a''', ''' ''', '''t''', '''e''', '''x''', '''t''', '''</s>''']
                  UpperCAmelCase_          =							tokenizer.convert_tokens_to_string(_snake_case)

                  self.assertIsInstance(_snake_case					,						_snake_case)







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

        """simple docstring"""


        UpperCAmelCase_          =							self.get_tokenizers()
        for tokenizer in tokenizers:
             with self.subTest(F"""{tokenizer.__class__.__name__}"""):
                  UpperCAmelCase_          =							[
                      '''bos_token''',
                      '''eos_token''',
                      '''unk_token''',
                      '''sep_token''',
                      '''pad_token''',
                      '''cls_token''',
                      '''mask_token''',
                  ]

                  UpperCAmelCase_          =							0
                  UpperCAmelCase_          =							tokenizer.convert_ids_to_tokens(
                      _snake_case					,						skip_special_tokens=_snake_case)

                  for attr in attributes_list:
                       setattr(_snake_case					,						attr + '''_id'''					,						_snake_case)
                       self.assertEqual(getattr(_snake_case					,						_snake_case)					,						_snake_case)
                       self.assertEqual(getattr(_snake_case					,						attr + '''_id''')					,						_snake_case)

                       setattr(_snake_case					,						attr + '''_id'''					,						_snake_case)
                       self.assertEqual(getattr(_snake_case					,						_snake_case)					,						_snake_case)
                       self.assertEqual(getattr(_snake_case					,						attr + '''_id''')					,						_snake_case)

                  setattr(_snake_case					,						'''additional_special_tokens_ids'''					,						[])
                  self.assertListEqual(getattr(_snake_case					,						'''additional_special_tokens''')					,						[])
                  self.assertListEqual(getattr(_snake_case					,						'''additional_special_tokens_ids''')					,						[])

                  setattr(_snake_case					,						'''additional_special_tokens_ids'''					,						[token_id_to_test_setters])
                  self.assertListEqual(getattr(_snake_case					,						'''additional_special_tokens''')					,						[token_to_test_setters])
                  self.assertListEqual(getattr(_snake_case					,						'''additional_special_tokens_ids''')					,						[token_id_to_test_setters])










import os
import sys
import unittest


lowerCamelCase											=       os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, 'utils'))

import check_dummies  # noqa: E402
from check_dummies import create_dummy_files, create_dummy_object, find_backend, read_init  # noqa: E402


# Align TRANSFORMERS_PATH in check_dummies with the current path
lowerCamelCase											=       os.path.join(git_repo_path, 'src', 'transformers')

lowerCamelCase											=       '\n{0} = None\n'

lowerCamelCase											=       '\nclass {0}(metaclass=DummyObject):\n    _backends = {1}\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, {1})\n'


lowerCamelCase											=       '\ndef {0}(*args, **kwargs):\n    requires_backends({0}, {1})\n'




class A  (   unittest.TestCase     ):






      def        lowerCamelCase       (					self						:     Tuple )					->						Union[str, Any]:
             """simple docstring"""




             _lowerCamelCase       :		int		=find_backend('    _import_structure["models.albert"].append("AlbertTokenizerFast")' )
             self.assertIsNone(lowercase_ )

             _lowerCamelCase       :		List[str]		=find_backend('    if not is_tokenizers_available():' )
             self.assertEqual(lowercase_			,   'tokenizers' )

             _lowerCamelCase       :		List[Any]		=find_backend('    if not is_tensorflow_text_available():' )
             self.assertEqual(lowercase_			,   'tensorflow_text' )

             _lowerCamelCase       :		int		=find_backend('    if not (is_sentencepiece_available() and is_tokenizers_available()):' )
             self.assertEqual(lowercase_			,   'sentencepiece_and_tokenizers' )

             _lowerCamelCase       :		Dict		=find_backend(
                 '    if not (is_sentencepiece_available() and is_tensorflow_text_available()):' )
             self.assertEqual(lowercase_			,   'sentencepiece_and_tensorflow_text' )

             _lowerCamelCase       :		List[Any]		=find_backend(
                 '    if not (is_sentencepiece_available() and is_tokenizers_available() and is_vision_available()):' )
             self.assertEqual(lowercase_			,   'sentencepiece_and_tokenizers_and_vision' )






      def        lowerCamelCase       (					self						:     Optional[Any] )					->						List[str]:
             """simple docstring"""




             _lowerCamelCase       :		Union[str, Any]		=read_init()
             # We don't assert on the exact list of keys to allow for smooth grow of backend-specific objects
             self.assertIn('torch'			,   lowercase_ )
             self.assertIn('tensorflow_text'			,   lowercase_ )
             self.assertIn('sentencepiece_and_tokenizers'			,   lowercase_ )

             # Likewise, we can't assert on the exact content of a key
             self.assertIn('BertModel'			,   objects['torch'] )
             self.assertIn('TFBertModel'			,   objects['tf'] )
             self.assertIn('FlaxBertModel'			,   objects['flax'] )
             self.assertIn('BertModel'			,   objects['torch'] )
             self.assertIn('TFBertTokenizer'			,   objects['tensorflow_text'] )
             self.assertIn('convert_slow_tokenizer'			,   objects['sentencepiece_and_tokenizers'] )






      def        lowerCamelCase       (					self						:     int )					->						Optional[Any]:
             """simple docstring"""




             _lowerCamelCase       :		Optional[Any]		=create_dummy_object('CONSTANT'			,   '\'torch\'' )
             self.assertEqual(lowercase_			,   '\nCONSTANT = None\n' )

             _lowerCamelCase       :		Dict		=create_dummy_object('function'			,   '\'torch\'' )
             self.assertEqual(
                 lowercase_			,   '\ndef function(*args, **kwargs):\n    requires_backends(function, \'torch\')\n' )

             _lowerCamelCase       :		Union[str, Any]		='\nclass FakeClass(metaclass=DummyObject):\n    _backends = \'torch\'\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, \'torch\')\n'
             _lowerCamelCase       :		Tuple		=create_dummy_object('FakeClass'			,   '\'torch\'' )
             self.assertEqual(lowercase_			,   lowercase_ )






      def        lowerCamelCase       (					self						:     List[Any] )					->						Dict:
             """simple docstring"""




             _lowerCamelCase       :		Dict		='# This file is autogenerated by the command `make fix-copies`, do not edit.\nfrom ..utils import DummyObject, requires_backends\n\n\nCONSTANT = None\n\n\ndef function(*args, **kwargs):\n    requires_backends(function, ["torch"])\n\n\nclass FakeClass(metaclass=DummyObject):\n    _backends = ["torch"]\n\n    def __init__(self, *args, **kwargs):\n        requires_backends(self, ["torch"])\n'
             _lowerCamelCase       :		Optional[int]		=create_dummy_files({'torch': ['CONSTANT', 'function', 'FakeClass']} )
             self.assertEqual(dummy_files['torch']			,   lowercase_ )


