


'''simple docstring'''



import os
import unittest

from transformers import MobileBertTokenizer, MobileBertTokenizerFast
from transformers.models.bert.tokenization_bert import (
    VOCAB_FILES_NAMES,
    BasicTokenizer,
    WordpieceTokenizer,
    _is_control,
    _is_punctuation,
    _is_whitespace,
)
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin, filter_non_english


@require_tokenizers
class 						_snake_case    (   __lowerCamelCase				,      unittest.TestCase		):
					lowerCAmelCase					:int            =    MobileBertTokenizer
					lowerCAmelCase					:Any            =    MobileBertTokenizerFast
					lowerCAmelCase					:int            =    True
					lowerCAmelCase					:Any            =    True
					lowerCAmelCase					:List[str]            =    filter_non_english
					lowerCAmelCase					:Any            =    'google/mobilebert-uncased'




					def  snake_case__  (					self):
								super().setUp()

								UpperCAmelCase__     :    str         =   [
								    """[UNK]""",
								    """[CLS]""",
								    """[SEP]""",
								    """[PAD]""",
								    """[MASK]""",
								    """want""",
								    """##want""",
								    """##ed""",
								    """wa""",
								    """un""",
								    """runn""",
								    """##ing""",
								    """,""",
								    """low""",
								    """lowest""",
								]
								UpperCAmelCase__     :    List[Any]         =   os.path.join(self.tmpdirname							,							VOCAB_FILES_NAMES["""vocab_file"""])
								with open(self.vocab_file							,							"""w"""							,							encoding="""utf-8""") as vocab_writer:
											vocab_writer.write("""""".join([x + """\n""" for x in vocab_tokens]))

								UpperCAmelCase__     :    Union[str, Any]         =   [
								    (tokenizer_def[0], self.pre_trained_model_path, tokenizer_def[2])  # else the 'google/' prefix is stripped
								    for tokenizer_def in self.tokenizers_list
								]




					def  snake_case__  (					self							,							_lowerCamelCase):
								UpperCAmelCase__     :    Optional[Any]         =   """UNwant\u00E9d,running"""
								UpperCAmelCase__     :    Dict         =   """unwanted, running"""
								return input_text, output_text




					def  snake_case__  (					self):
								UpperCAmelCase__     :    str         =   self.tokenizer_class(self.vocab_file)

								UpperCAmelCase__     :    Optional[int]         =   tokenizer.tokenize("""UNwant\u00E9d,running""")
								self.assertListEqual(UpperCamelCase_							,							["""un""", """##want""", """##ed""", """,""", """runn""", """##ing"""])
								self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCamelCase_)							,							[9, 6, 7, 12, 10, 11])




					def  snake_case__  (					self):
								if not self.test_rust_tokenizer:
											return

								UpperCAmelCase__     :    List[Any]         =   self.get_tokenizer()
								UpperCAmelCase__     :    int         =   self.get_rust_tokenizer()

								UpperCAmelCase__     :    Optional[Any]         =   """UNwant\u00E9d,running"""

								UpperCAmelCase__     :    Union[str, Any]         =   tokenizer.tokenize(UpperCamelCase_)
								UpperCAmelCase__     :    Dict         =   rust_tokenizer.tokenize(UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

								UpperCAmelCase__     :    Dict         =   tokenizer.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
								UpperCAmelCase__     :    Dict         =   rust_tokenizer.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

								UpperCAmelCase__     :    Any         =   self.get_rust_tokenizer()
								UpperCAmelCase__     :    Optional[Any]         =   tokenizer.encode(UpperCamelCase_)
								UpperCAmelCase__     :    int         =   rust_tokenizer.encode(UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

								# With lower casing
								UpperCAmelCase__     :    List[str]         =   self.get_tokenizer(do_lower_case=UpperCamelCase_)
								UpperCAmelCase__     :    str         =   self.get_rust_tokenizer(do_lower_case=UpperCamelCase_)

								UpperCAmelCase__     :    int         =   """UNwant\u00E9d,running"""

								UpperCAmelCase__     :    Union[str, Any]         =   tokenizer.tokenize(UpperCamelCase_)
								UpperCAmelCase__     :    Optional[Any]         =   rust_tokenizer.tokenize(UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

								UpperCAmelCase__     :    Optional[Any]         =   tokenizer.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
								UpperCAmelCase__     :    str         =   rust_tokenizer.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

								UpperCAmelCase__     :    Optional[Any]         =   self.get_rust_tokenizer()
								UpperCAmelCase__     :    Any         =   tokenizer.encode(UpperCamelCase_)
								UpperCAmelCase__     :    str         =   rust_tokenizer.encode(UpperCamelCase_)
								self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Dict         =   BasicTokenizer()

								self.assertListEqual(tokenizer.tokenize("""ah\u535A\u63A8zz""")							,							["""ah""", """\u535A""", """\u63A8""", """zz"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Optional[int]         =   BasicTokenizer(do_lower_case=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """)							,							["""hello""", """!""", """how""", """are""", """you""", """?"""])
								self.assertListEqual(tokenizer.tokenize("""H\u00E9llo""")							,							["""hello"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Tuple         =   BasicTokenizer(do_lower_case=UpperCamelCase_							,							strip_accents=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """)							,							["""hällo""", """!""", """how""", """are""", """you""", """?"""])
								self.assertListEqual(tokenizer.tokenize("""H\u00E9llo""")							,							["""h\u00E9llo"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    List[str]         =   BasicTokenizer(do_lower_case=UpperCamelCase_							,							strip_accents=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """)							,							["""hallo""", """!""", """how""", """are""", """you""", """?"""])
								self.assertListEqual(tokenizer.tokenize("""H\u00E9llo""")							,							["""hello"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Tuple         =   BasicTokenizer(do_lower_case=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """)							,							["""hallo""", """!""", """how""", """are""", """you""", """?"""])
								self.assertListEqual(tokenizer.tokenize("""H\u00E9llo""")							,							["""hello"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    List[str]         =   BasicTokenizer(do_lower_case=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """)							,							["""HeLLo""", """!""", """how""", """Are""", """yoU""", """?"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Union[str, Any]         =   BasicTokenizer(do_lower_case=UpperCamelCase_							,							strip_accents=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """)							,							["""HäLLo""", """!""", """how""", """Are""", """yoU""", """?"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Optional[Any]         =   BasicTokenizer(do_lower_case=UpperCamelCase_							,							strip_accents=UpperCamelCase_)

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """)							,							["""HaLLo""", """!""", """how""", """Are""", """yoU""", """?"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    List[str]         =   BasicTokenizer(do_lower_case=UpperCamelCase_							,							never_split=["""[UNK]"""])

								self.assertListEqual(
								    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU? [UNK]""")							,							["""HeLLo""", """!""", """how""", """Are""", """yoU""", """?""", """[UNK]"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Union[str, Any]         =   ["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

								UpperCAmelCase__     :    Union[str, Any]         =   {}
								for i, token in enumerate(UpperCamelCase_):
											UpperCAmelCase__     :    List[Any]         =   i
								UpperCAmelCase__     :    Union[str, Any]         =   WordpieceTokenizer(vocab=UpperCamelCase_							,							unk_token="""[UNK]""")

								self.assertListEqual(tokenizer.tokenize("""""")							,							[])

								self.assertListEqual(tokenizer.tokenize("""unwanted running""")							,							["""un""", """##want""", """##ed""", """runn""", """##ing"""])

								self.assertListEqual(tokenizer.tokenize("""unwantedX running""")							,							["""[UNK]""", """runn""", """##ing"""])




					def  snake_case__  (					self):
								self.assertTrue(_is_whitespace(""" """))
								self.assertTrue(_is_whitespace("""\t"""))
								self.assertTrue(_is_whitespace("""\r"""))
								self.assertTrue(_is_whitespace("""\n"""))
								self.assertTrue(_is_whitespace("""\u00A0"""))

								self.assertFalse(_is_whitespace("""A"""))
								self.assertFalse(_is_whitespace("""-"""))




					def  snake_case__  (					self):
								self.assertTrue(_is_control("""\u0005"""))

								self.assertFalse(_is_control("""A"""))
								self.assertFalse(_is_control(""" """))
								self.assertFalse(_is_control("""\t"""))
								self.assertFalse(_is_control("""\r"""))




					def  snake_case__  (					self):
								self.assertTrue(_is_punctuation("""-"""))
								self.assertTrue(_is_punctuation("""$"""))
								self.assertTrue(_is_punctuation("""`"""))
								self.assertTrue(_is_punctuation("""."""))

								self.assertFalse(_is_punctuation("""A"""))
								self.assertFalse(_is_punctuation(""" """))




					def  snake_case__  (					self):
								UpperCAmelCase__     :    Tuple         =   self.get_tokenizer()
								UpperCAmelCase__     :    List[str]         =   self.get_rust_tokenizer()

								# Example taken from the issue https://github.com/huggingface/tokenizers/issues/340
								self.assertListEqual([tokenizer.tokenize(UpperCamelCase_) for t in ["""Test""", """\xad""", """test"""]]							,							[["""[UNK]"""], [], ["""[UNK]"""]])

								self.assertListEqual(
								    [rust_tokenizer.tokenize(UpperCamelCase_) for t in ["""Test""", """\xad""", """test"""]]							,							[["""[UNK]"""], [], ["""[UNK]"""]])




					@slow
					def  snake_case__  (					self):
								UpperCAmelCase__     :    List[str]         =   self.tokenizer_class.from_pretrained("""google/mobilebert-uncased""")

								UpperCAmelCase__     :    str         =   tokenizer.encode("""sequence builders"""							,							add_special_tokens=UpperCamelCase_)
								UpperCAmelCase__     :    Any         =   tokenizer.encode("""multi-sequence build"""							,							add_special_tokens=UpperCamelCase_)

								UpperCAmelCase__     :    Optional[int]         =   tokenizer.build_inputs_with_special_tokens(UpperCamelCase_)
								UpperCAmelCase__     :    int         =   tokenizer.build_inputs_with_special_tokens(UpperCamelCase_							,							UpperCamelCase_)

								assert encoded_sentence == [101] + text + [102]
								assert encoded_pair == [101] + text + [102] + text_a + [102]




					def  snake_case__  (					self):
								for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
											with self.subTest(f'''{tokenizer.__class__.__name__} ({pretrained_name})'''):
														UpperCAmelCase__     :    Dict         =   self.rust_tokenizer_class.from_pretrained(UpperCamelCase_							,							**UpperCamelCase_)

														UpperCAmelCase__     :    Any         =   f'''A, naïve {tokenizer_r.mask_token} AllenNLP sentence.'''
														UpperCAmelCase__     :    Optional[Any]         =   tokenizer_r.encode_plus(
														    UpperCamelCase_							,							return_attention_mask=UpperCamelCase_							,							return_token_type_ids=UpperCamelCase_							,							return_offsets_mapping=UpperCamelCase_							,							add_special_tokens=UpperCamelCase_							,							)

														UpperCAmelCase__     :    List[Any]         =   tokenizer_r.do_lower_case if hasattr(UpperCamelCase_							,							"""do_lower_case""") else False
														UpperCAmelCase__     :    int         =   (
														    [
														        ((0, 0), tokenizer_r.cls_token),
														        ((0, 1), """A"""),
														        ((1, 2), ""","""),
														        ((3, 5), """na"""),
														        ((5, 6), """##ï"""),
														        ((6, 8), """##ve"""),
														        ((9, 15), tokenizer_r.mask_token),
														        ((16, 21), """Allen"""),
														        ((21, 23), """##NL"""),
														        ((23, 24), """##P"""),
														        ((25, 33), """sentence"""),
														        ((33, 34), """."""),
														        ((0, 0), tokenizer_r.sep_token),
														    ]
														    if not do_lower_case
														    else [
														        ((0, 0), tokenizer_r.cls_token),
														        ((0, 1), """a"""),
														        ((1, 2), ""","""),
														        ((3, 8), """naive"""),
														        ((9, 15), tokenizer_r.mask_token),
														        ((16, 21), """allen"""),
														        ((21, 23), """##nl"""),
														        ((23, 24), """##p"""),
														        ((25, 33), """sentence"""),
														        ((33, 34), """."""),
														        ((0, 0), tokenizer_r.sep_token),
														    ]
														)

														self.assertEqual(
														    [e[1] for e in expected_results]							,							tokenizer_r.convert_ids_to_tokens(tokens["""input_ids"""]))
														self.assertEqual([e[0] for e in expected_results]							,							tokens["""offset_mapping"""])




					def  snake_case__  (					self):
								UpperCAmelCase__     :    int         =   ["""的""", """人""", """有"""]
								UpperCAmelCase__     :    List[Any]         =   """""".join(UpperCamelCase_)
								for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
											with self.subTest(f'''{tokenizer.__class__.__name__} ({pretrained_name})'''):
														UpperCAmelCase__     :    int         =   True
														UpperCAmelCase__     :    Union[str, Any]         =   self.tokenizer_class.from_pretrained(UpperCamelCase_							,							**UpperCamelCase_)
														UpperCAmelCase__     :    int         =   self.rust_tokenizer_class.from_pretrained(UpperCamelCase_							,							**UpperCamelCase_)

														UpperCAmelCase__     :    Optional[int]         =   tokenizer_p.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
														UpperCAmelCase__     :    str         =   tokenizer_r.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)

														UpperCAmelCase__     :    int         =   tokenizer_r.convert_ids_to_tokens(UpperCamelCase_)
														UpperCAmelCase__     :    Union[str, Any]         =   tokenizer_p.convert_ids_to_tokens(UpperCamelCase_)

														# it is expected that each Chinese character is not preceded by "##"
														self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)
														self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)

														UpperCAmelCase__     :    Tuple         =   False
														UpperCAmelCase__     :    Union[str, Any]         =   self.rust_tokenizer_class.from_pretrained(UpperCamelCase_							,							**UpperCamelCase_)
														UpperCAmelCase__     :    List[Any]         =   self.tokenizer_class.from_pretrained(UpperCamelCase_							,							**UpperCamelCase_)

														UpperCAmelCase__     :    Dict         =   tokenizer_r.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)
														UpperCAmelCase__     :    int         =   tokenizer_p.encode(UpperCamelCase_							,							add_special_tokens=UpperCamelCase_)

														UpperCAmelCase__     :    int         =   tokenizer_r.convert_ids_to_tokens(UpperCamelCase_)
														UpperCAmelCase__     :    List[Any]         =   tokenizer_p.convert_ids_to_tokens(UpperCamelCase_)

														# it is expected that only the first Chinese character is not preceded by "##".
														UpperCAmelCase__     :    Dict         =   [
														    f'''##{token}''' if idx != 0 else token for idx, token in enumerate(UpperCamelCase_)
														]
														self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)
														self.assertListEqual(UpperCamelCase_							,							UpperCamelCase_)





