


'''simple docstring'''

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


a_						:       str      			=				"""▁"""

a_						:       List[Any]      			=				get_tests_dir("""fixtures/test_sentencepiece.model""")





@require_sentencepiece
@require_tokenizers
class   snake_case					(					lowercase__    ,		unittest.TestCase    ):





  """simple docstring"""




  _lowerCamelCase						=						BigBirdTokenizer
  _lowerCamelCase						=						BigBirdTokenizerFast
  _lowerCamelCase						=						True
  _lowerCamelCase						=						True






  def 						snake_case	(					self  ):







   """simple docstring"""
   super().setUp()

   lowerCamelCase_  				=						self.tokenizer_class(UpperCamelCase    ,						keep_accents=UpperCamelCase  )
   tokenizer.save_pretrained(self.tmpdirname  )






  def 						snake_case	(					self  ):







   """simple docstring"""
   lowerCamelCase_  				=						"<s>"
   lowerCamelCase_  				=						1

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






  def 						snake_case	(					self  ):







   """simple docstring"""
   lowerCamelCase_  				=						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(UpperCamelCase  )    ,						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_  				=						self.get_tokenizer()
   lowerCamelCase_  				=						self.get_rust_tokenizer()

   lowerCamelCase_  				=						"I was born in 92000, and this is falsé."

   lowerCamelCase_  				=						tokenizer.tokenize(UpperCamelCase  )
   lowerCamelCase_  				=						rust_tokenizer.tokenize(UpperCamelCase  )
   self.assertListEqual(UpperCamelCase    ,						UpperCamelCase  )

   lowerCamelCase_  				=						tokenizer.encode(UpperCamelCase    ,						add_special_tokens=UpperCamelCase  )
   lowerCamelCase_  				=						rust_tokenizer.encode(UpperCamelCase    ,						add_special_tokens=UpperCamelCase  )
   self.assertListEqual(UpperCamelCase    ,						UpperCamelCase  )

   lowerCamelCase_  				=						self.get_rust_tokenizer()
   lowerCamelCase_  				=						tokenizer.encode(UpperCamelCase  )
   lowerCamelCase_  				=						rust_tokenizer.encode(UpperCamelCase  )
   self.assertListEqual(UpperCamelCase    ,						UpperCamelCase  )






  def 						snake_case	(					self  ):







   """simple docstring"""
   lowerCamelCase_  				=						BigBirdTokenizer(UpperCamelCase    ,						keep_accents=UpperCamelCase  )

   lowerCamelCase_  				=						tokenizer.tokenize("This is a test"  )
   self.assertListEqual(UpperCamelCase    ,						["▁This", "▁is", "▁a", "▁t", "est"]  )

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

   lowerCamelCase_  				=						tokenizer.tokenize("I was born in 92000, and this is falsé."  )
   self.assertListEqual(
       UpperCamelCase    ,						[
           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_  				=						tokenizer.convert_tokens_to_ids(UpperCamelCase  )
   self.assertListEqual(
       UpperCamelCase    ,						[8, 21, 84, 55, 24, 19, 7, 0, 602, 347, 347, 347, 3, 12, 66, 46, 72, 80, 6, 0, 4]    ,						)

   lowerCamelCase_  				=						tokenizer.convert_ids_to_tokens(UpperCamelCase  )
   self.assertListEqual(
       UpperCamelCase    ,						[
           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_  				=						"Hello World!"
   lowerCamelCase_  				=						[65, 1_8536, 2260, 101, 66]

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






  @slow
  def 						snake_case	(					self  ):







   """simple docstring"""
   lowerCamelCase_  				=						(
       "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_  				=						[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(UpperCamelCase    ,						self.big_tokenizer.encode(UpperCamelCase  )  )






  @require_torch
  @slow
  def 						snake_case	(					self  ):







   """simple docstring"""
   import torch

   from transformers import BigBirdConfig, BigBirdModel

   # Build sequence
   lowerCamelCase_  				=						list(self.big_tokenizer.get_vocab().keys()  )[:10]
   lowerCamelCase_  				=						" ".join(UpperCamelCase  )
   lowerCamelCase_  				=						self.big_tokenizer.encode_plus(UpperCamelCase    ,						return_tensors="pt"    ,						return_token_type_ids=UpperCamelCase  )
   lowerCamelCase_  				=						self.big_tokenizer.batch_encode_plus(
       [sequence + " " + sequence]    ,						return_tensors="pt"    ,						return_token_type_ids=UpperCamelCase  )

   lowerCamelCase_  				=						BigBirdConfig(attention_type="original_full"  )
   lowerCamelCase_  				=						BigBirdModel(UpperCamelCase  )

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

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






  @slow
  def 						snake_case	(					self  ):







   """simple docstring"""
   lowerCamelCase_  				=						BigBirdTokenizer.from_pretrained("google/bigbird-roberta-base"  )
   lowerCamelCase_  				=						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"""
   # fmt: off
   lowerCamelCase_  				=						{"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=UpperCamelCase    ,						model_name="google/bigbird-roberta-base"    ,						revision="215c99f1600e06f83acce68422f2035b2b5c3510"    ,						)







'''simple docstring'''

def __snake_case      (   UpperCAmelCase_		:		int = 100	):
 lowerCamelCase_  				=						n * (n + 1) * (2 * n + 1) / 6
 lowerCamelCase_  				=						(n * (n + 1) / 2) ** 2
 return int(square_of_sum - sum_of_squares	)


if __name__ == "__main__":
     print(f'''{solution() = }''')



