





'''simple docstring'''






import json
import os
import unittest

from transformers import DebertaTokenizer, DebertaTokenizerFast
from transformers.models.deberta.tokenization_deberta import VOCAB_FILES_NAMES
from transformers.testing_utils import slow

from ...test_tokenization_common import TokenizerTesterMixin




class 					UpperCAmelCase	(     snake_case_				,						unittest.TestCase  ):
      _lowercase:     List[str]			=  DebertaTokenizer
      _lowercase:     Union[str, Any]			=  True
      _lowercase:     Union[str, Any]			=  DebertaTokenizerFast




      def 							lowercase__       (     self     :					Any    )							-> List[Any]:
            super().setUp()

            # Adapted from Sennrich et al. 2015 and https://github.com/rsennrich/subword-nmt
            _lowerCAmelCase					=				[
                """l""",
                """o""",
                """w""",
                """e""",
                """r""",
                """s""",
                """t""",
                """i""",
                """d""",
                """n""",
                """\u0120""",
                """\u0120l""",
                """\u0120n""",
                """\u0120lo""",
                """\u0120low""",
                """er""",
                """\u0120lowest""",
                """\u0120newer""",
                """\u0120wider""",
                """[UNK]""",
            ]
            _lowerCAmelCase					=				dict(zip(__snake_case     ,     range(len(__snake_case    )    )    )    )
            _lowerCAmelCase					=				["""#version: 0.2""", """\u0120 l""", """\u0120l o""", """\u0120lo w""", """e r""", """"""]
            _lowerCAmelCase					=				{"""unk_token""": """[UNK]"""}

            _lowerCAmelCase					=				os.path.join(self.tmpdirname     ,     VOCAB_FILES_NAMES["""vocab_file"""]    )
            _lowerCAmelCase					=				os.path.join(self.tmpdirname     ,     VOCAB_FILES_NAMES["""merges_file"""]    )
            with open(self.vocab_file     ,     """w"""     ,     encoding="""utf-8"""    ) as fp:
                  fp.write(json.dumps(__snake_case    ) + """\n"""    )
            with open(self.merges_file     ,     """w"""     ,     encoding="""utf-8"""    ) as fp:
                  fp.write("""\n""".join(__snake_case    )    )




      def 							lowercase__       (     self     :					List[str]     ,     **__snake_case     :					Any    )							-> Dict:
            kwargs.update(self.special_tokens_map    )
            return self.tokenizer_class.from_pretrained(self.tmpdirname     ,     **__snake_case    )




      def 							lowercase__       (     self     :					Dict     ,     __snake_case     :					Dict    )							-> Optional[Any]:
            _lowerCAmelCase					=				"""lower newer"""
            _lowerCAmelCase					=				"""lower newer"""
            return input_text, output_text




      def 							lowercase__       (     self     :					Union[str, Any]    )							-> List[str]:
            _lowerCAmelCase					=				self.get_tokenizer()
            _lowerCAmelCase					=				"""lower newer"""
            _lowerCAmelCase					=				["""l""", """o""", """w""", """er""", """\u0120""", """n""", """e""", """w""", """er"""]
            _lowerCAmelCase					=				tokenizer.tokenize(__snake_case    )
            self.assertListEqual(__snake_case     ,     __snake_case    )

            _lowerCAmelCase					=				tokens + [tokenizer.unk_token]
            _lowerCAmelCase					=				[0, 1, 2, 15, 10, 9, 3, 2, 15, 19]
            self.assertListEqual(tokenizer.convert_tokens_to_ids(__snake_case    )     ,     __snake_case    )




      def 							lowercase__       (     self     :					Union[str, Any]    )							-> Tuple:
            _lowerCAmelCase					=				self.get_tokenizer()
            _lowerCAmelCase					=				tokenizer("""Hello"""     ,     """World"""    )
            _lowerCAmelCase					=				[0, 0, 0, 0, 0, 0, 0, 1, 1, 1, 1, 1, 1]
            self.assertListEqual(tokd["""token_type_ids"""]     ,     __snake_case    )




      @slow
      def 							lowercase__       (     self     :					int    )							-> Union[str, Any]:
            _lowerCAmelCase					=				self.tokenizer_class.from_pretrained("""microsoft/deberta-base"""    )

            _lowerCAmelCase					=				tokenizer.encode("""sequence builders"""     ,     add_special_tokens=__snake_case    )
            _lowerCAmelCase					=				tokenizer.encode("""multi-sequence build"""     ,     add_special_tokens=__snake_case    )

            _lowerCAmelCase					=				tokenizer.encode(
                """sequence builders"""     ,     add_special_tokens=__snake_case     ,     add_prefix_space=__snake_case    )
            _lowerCAmelCase					=				tokenizer.encode(
                """sequence builders"""     ,     """multi-sequence build"""     ,     add_special_tokens=__snake_case     ,     add_prefix_space=__snake_case    )

            _lowerCAmelCase					=				tokenizer.build_inputs_with_special_tokens(__snake_case    )
            _lowerCAmelCase					=				tokenizer.build_inputs_with_special_tokens(__snake_case     ,     __snake_case    )

            assert encoded_sentence == encoded_text_from_decode
            assert encoded_pair == encoded_pair_from_decode




      @slow
      def 							lowercase__       (     self     :					Any    )							-> str:
            _lowerCAmelCase					=				[self.tokenizer_class]
            if self.test_rust_tokenizer:
                  tokenizer_classes.append(self.rust_tokenizer_class    )

            for tokenizer_class in tokenizer_classes:
                  _lowerCAmelCase					=				tokenizer_class.from_pretrained("""microsoft/deberta-base"""    )

                  _lowerCAmelCase					=				[
                      """ALBERT: A Lite BERT for Self-supervised Learning of Language Representations""",
                      """ALBERT incorporates two parameter reduction techniques""",
                      """The first one is a factorized embedding parameterization. By decomposing the large vocabulary"""
                      """ embedding matrix into two small matrices, we separate the size of the hidden layers from the size of"""
                      """ vocabulary embedding.""",
                  ]

                  _lowerCAmelCase					=				tokenizer(__snake_case     ,     padding=__snake_case    )
                  _lowerCAmelCase					=				[tokenizer.decode(__snake_case     ,     skip_special_tokens=__snake_case    ) for seq in encoding["""input_ids"""]]

                  # fmt: off
                  _lowerCAmelCase					=				{
                      """input_ids""": [
                          [1, 21_18, 1_11_26, 5_65, 35, 83, 2_51_91, 1_63, 1_88_54, 13, 1_21_56, 12, 1_61_01, 2_53_76, 1_38_07, 9, 2_22_05, 2_78_93, 16_35, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [1, 21_18, 1_11_26, 5_65, 2_45_36, 80, 4_37_97, 48_78, 73_73, 2, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [1, 1_33, 78, 65, 16, 10, 37_24, 15_38, 3_31_83, 1_13_03, 4_37_97, 19_38, 4, 8_70, 2_41_65, 2_91_05, 5, 7_39, 3_26_44, 3_31_83, 1_13_03, 3_61_73, 88, 80, 6_50, 78_21, 4_59_40, 6, 52, 25_59, 5, 18_36, 9, 5, 73_97, 1_31_71, 31, 5, 18_36, 9, 3_26_44, 3_31_83, 1_13_03, 4, 2]
                      ],
                      """token_type_ids""": [
                          [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0]
                      ],
                      """attention_mask""": [
                          [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0],
                          [1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1, 1]
                      ]
                  }
                  # fmt: on

                  _lowerCAmelCase					=				[
                      """ALBERT: A Lite BERT for Self-supervised Learning of Language Representations""",
                      """ALBERT incorporates two parameter reduction techniques""",
                      """The first one is a factorized embedding parameterization. By decomposing the large vocabulary"""
                      """ embedding matrix into two small matrices, we separate the size of the hidden layers from the size of"""
                      """ vocabulary embedding.""",
                  ]

                  self.assertDictEqual(encoding.data     ,     __snake_case    )

                  for expected, decoded in zip(__snake_case     ,     __snake_case    ):
                        self.assertEqual(__snake_case     ,     __snake_case    )













'''simple docstring'''






import os
import unittest

from transformers import BatchEncoding
from transformers.models.bert.tokenization_bert import (
    BasicTokenizer,
    WordpieceTokenizer,
    _is_control,
    _is_punctuation,
    _is_whitespace,
)
from transformers.models.prophetnet.tokenization_prophetnet import VOCAB_FILES_NAMES, ProphetNetTokenizer
from transformers.testing_utils import require_torch, slow

from ...test_tokenization_common import TokenizerTesterMixin




class 					UpperCAmelCase	(     snake_case_				,						unittest.TestCase  ):
      _lowercase:     Optional[int]			=  ProphetNetTokenizer
      _lowercase:     Union[str, Any]			=  False




      def 							lowercase__       (     self     :					Dict    )							-> Optional[int]:
            super().setUp()

            _lowerCAmelCase					=				[
                """[UNK]""",
                """[CLS]""",
                """[SEP]""",
                """[PAD]""",
                """[MASK]""",
                """want""",
                """##want""",
                """##ed""",
                """wa""",
                """un""",
                """runn""",
                """##ing""",
                """,""",
                """low""",
                """lowest""",
            ]
            _lowerCAmelCase					=				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 							lowercase__       (     self     :					Tuple     ,     __snake_case     :					str    )							-> List[str]:
            _lowerCAmelCase					=				"""UNwant\u00E9d,running"""
            _lowerCAmelCase					=				"""unwanted, running"""
            return input_text, output_text




      def 							lowercase__       (     self     :					Union[str, Any]    )							-> Optional[int]:
            _lowerCAmelCase					=				self.tokenizer_class(self.vocab_file    )

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




      def 							lowercase__       (     self     :					List[Any]    )							-> List[str]:
            _lowerCAmelCase					=				BasicTokenizer()

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




      def 							lowercase__       (     self     :					Optional[Any]    )							-> List[Any]:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case    )

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




      def 							lowercase__       (     self     :					Dict    )							-> int:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case     ,     strip_accents=__snake_case    )

            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     :					Dict    )							-> int:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case     ,     strip_accents=__snake_case    )

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




      def 							lowercase__       (     self     :					Dict    )							-> Optional[Any]:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case    )

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




      def 							lowercase__       (     self     :					str    )							-> List[str]:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case    )

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




      def 							lowercase__       (     self     :					Union[str, Any]    )							-> str:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case     ,     strip_accents=__snake_case    )

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




      def 							lowercase__       (     self     :					Tuple    )							-> str:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case     ,     strip_accents=__snake_case    )

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




      def 							lowercase__       (     self     :					Dict    )							-> Optional[int]:
            _lowerCAmelCase					=				BasicTokenizer(do_lower_case=__snake_case     ,     never_split=["""[UNK]"""]    )

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




      def 							lowercase__       (     self     :					Any    )							-> Dict:
            _lowerCAmelCase					=				["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

            _lowerCAmelCase					=				{}
            for i, token in enumerate(__snake_case    ):
                  _lowerCAmelCase					=				i
            _lowerCAmelCase					=				WordpieceTokenizer(vocab=__snake_case     ,     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"""]    )




      @require_torch
      def 							lowercase__       (     self     :					Optional[int]    )							-> Optional[Any]:
            _lowerCAmelCase					=				self.tokenizer_class.from_pretrained("""microsoft/prophetnet-large-uncased"""    )

            _lowerCAmelCase					=				["""A long paragraph for summarization.""", """Another paragraph for summarization."""]
            _lowerCAmelCase					=				[10_37, 21_46, 2_04_23, 20_05, 76_80, 78_49, 39_89, 10_12, 1_02]
            _lowerCAmelCase					=				tokenizer(__snake_case     ,     padding=__snake_case     ,     return_tensors="""pt"""    )
            self.assertIsInstance(__snake_case     ,     __snake_case    )
            _lowerCAmelCase					=				list(batch.input_ids.numpy()[0]    )
            self.assertListEqual(__snake_case     ,     __snake_case    )

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




      def 							lowercase__       (     self     :					List[Any]    )							-> 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     :					str    )							-> Union[str, 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 							lowercase__       (     self     :					int    )							-> int:
            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(""" """    )    )




      @slow
      def 							lowercase__       (     self     :					int    )							-> Optional[int]:
            _lowerCAmelCase					=				self.tokenizer_class.from_pretrained("""microsoft/prophetnet-large-uncased"""    )

            _lowerCAmelCase					=				tokenizer.encode("""sequence builders"""     ,     add_special_tokens=__snake_case    )
            _lowerCAmelCase					=				tokenizer.encode("""multi-sequence build"""     ,     add_special_tokens=__snake_case    )

            _lowerCAmelCase					=				tokenizer.build_inputs_with_special_tokens(__snake_case    )
            _lowerCAmelCase					=				tokenizer.build_inputs_with_special_tokens(__snake_case     ,     __snake_case    )

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






