

import json
import os
import unittest

from transformers.models.roc_bert.tokenization_roc_bert import (
    VOCAB_FILES_NAMES,
    RoCBertBasicTokenizer,
    RoCBertTokenizer,
    RoCBertWordpieceTokenizer,
    _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 						__magic_name__					(							snake_case			,      unittest.TestCase				):
     UpperCamelCase_	:Optional[Any]										=      RoCBertTokenizer
     UpperCamelCase_	:List[Any]										=      None
     UpperCamelCase_	:Tuple										=      False
     UpperCamelCase_	:str										=      True
     UpperCamelCase_	:Optional[Any]										=      filter_non_english






     def 	UpperCAmelCase_				(				self	)->     Tuple:
         super().setUp()

         UpperCamelCase_			  =						["[UNK]", "[CLS]", "[SEP]", "[PAD]", "[MASK]", "你", "好", "是", "谁", "a", "b", "c", "d"]
         UpperCamelCase_			  =						{}
         UpperCamelCase_			  =						{}
         for i, value in enumerate(_lowercase	):
             UpperCamelCase_			  =						i
             UpperCamelCase_			  =						i
         UpperCamelCase_			  =						os.path.join(self.tmpdirname		,						VOCAB_FILES_NAMES["vocab_file"]	)
         UpperCamelCase_			  =						os.path.join(self.tmpdirname		,						VOCAB_FILES_NAMES["word_shape_file"]	)
         UpperCamelCase_			  =						os.path.join(self.tmpdirname		,						VOCAB_FILES_NAMES["word_pronunciation_file"]	)
         with open(self.vocab_file		,						"w"		,						encoding="utf-8"	) as vocab_writer:
             vocab_writer.write("".join([x + "\n" for x in vocab_tokens]	)	)
         with open(self.word_shape_file		,						"w"		,						encoding="utf-8"	) as word_shape_writer:
             json.dump(_lowercase		,						_lowercase		,						ensure_ascii=_lowercase	)
         with open(self.word_pronunciation_file		,						"w"		,						encoding="utf-8"	) as word_pronunciation_writer:
             json.dump(_lowercase		,						_lowercase		,						ensure_ascii=_lowercase	)






     def 	UpperCAmelCase_				(				self	)->     int:
         UpperCamelCase_			  =						self.tokenizer_class(self.vocab_file		,						self.word_shape_file		,						self.word_pronunciation_file	)

         UpperCamelCase_			  =						tokenizer.tokenize("你好[SEP]你是谁"	)
         self.assertListEqual(_lowercase		,						["你", "好", "[SEP]", "你", "是", "谁"]	)
         self.assertListEqual(tokenizer.convert_tokens_to_ids(_lowercase	)		,						[5, 6, 2, 5, 7, 8]	)
         self.assertListEqual(tokenizer.convert_tokens_to_shape_ids(_lowercase	)		,						[5, 6, 2, 5, 7, 8]	)
         self.assertListEqual(tokenizer.convert_tokens_to_pronunciation_ids(_lowercase	)		,						[5, 6, 2, 5, 7, 8]	)






     def 	UpperCAmelCase_				(				self	)->     Optional[int]:
         UpperCamelCase_			  =						RoCBertBasicTokenizer()

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






     def 	UpperCAmelCase_				(				self	)->     Optional[Any]:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     List[str]:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase		,						strip_accents=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     List[Any]:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase		,						strip_accents=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     Union[str, Any]:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     str:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     str:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase		,						strip_accents=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     Tuple:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase		,						strip_accents=_lowercase	)

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






     def 	UpperCAmelCase_				(				self	)->     Tuple:
         UpperCamelCase_			  =						RoCBertBasicTokenizer(do_lower_case=_lowercase		,						never_split=["[UNK]"]	)

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






     def 	UpperCAmelCase_				(				self	)->     List[str]:
         UpperCamelCase_			  =						["[UNK]", "[CLS]", "[SEP]", "want", "##want", "##ed", "wa", "un", "runn", "##ing"]

         UpperCamelCase_			  =						{}
         for i, token in enumerate(_lowercase	):
             UpperCamelCase_			  =						i
         UpperCamelCase_			  =						RoCBertWordpieceTokenizer(vocab=_lowercase		,						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 	UpperCAmelCase_				(				self	)->     Any:
         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 	UpperCAmelCase_				(				self	)->     Tuple:
         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 	UpperCAmelCase_				(				self	)->     List[Any]:
         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 	UpperCAmelCase_				(				self	)->     Dict:
         UpperCamelCase_			  =						self.get_tokenizer()

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

         if self.test_rust_tokenizer:
             UpperCamelCase_			  =						self.get_rust_tokenizer()
             self.assertListEqual(
                 [rust_tokenizer.tokenize(_lowercase	) for t in ["Test", "\xad", "test"]]		,						[["[UNK]"], [], ["[UNK]"]]	)






     def 	UpperCAmelCase_				(				self	)->     Union[str, Any]:
         for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
             with self.subTest(F"{tokenizer.__class__.__name__} ({pretrained_name})"	):
                 UpperCamelCase_			  =						self.rust_tokenizer_class.from_pretrained(_lowercase		,						**_lowercase	)

                 UpperCamelCase_			  =						F"A, naïve {tokenizer_r.mask_token} AllenNLP sentence."
                 UpperCamelCase_			  =						tokenizer_r.encode_plus(
                     _lowercase		,						return_attention_mask=_lowercase		,						return_token_type_ids=_lowercase		,						return_offsets_mapping=_lowercase		,						add_special_tokens=_lowercase		,						)

                 UpperCamelCase_			  =						tokenizer_r.do_lower_case if hasattr(_lowercase		,						"do_lower_case"	) else False
                 UpperCamelCase_			  =						(
                     [
                         ((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 	UpperCAmelCase_				(				self	)->     List[str]:
         UpperCamelCase_			  =						["的", "人", "有"]
         UpperCamelCase_			  =						"".join(_lowercase	)
         for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
             with self.subTest(F"{tokenizer.__class__.__name__} ({pretrained_name})"	):
                 UpperCamelCase_			  =						True
                 UpperCamelCase_			  =						self.tokenizer_class.from_pretrained(_lowercase		,						**_lowercase	)
                 UpperCamelCase_			  =						self.rust_tokenizer_class.from_pretrained(_lowercase		,						**_lowercase	)

                 UpperCamelCase_			  =						tokenizer_p.encode(_lowercase		,						add_special_tokens=_lowercase	)
                 UpperCamelCase_			  =						tokenizer_r.encode(_lowercase		,						add_special_tokens=_lowercase	)

                 UpperCamelCase_			  =						tokenizer_r.convert_ids_to_tokens(_lowercase	)
                 UpperCamelCase_			  =						tokenizer_p.convert_ids_to_tokens(_lowercase	)

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

                 UpperCamelCase_			  =						False
                 UpperCamelCase_			  =						self.rust_tokenizer_class.from_pretrained(_lowercase		,						**_lowercase	)
                 UpperCamelCase_			  =						self.tokenizer_class.from_pretrained(_lowercase		,						**_lowercase	)

                 UpperCamelCase_			  =						tokenizer_r.encode(_lowercase		,						add_special_tokens=_lowercase	)
                 UpperCamelCase_			  =						tokenizer_p.encode(_lowercase		,						add_special_tokens=_lowercase	)

                 UpperCamelCase_			  =						tokenizer_r.convert_ids_to_tokens(_lowercase	)
                 UpperCamelCase_			  =						tokenizer_p.convert_ids_to_tokens(_lowercase	)

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






     @slow
     def 	UpperCAmelCase_				(				self	)->     List[str]:
         UpperCamelCase_			  =						self.tokenizer_class(self.vocab_file		,						self.word_shape_file		,						self.word_pronunciation_file	)

         UpperCamelCase_			  =						tokenizer.encode("你好"		,						add_special_tokens=_lowercase	)
         UpperCamelCase_			  =						tokenizer.encode("你是谁"		,						add_special_tokens=_lowercase	)

         UpperCamelCase_			  =						tokenizer.build_inputs_with_special_tokens(_lowercase	)
         UpperCamelCase_			  =						tokenizer.build_inputs_with_special_tokens(_lowercase		,						_lowercase	)

         assert encoded_sentence == [1] + text + [2]
         assert encoded_pair == [1] + text + [2] + text_a + [2]






     def 	UpperCAmelCase_				(				self	)->     Dict:
         UpperCamelCase_			  =						self.get_tokenizers(do_lower_case=_lowercase	)
         for tokenizer in tokenizers:
             with self.subTest(F"{tokenizer.__class__.__name__}"	):
                 UpperCamelCase_			  =						"你好，你是谁"
                 UpperCamelCase_			  =						tokenizer.tokenize(_lowercase	)
                 UpperCamelCase_			  =						tokenizer.convert_tokens_to_ids(_lowercase	)
                 UpperCamelCase_			  =						tokenizer.convert_tokens_to_shape_ids(_lowercase	)
                 UpperCamelCase_			  =						tokenizer.convert_tokens_to_pronunciation_ids(_lowercase	)
                 UpperCamelCase_			  =						tokenizer.prepare_for_model(
                     _lowercase		,						_lowercase		,						_lowercase		,						add_special_tokens=_lowercase	)

                 UpperCamelCase_			  =						tokenizer.encode_plus(_lowercase		,						add_special_tokens=_lowercase	)

                 self.assertEqual(_lowercase		,						_lowercase	)



from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tokenizers_available


SCREAMING_SNAKE_CASE		:List[Any]						=     {"""tokenization_herbert""": ["""HerbertTokenizer"""]}

try:
 if not is_tokenizers_available():
  raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
 pass
else:
 SCREAMING_SNAKE_CASE		:Union[str, Any]						=     ["""HerbertTokenizerFast"""]


if TYPE_CHECKING:
 from .tokenization_herbert import HerbertTokenizer

 try:
  if not is_tokenizers_available():
   raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
  pass
 else:
  from .tokenization_herbert_fast import HerbertTokenizerFast

else:
 import sys

 SCREAMING_SNAKE_CASE		:List[str]						=     _LazyModule(__name__, globals()["""__file__"""], _import_structure, module_spec=__spec__)