from tempfile import TemporaryDirectory
from unittest import TestCase
from unittest.mock import MagicMock, patch

from transformers import AutoModel, TFAutoModel
from transformers.onnx import FeaturesManager
from transformers.testing_utils import SMALL_MODEL_IDENTIFIER, require_tf, require_torch





@require_torch
@require_tf
class     lowerCamelCase__(					__lowerCamelCase):
      def 			lowerCAmelCase__						(				self: List[Any]   ):
             __lowerCamelCase   =						SMALL_MODEL_IDENTIFIER
             __lowerCamelCase   =						"""pt"""
             __lowerCamelCase   =						"""tf"""
      def 			lowerCAmelCase__						(				self: Optional[Any]    ,       UpperCamelCase_: Optional[Any]   ):
             __lowerCamelCase   =						AutoModel.from_pretrained(self.test_model   )
             model_pt.save_pretrained(UpperCamelCase_   )
      def 			lowerCAmelCase__						(				self: Tuple    ,       UpperCamelCase_: Union[str, Any]   ):
             __lowerCamelCase   =						TFAutoModel.from_pretrained(self.test_model    ,       from_pt=UpperCamelCase_   )
             model_tf.save_pretrained(UpperCamelCase_   )
      def 			lowerCAmelCase__						(				self: Union[str, Any]   ):
             __lowerCamelCase   =						"""mock_framework"""

             # Framework provided - return whatever the user provides
             __lowerCamelCase   =						FeaturesManager.determine_framework(self.test_model    ,       UpperCamelCase_   )
             self.assertEqual(UpperCamelCase_    ,       UpperCamelCase_   )

             # Local checkpoint and framework provided - return provided framework
             # PyTorch checkpoint
             with TemporaryDirectory() as local_pt_ckpt:
                    self._setup_pt_ckpt(UpperCamelCase_   )
                    __lowerCamelCase   =						FeaturesManager.determine_framework(UpperCamelCase_    ,       UpperCamelCase_   )
                    self.assertEqual(UpperCamelCase_    ,       UpperCamelCase_   )

             # TensorFlow checkpoint
             with TemporaryDirectory() as local_tf_ckpt:
                    self._setup_tf_ckpt(UpperCamelCase_   )
                    __lowerCamelCase   =						FeaturesManager.determine_framework(UpperCamelCase_    ,       UpperCamelCase_   )
                    self.assertEqual(UpperCamelCase_    ,       UpperCamelCase_   )
      def 			lowerCAmelCase__						(				self: int   ):
             # PyTorch checkpoint
             with TemporaryDirectory() as local_pt_ckpt:
                    self._setup_pt_ckpt(UpperCamelCase_   )
                    __lowerCamelCase   =						FeaturesManager.determine_framework(UpperCamelCase_   )
                    self.assertEqual(UpperCamelCase_    ,       self.framework_pt   )

             # TensorFlow checkpoint
             with TemporaryDirectory() as local_tf_ckpt:
                    self._setup_tf_ckpt(UpperCamelCase_   )
                    __lowerCamelCase   =						FeaturesManager.determine_framework(UpperCamelCase_   )
                    self.assertEqual(UpperCamelCase_    ,       self.framework_tf   )

             # Invalid local checkpoint
             with TemporaryDirectory() as local_invalid_ckpt:
                    with self.assertRaises(UpperCamelCase_   ):
                           __lowerCamelCase   =						FeaturesManager.determine_framework(UpperCamelCase_   )




      def 			lowerCAmelCase__						(				self: Optional[int]   ):
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             with patch("""transformers.onnx.features.is_tf_available"""    ,       UpperCamelCase_   ):
                    __lowerCamelCase   =						FeaturesManager.determine_framework(self.test_model   )
                    self.assertEqual(UpperCamelCase_    ,       self.framework_pt   )

             # PyTorch not in environment -> use TensorFlow
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             with patch("""transformers.onnx.features.is_torch_available"""    ,       UpperCamelCase_   ):
                    __lowerCamelCase   =						FeaturesManager.determine_framework(self.test_model   )
                    self.assertEqual(UpperCamelCase_    ,       self.framework_tf   )

             # Both in environment -> use PyTorch
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             with patch("""transformers.onnx.features.is_tf_available"""    ,       UpperCamelCase_   ), patch(
                 """transformers.onnx.features.is_torch_available"""    ,       UpperCamelCase_   ):
                    __lowerCamelCase   =						FeaturesManager.determine_framework(self.test_model   )
                    self.assertEqual(UpperCamelCase_    ,       self.framework_pt   )

             # Both not in environment -> raise error
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             __lowerCamelCase   =						MagicMock(return_value=UpperCamelCase_   )
             with patch("""transformers.onnx.features.is_tf_available"""    ,       UpperCamelCase_   ), patch(
                 """transformers.onnx.features.is_torch_available"""    ,       UpperCamelCase_   ):
                    with self.assertRaises(UpperCamelCase_   ):
                           __lowerCamelCase   =						FeaturesManager.determine_framework(self.test_model   )

