



import unittest

from transformers import BigBirdTokenizer, BigBirdTokenizerFast
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, require_torch, slow
from transformers.utils import cached_property

from ...test_tokenization_common import TokenizerTesterMixin


_snake_case	=						"▁"

_snake_case	=						get_tests_dir("fixtures/test_sentencepiece.model")



@require_sentencepiece
@require_tokenizers
class 						UpperCAmelCase_					( a     ,      unittest.TestCase):
       lowerCamelCase__         =	BigBirdTokenizer
       lowerCamelCase__         =	BigBirdTokenizerFast
       lowerCamelCase__         =	True
       lowerCamelCase__         =	True







       def      snake_case__   (     self):

              '''simple docstring'''

              super().setUp()

              _lowerCAmelCase  :							List[str]				     =					self.tokenizer_class(__a,    keep_accents=__a)
              tokenizer.save_pretrained(self.tmpdirname)







       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							List[Any]				     =					"<s>"
              _lowerCAmelCase  :							Union[str, Any]				     =					1

              self.assertEqual(self.get_tokenizer()._convert_token_to_id(__a),    __a)
              self.assertEqual(self.get_tokenizer()._convert_id_to_token(__a),    __a)







       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							Tuple				     =					list(self.get_tokenizer().get_vocab().keys())

              self.assertEqual(vocab_keys[0],    "<unk>")
              self.assertEqual(vocab_keys[1],    "<s>")
              self.assertEqual(vocab_keys[-1],    "[MASK]")
              self.assertEqual(len(__a),    1004)







       def      snake_case__   (     self):

              '''simple docstring'''

              self.assertEqual(self.get_tokenizer().vocab_size,    1000)







       def      snake_case__   (     self):

              '''simple docstring'''

              if not self.test_rust_tokenizer:
                     return

              _lowerCAmelCase  :							List[str]				     =					self.get_tokenizer()
              _lowerCAmelCase  :							Tuple				     =					self.get_rust_tokenizer()

              _lowerCAmelCase  :							List[Any]				     =					"I was born in 92000, and this is falsé."

              _lowerCAmelCase  :							Optional[int]				     =					tokenizer.tokenize(__a)
              _lowerCAmelCase  :							Any				     =					rust_tokenizer.tokenize(__a)
              self.assertListEqual(__a,    __a)

              _lowerCAmelCase  :							int				     =					tokenizer.encode(__a,    add_special_tokens=__a)
              _lowerCAmelCase  :							Dict				     =					rust_tokenizer.encode(__a,    add_special_tokens=__a)
              self.assertListEqual(__a,    __a)

              _lowerCAmelCase  :							str				     =					self.get_rust_tokenizer()
              _lowerCAmelCase  :							List[Any]				     =					tokenizer.encode(__a)
              _lowerCAmelCase  :							Any				     =					rust_tokenizer.encode(__a)
              self.assertListEqual(__a,    __a)







       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							str				     =					BigBirdTokenizer(__a,    keep_accents=__a)

              _lowerCAmelCase  :							List[Any]				     =					tokenizer.tokenize("This is a test")
              self.assertListEqual(__a,    ["▁This", "▁is", "▁a", "▁t", "est"])

              self.assertListEqual(
                  tokenizer.convert_tokens_to_ids(__a),    [285, 46, 10, 170, 382],    )

              _lowerCAmelCase  :							int				     =					tokenizer.tokenize("I was born in 92000, and this is falsé.")
              self.assertListEqual(
                  __a,    [
                      SPIECE_UNDERLINE + "I",
                      SPIECE_UNDERLINE + "was",
                      SPIECE_UNDERLINE + "b",
                      "or",
                      "n",
                      SPIECE_UNDERLINE + "in",
                      SPIECE_UNDERLINE + "",
                      "9",
                      "2",
                      "0",
                      "0",
                      "0",
                      ",",
                      SPIECE_UNDERLINE + "and",
                      SPIECE_UNDERLINE + "this",
                      SPIECE_UNDERLINE + "is",
                      SPIECE_UNDERLINE + "f",
                      "al",
                      "s",
                      "é",
                      ".",
                  ],    )
              _lowerCAmelCase  :							Dict				     =					tokenizer.convert_tokens_to_ids(__a)
              self.assertListEqual(
                  __a,    [8, 21, 84, 55, 24, 19, 7, 0, 602, 347, 347, 347, 3, 12, 66, 46, 72, 80, 6, 0, 4],    )

              _lowerCAmelCase  :							List[Any]				     =					tokenizer.convert_ids_to_tokens(__a)
              self.assertListEqual(
                  __a,    [
                      SPIECE_UNDERLINE + "I",
                      SPIECE_UNDERLINE + "was",
                      SPIECE_UNDERLINE + "b",
                      "or",
                      "n",
                      SPIECE_UNDERLINE + "in",
                      SPIECE_UNDERLINE + "",
                      "<unk>",
                      "2",
                      "0",
                      "0",
                      "0",
                      ",",
                      SPIECE_UNDERLINE + "and",
                      SPIECE_UNDERLINE + "this",
                      SPIECE_UNDERLINE + "is",
                      SPIECE_UNDERLINE + "f",
                      "al",
                      "s",
                      "<unk>",
                      ".",
                  ],    )







       @cached_property
       def      snake_case__   (     self):

              '''simple docstring'''

              return BigBirdTokenizer.from_pretrained("google/bigbird-roberta-base")







       @slow
       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							Any				     =					"Hello World!"
              _lowerCAmelCase  :							Any				     =					[65, 1_8536, 2260, 101, 66]

              self.assertListEqual(__a,    self.big_tokenizer.encode(__a))







       @slow
       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							Union[str, Any]				     =					(
                  "This is a very long text with a lot of weird characters, such as: . , ~ ? ( ) \" [ ] ! : - . Also we will"
                  " add words that should not exsist and be tokenized to <unk>, such as saoneuhaoesuth"
              )
              # fmt: off
              _lowerCAmelCase  :							Optional[Any]				     =					[65, 871, 419, 358, 946, 991, 2521, 452, 358, 1357, 387, 7751, 3536, 112, 985, 456, 126, 865, 938, 5400, 5734, 458, 1368, 467, 786, 2462, 5246, 1159, 633, 865, 4519, 457, 582, 852, 2557, 427, 916, 508, 405, 3_4324, 497, 391, 408, 1_1342, 1244, 385, 100, 938, 985, 456, 574, 362, 1_2597, 3200, 3129, 1172, 66]  # noqa: E231
              # fmt: on
              self.assertListEqual(__a,    self.big_tokenizer.encode(__a))







       @require_torch
       @slow
       def      snake_case__   (     self):

              '''simple docstring'''

              import torch

              from transformers import BigBirdConfig, BigBirdModel

              # Build sequence
              _lowerCAmelCase  :							Any				     =					list(self.big_tokenizer.get_vocab().keys())[:10]
              _lowerCAmelCase  :							str				     =					" ".join(__a)
              _lowerCAmelCase  :							int				     =					self.big_tokenizer.encode_plus(__a,    return_tensors="pt",    return_token_type_ids=__a)
              _lowerCAmelCase  :							List[Any]				     =					self.big_tokenizer.batch_encode_plus(
                  [sequence + " " + sequence],    return_tensors="pt",    return_token_type_ids=__a)

              _lowerCAmelCase  :							Tuple				     =					BigBirdConfig(attention_type="original_full")
              _lowerCAmelCase  :							List[Any]				     =					BigBirdModel(__a)

              assert model.get_input_embeddings().weight.shape[0] >= self.big_tokenizer.vocab_size

              with torch.no_grad():
                     model(**__a)
                     model(**__a)







       @slow
       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							Tuple				     =					BigBirdTokenizer.from_pretrained("google/bigbird-roberta-base")
              _lowerCAmelCase  :							Tuple				     =					tokenizer.decode(tokenizer("Paris is the [MASK].").input_ids)

              self.assertTrue(decoded_text == "[CLS] Paris is the[MASK].[SEP]")







       @slow
       def      snake_case__   (     self):

              '''simple docstring'''

              _lowerCAmelCase  :							List[Any]				     =					{"input_ids": [[65, 3_9286, 458, 3_6335, 2001, 456, 1_3073, 1_3266, 455, 113, 7746, 1741, 1_1157, 391, 1_3073, 1_3266, 455, 113, 3967, 3_5412, 113, 4936, 109, 3870, 2377, 113, 3_0084, 4_5720, 458, 134, 1_7496, 112, 503, 1_1672, 113, 118, 112, 5665, 1_3347, 3_8687, 112, 1496, 3_1389, 112, 3268, 4_7264, 134, 962, 112, 1_6377, 8035, 2_3130, 430, 1_2169, 1_5518, 2_8592, 458, 146, 4_1697, 109, 391, 1_2169, 1_5518, 1_6689, 458, 146, 4_1358, 109, 452, 726, 4034, 111, 763, 3_5412, 5082, 388, 1903, 111, 9051, 391, 2870, 4_8918, 1900, 1123, 550, 998, 112, 9586, 1_5985, 455, 391, 410, 2_2955, 3_7636, 114, 66], [65, 448, 1_7496, 419, 3663, 385, 763, 113, 2_7533, 2870, 3283, 1_3043, 1639, 2_4713, 523, 656, 2_4013, 1_8550, 2521, 517, 2_7014, 2_1244, 420, 1212, 1465, 391, 927, 4833, 388, 578, 1_1786, 114, 66, 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], [65, 484, 2169, 7687, 2_1932, 1_8146, 726, 363, 1_7032, 3391, 114, 66, 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, 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, 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, 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, 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], [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, 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]]}  # noqa: E501
              # fmt: on

              self.tokenizer_integration_test_util(
                  expected_encoding=__a,    model_name="google/bigbird-roberta-base",    revision="215c99f1600e06f83acce68422f2035b2b5c3510",    )





def        A       (      _lowerCamelCase						):





							'''simple docstring'''




							if length <= 0 or not isinstance(_lowerCamelCase       ,				_lowerCamelCase						):
														raise ValueError("Length must be a positive integer."						)
							return [n * (2 * n - 1) for n in range(_lowerCamelCase						)]


if __name__ == "__main__":
						print(hexagonal_numbers(length=5))
						print(hexagonal_numbers(length=10))
