






import os
import unittest

from transformers import ElectraTokenizerFast
from transformers.models.electra.tokenization_electra import (
    VOCAB_FILES_NAMES,
    BasicTokenizer,
    ElectraTokenizer,
    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 					a					(				lowercase__							, unittest.TestCase     ):



      """simple docstring"""


      a						:				Optional[Any]							 =			ElectraTokenizer
      a						:				Any							 =			ElectraTokenizerFast
      a						:				Optional[Any]							 =			True
      a						:				Tuple							 =			True
      a						:				Tuple							 =			filter_non_english
      def 			UpperCAmelCase   (   self      :    List[Any]      )							->   str:
        super().setUp()

        __UpperCAmelCase     =						[
            "[UNK]",
            "[CLS]",
            "[SEP]",
            "[PAD]",
            "[MASK]",
            "want",
            "##want",
            "##ed",
            "wa",
            "un",
            "runn",
            "##ing",
            ",",
            "low",
            "lowest",
        ]
        __UpperCAmelCase     =						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]      )      )
      def 			UpperCAmelCase   (   self      :    Tuple	,   __lowercase      :    Any      )							->   Tuple:
        __UpperCAmelCase     =						"UNwant\u00E9d,running"
        __UpperCAmelCase     =						"unwanted, running"
        return input_text, output_text
      def 			UpperCAmelCase   (   self      :    Union[str, Any]      )							->   Any:
        __UpperCAmelCase     =						self.tokenizer_class(self.vocab_file      )

        __UpperCAmelCase     =						tokenizer.tokenize("""UNwant\u00E9d,running"""      )
        self.assertListEqual(__lowercase	,   ["""un""", """##want""", """##ed""", """,""", """runn""", """##ing"""]      )
        self.assertListEqual(tokenizer.convert_tokens_to_ids(__lowercase      )	,   [9, 6, 7, 12, 10, 11]      )
      def 			UpperCAmelCase   (   self      :    Tuple      )							->   str:






        if not self.test_rust_tokenizer:
          return

        __UpperCAmelCase     =						self.get_tokenizer()
        __UpperCAmelCase     =						self.get_rust_tokenizer()

        __UpperCAmelCase     =						"UNwant\u00E9d,running"

        __UpperCAmelCase     =						tokenizer.tokenize(__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.tokenize(__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )

        __UpperCAmelCase     =						tokenizer.encode(__lowercase	,   add_special_tokens=__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.encode(__lowercase	,   add_special_tokens=__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )

        __UpperCAmelCase     =						self.get_rust_tokenizer()
        __UpperCAmelCase     =						tokenizer.encode(__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.encode(__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )

        # With lower casing
        __UpperCAmelCase     =						self.get_tokenizer(do_lower_case=__lowercase      )
        __UpperCAmelCase     =						self.get_rust_tokenizer(do_lower_case=__lowercase      )

        __UpperCAmelCase     =						"UNwant\u00E9d,running"

        __UpperCAmelCase     =						tokenizer.tokenize(__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.tokenize(__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )

        __UpperCAmelCase     =						tokenizer.encode(__lowercase	,   add_special_tokens=__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.encode(__lowercase	,   add_special_tokens=__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )

        __UpperCAmelCase     =						self.get_rust_tokenizer()
        __UpperCAmelCase     =						tokenizer.encode(__lowercase      )
        __UpperCAmelCase     =						rust_tokenizer.encode(__lowercase      )
        self.assertListEqual(__lowercase	,   __lowercase      )
      def 			UpperCAmelCase   (   self      :    List[Any]      )							->   Union[str, Any]:
        __UpperCAmelCase     =						BasicTokenizer()

        self.assertListEqual(tokenizer.tokenize("""ah\u535A\u63A8zz"""      )	,   ["""ah""", """\u535A""", """\u63A8""", """zz"""]      )
      def 			UpperCAmelCase   (   self      :    Optional[int]      )							->   Dict:
        __UpperCAmelCase     =						BasicTokenizer(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      :    Optional[Any]      )							->   List[Any]:
        __UpperCAmelCase     =						BasicTokenizer(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      :    Optional[Any]      )							->   Optional[Any]:
        __UpperCAmelCase     =						BasicTokenizer(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      :    List[Any]      )							->   Any:
        __UpperCAmelCase     =						BasicTokenizer(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      :    List[Any]      )							->   Tuple:
        __UpperCAmelCase     =						BasicTokenizer(do_lower_case=__lowercase      )

        self.assertListEqual(
            tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """      )	,   ["""HeLLo""", """!""", """how""", """Are""", """yoU""", """?"""]      )
      def 			UpperCAmelCase   (   self      :    List[str]      )							->   Union[str, Any]:
        __UpperCAmelCase     =						BasicTokenizer(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      :    Any      )							->   int:
        __UpperCAmelCase     =						BasicTokenizer(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      )							->   Union[str, Any]:
        __UpperCAmelCase     =						BasicTokenizer(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[Any]      )							->   List[Any]:
        __UpperCAmelCase     =						["[UNK]", "[CLS]", "[SEP]", "want", "##want", "##ed", "wa", "un", "runn", "##ing"]

        __UpperCAmelCase     =						{}
        for i, token in enumerate(__lowercase      ):
          __UpperCAmelCase     =						i
        __UpperCAmelCase     =						WordpieceTokenizer(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      :    List[str]      )							->   Dict:
        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      :    Optional[int]      )							->   Any:
        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[str]      )							->   Union[str, 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      :    Tuple      )							->   List[Any]:
        __UpperCAmelCase     =						self.get_tokenizer()
        __UpperCAmelCase     =						self.get_rust_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]"""]]      )

        self.assertListEqual(
            [rust_tokenizer.tokenize(__lowercase      ) for t in ["""Test""", """\xad""", """test"""]]	,   [["""[UNK]"""], [], ["""[UNK]"""]]      )
      @slow
      def 			UpperCAmelCase   (   self      :    List[str]      )							->   Dict:
        __UpperCAmelCase     =						self.tokenizer_class.from_pretrained("""google/electra-base-discriminator"""      )

        __UpperCAmelCase     =						tokenizer.encode("""sequence builders"""	,   add_special_tokens=__lowercase      )
        __UpperCAmelCase     =						tokenizer.encode("""multi-sequence build"""	,   add_special_tokens=__lowercase      )

        __UpperCAmelCase     =						tokenizer.build_inputs_with_special_tokens(__lowercase      )
        __UpperCAmelCase     =						tokenizer.build_inputs_with_special_tokens(__lowercase	,   __lowercase      )

        assert encoded_sentence == [101] + text + [102]
        assert encoded_pair == [101] + text + [102] + text_a + [102]
      def 			UpperCAmelCase   (   self      :    Any      )							->   str:






        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      :    Any      )							->   Dict:
        __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      )










from typing import TYPE_CHECKING

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


a	: str      =				{"configuration_hubert": ["HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP", "HubertConfig"]}

try:
       if not is_torch_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a	: str      =				[
           "HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
           "HubertForCTC",
           "HubertForSequenceClassification",
           "HubertModel",
           "HubertPreTrainedModel",
       ]


try:
       if not is_tf_available():
              raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
       pass
else:
       a	: List[str]      =				[
           "TF_HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST",
           "TFHubertForCTC",
           "TFHubertModel",
           "TFHubertPreTrainedModel",
       ]

if TYPE_CHECKING:
       from .configuration_hubert import HUBERT_PRETRAINED_CONFIG_ARCHIVE_MAP, HubertConfig

       try:
              if not is_torch_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_hubert import (
                  HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
                  HubertForCTC,
                  HubertForSequenceClassification,
                  HubertModel,
                  HubertPreTrainedModel,
              )

       try:
              if not is_tf_available():
                     raise OptionalDependencyNotAvailable()
       except OptionalDependencyNotAvailable:
              pass
       else:
              from .modeling_tf_hubert import (
                  TF_HUBERT_PRETRAINED_MODEL_ARCHIVE_LIST,
                  TFHubertForCTC,
                  TFHubertModel,
                  TFHubertPreTrainedModel,
              )


else:
       import sys

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


