
'''simple docstring'''






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        __snake_case	(       _SCREAMING_SNAKE_CASE     ,unittest.TestCase):







    """simple docstring"""



    lowercase       			= RoCBertTokenizer
    lowercase       			= None
    lowercase       			= False
    lowercase       			= True
    lowercase       			= filter_non_english

    def __lowercase       (       self   :  List[Any]		)       ->  Union[str, Any]:
      super().setUp()

      lowerCAmelCase_      :				Dict      =       ["""[UNK]""", """[CLS]""", """[SEP]""", """[PAD]""", """[MASK]""", """你""", """好""", """是""", """谁""", """a""", """b""", """c""", """d"""]
      lowerCAmelCase_      :				Union[str, Any]      =       {}
      lowerCAmelCase_      :				Union[str, Any]      =       {}
      for i, value in enumerate(lowerCamelCase		):
        lowerCAmelCase_      :				int      =       i
        lowerCAmelCase_      :				Optional[Any]      =       i
      lowerCAmelCase_      :				Optional[int]      =       os.path.join(self.tmpdirname   ,						VOCAB_FILES_NAMES["""vocab_file"""]		)
      lowerCAmelCase_      :				Optional[Any]      =       os.path.join(self.tmpdirname   ,						VOCAB_FILES_NAMES["""word_shape_file"""]		)
      lowerCAmelCase_      :				List[Any]      =       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(lowerCamelCase   ,						lowerCamelCase   ,						ensure_ascii=lowerCamelCase		)
      with open(self.word_pronunciation_file   ,						"""w"""   ,						encoding="""utf-8"""		) as word_pronunciation_writer:
        json.dump(lowerCamelCase   ,						lowerCamelCase   ,						ensure_ascii=lowerCamelCase		)

    def __lowercase       (       self   :  List[str]		)       ->  Tuple:
      lowerCAmelCase_      :				List[str]      =       self.tokenizer_class(self.vocab_file   ,						self.word_shape_file   ,						self.word_pronunciation_file		)

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

    def __lowercase       (       self   :  Dict		)       ->  str:
      lowerCAmelCase_      :				List[Any]      =       RoCBertBasicTokenizer()

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

    def __lowercase       (       self   :  List[str]		)       ->  Tuple:
      lowerCAmelCase_      :				Optional[Any]      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase		)

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

    def __lowercase       (       self   :  Dict		)       ->  Optional[int]:
      lowerCAmelCase_      :				Tuple      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase   ,						strip_accents=lowerCamelCase		)

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

    def __lowercase       (       self   :  Optional[Any]		)       ->  Any:
      lowerCAmelCase_      :				List[Any]      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase   ,						strip_accents=lowerCamelCase		)

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

    def __lowercase       (       self   :  Union[str, Any]		)       ->  List[str]:
      lowerCAmelCase_      :				Optional[int]      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase		)

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

    def __lowercase       (       self   :  int		)       ->  Optional[Any]:
      lowerCAmelCase_      :				Union[str, Any]      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase		)

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

    def __lowercase       (       self   :  List[Any]		)       ->  Union[str, Any]:
      lowerCAmelCase_      :				str      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase   ,						strip_accents=lowerCamelCase		)

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

    def __lowercase       (       self   :  Union[str, Any]		)       ->  str:
      lowerCAmelCase_      :				Optional[Any]      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase   ,						strip_accents=lowerCamelCase		)

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

    def __lowercase       (       self   :  Tuple		)       ->  Any:
      lowerCAmelCase_      :				Dict      =       RoCBertBasicTokenizer(do_lower_case=lowerCamelCase   ,						never_split=["""[UNK]"""]		)

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

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

      lowerCAmelCase_      :				Dict      =       {}
      for i, token in enumerate(lowerCamelCase		):
        lowerCAmelCase_      :				Any      =       i
      lowerCAmelCase_      :				Optional[Any]      =       RoCBertWordpieceTokenizer(vocab=lowerCamelCase   ,						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 __lowercase       (       self   :  Optional[int]		)       ->  List[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 __lowercase       (       self   :  Tuple		)       ->  Dict:
      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 __lowercase       (       self   :  str		)       ->  Tuple:
      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 __lowercase       (       self   :  str		)       ->  Optional[Any]:
      lowerCAmelCase_      :				int      =       self.get_tokenizer()

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

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

    def __lowercase       (       self   :  Optional[Any]		)       ->  Dict:
      for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
        with self.subTest(F'{tokenizer.__class__.__name__} ({pretrained_name})'		):
          lowerCAmelCase_      :				Optional[int]      =       self.rust_tokenizer_class.from_pretrained(lowerCamelCase   ,						**lowerCamelCase		)

          lowerCAmelCase_      :				Any      =       F'A, naïve {tokenizer_r.mask_token} AllenNLP sentence.'
          lowerCAmelCase_      :				int      =       tokenizer_r.encode_plus(
              lowerCamelCase   ,						return_attention_mask=lowerCamelCase   ,						return_token_type_ids=lowerCamelCase   ,						return_offsets_mapping=lowerCamelCase   ,						add_special_tokens=lowerCamelCase   ,						)

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

          lowerCAmelCase_      :				Union[str, Any]      =       tokenizer_p.encode(lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)
          lowerCAmelCase_      :				int      =       tokenizer_r.encode(lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)

          lowerCAmelCase_      :				Any      =       tokenizer_r.convert_ids_to_tokens(lowerCamelCase		)
          lowerCAmelCase_      :				Union[str, Any]      =       tokenizer_p.convert_ids_to_tokens(lowerCamelCase		)

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

          lowerCAmelCase_      :				Dict      =       False
          lowerCAmelCase_      :				Union[str, Any]      =       self.rust_tokenizer_class.from_pretrained(lowerCamelCase   ,						**lowerCamelCase		)
          lowerCAmelCase_      :				Any      =       self.tokenizer_class.from_pretrained(lowerCamelCase   ,						**lowerCamelCase		)

          lowerCAmelCase_      :				Tuple      =       tokenizer_r.encode(lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)
          lowerCAmelCase_      :				List[Any]      =       tokenizer_p.encode(lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)

          lowerCAmelCase_      :				Any      =       tokenizer_r.convert_ids_to_tokens(lowerCamelCase		)
          lowerCAmelCase_      :				Union[str, Any]      =       tokenizer_p.convert_ids_to_tokens(lowerCamelCase		)

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

    @slow
    def __lowercase       (       self   :  Union[str, Any]		)       ->  Tuple:
      lowerCAmelCase_      :				Optional[Any]      =       self.tokenizer_class(self.vocab_file   ,						self.word_shape_file   ,						self.word_pronunciation_file		)

      lowerCAmelCase_      :				int      =       tokenizer.encode("""你好"""   ,						add_special_tokens=lowerCamelCase		)
      lowerCAmelCase_      :				Optional[int]      =       tokenizer.encode("""你是谁"""   ,						add_special_tokens=lowerCamelCase		)

      lowerCAmelCase_      :				List[Any]      =       tokenizer.build_inputs_with_special_tokens(lowerCamelCase		)
      lowerCAmelCase_      :				Dict      =       tokenizer.build_inputs_with_special_tokens(lowerCamelCase   ,						lowerCamelCase		)

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





    def __lowercase       (       self   :  Any		)       ->  List[str]:
      lowerCAmelCase_      :				List[Any]      =       self.get_tokenizers(do_lower_case=lowerCamelCase		)
      for tokenizer in tokenizers:
        with self.subTest(F'{tokenizer.__class__.__name__}'		):
          lowerCAmelCase_      :				List[Any]      =       """你好，你是谁"""
          lowerCAmelCase_      :				Any      =       tokenizer.tokenize(lowerCamelCase		)
          lowerCAmelCase_      :				List[str]      =       tokenizer.convert_tokens_to_ids(lowerCamelCase		)
          lowerCAmelCase_      :				Optional[Any]      =       tokenizer.convert_tokens_to_shape_ids(lowerCamelCase		)
          lowerCAmelCase_      :				int      =       tokenizer.convert_tokens_to_pronunciation_ids(lowerCamelCase		)
          lowerCAmelCase_      :				List[str]      =       tokenizer.prepare_for_model(
              lowerCamelCase   ,						lowerCamelCase   ,						lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)

          lowerCAmelCase_      :				Dict      =       tokenizer.encode_plus(lowerCamelCase   ,						add_special_tokens=lowerCamelCase		)

          self.assertEqual(lowerCamelCase   ,						lowerCamelCase		)







'''simple docstring'''






from typing import TYPE_CHECKING

from ...utils import OptionalDependencyNotAvailable, _LazyModule, is_tokenizers_available, is_torch_available


__A :  List[Any]    							=				{
    "configuration_roc_bert": ["ROC_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "RoCBertConfig"],
    "tokenization_roc_bert": ["RoCBertTokenizer"],
}

try:
  if not is_tokenizers_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  pass

try:
  if not is_torch_available():
    raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
  pass
else:
  __A :  Union[str, Any]    							=				[
      "ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST",
      "RoCBertForCausalLM",
      "RoCBertForMaskedLM",
      "RoCBertForMultipleChoice",
      "RoCBertForPreTraining",
      "RoCBertForQuestionAnswering",
      "RoCBertForSequenceClassification",
      "RoCBertForTokenClassification",
      "RoCBertLayer",
      "RoCBertModel",
      "RoCBertPreTrainedModel",
      "load_tf_weights_in_roc_bert",
  ]

if TYPE_CHECKING:
  from .configuration_roc_bert import ROC_BERT_PRETRAINED_CONFIG_ARCHIVE_MAP, RoCBertConfig
  from .tokenization_roc_bert import RoCBertTokenizer

  try:
    if not is_tokenizers_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    raise OptionalDependencyNotAvailable()

  try:
    if not is_torch_available():
      raise OptionalDependencyNotAvailable()
  except OptionalDependencyNotAvailable:
    pass
  else:
    from .modeling_roc_bert import (
        ROC_BERT_PRETRAINED_MODEL_ARCHIVE_LIST,
        RoCBertForCausalLM,
        RoCBertForMaskedLM,
        RoCBertForMultipleChoice,
        RoCBertForPreTraining,
        RoCBertForQuestionAnswering,
        RoCBertForSequenceClassification,
        RoCBertForTokenClassification,
        RoCBertLayer,
        RoCBertModel,
        RoCBertPreTrainedModel,
        load_tf_weights_in_roc_bert,
    )


else:
  import sys

  __A :  int    							=				_LazyModule(__name__, globals()["__file__"], _import_structure, module_spec=__spec__)



