






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  UpperCamelCase    (			A_	,  unittest.TestCase   ):
 UpperCamelCase      : Tuple    =      ElectraTokenizer
 UpperCamelCase      : Tuple    =      ElectraTokenizerFast
 UpperCamelCase      : Any    =      True
 UpperCamelCase      : List[str]    =      True
 UpperCamelCase      : Any    =      filter_non_english


 def _lowercase			(  self	:		Optional[Any]							)      -> Optional[Any]:
   super().setUp()

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


 def _lowercase			(  self	:		Union[str, Any]							)      -> Tuple:
   _a						: Optional[Any]    =			self.tokenizer_class(self.vocab_file							)

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


 def _lowercase			(  self	:		Tuple							)      -> Any:
   if not self.test_rust_tokenizer:
     return

   _a						: Dict    =			self.get_tokenizer()
   _a						: Optional[int]    =			self.get_rust_tokenizer()

   _a						: str    =			"""UNwant\u00E9d,running"""

   _a						: Union[str, Any]    =			tokenizer.tokenize(UpperCAmelCase__							)
   _a						: Optional[Any]    =			rust_tokenizer.tokenize(UpperCAmelCase__							)
   self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)

   _a						: str    =			tokenizer.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
   _a						: Optional[Any]    =			rust_tokenizer.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
   self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)

   _a						: str    =			self.get_rust_tokenizer()
   _a						: Union[str, Any]    =			tokenizer.encode(UpperCAmelCase__							)
   _a						: Optional[int]    =			rust_tokenizer.encode(UpperCAmelCase__							)
   self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)

   # With lower casing
   _a						: Optional[Any]    =			self.get_tokenizer(do_lower_case=UpperCAmelCase__							)
   _a						: Tuple    =			self.get_rust_tokenizer(do_lower_case=UpperCAmelCase__							)

   _a						: Union[str, Any]    =			"""UNwant\u00E9d,running"""

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

   _a						: List[str]    =			tokenizer.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
   _a						: Any    =			rust_tokenizer.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
   self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)

   _a						: Optional[Any]    =			self.get_rust_tokenizer()
   _a						: Tuple    =			tokenizer.encode(UpperCAmelCase__							)
   _a						: Optional[int]    =			rust_tokenizer.encode(UpperCAmelCase__							)
   self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)


 def _lowercase			(  self	:		Optional[int]							)      -> int:
   _a						: Optional[Any]    =			BasicTokenizer()

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


 def _lowercase			(  self	:		Union[str, Any]							)      -> Dict:
   _a						: List[Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							)

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


 def _lowercase			(  self	:		str							)      -> Dict:
   _a						: str    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							,		strip_accents=UpperCAmelCase__							)

   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	:		Tuple							)      -> Union[str, Any]:
   _a						: Optional[Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							,		strip_accents=UpperCAmelCase__							)

   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							)      -> Union[str, Any]:
   _a						: Optional[Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							)

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


 def _lowercase			(  self	:		Optional[int]							)      -> str:
   _a						: Union[str, Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							)

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


 def _lowercase			(  self	:		str							)      -> Dict:
   _a						: List[Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							,		strip_accents=UpperCAmelCase__							)

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


 def _lowercase			(  self	:		Optional[Any]							)      -> Optional[Any]:
   _a						: Union[str, Any]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							,		strip_accents=UpperCAmelCase__							)

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


 def _lowercase			(  self	:		Optional[int]							)      -> str:
   _a						: Optional[int]    =			BasicTokenizer(do_lower_case=UpperCAmelCase__							,		never_split=["""[UNK]"""]							)

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


 def _lowercase			(  self	:		Tuple							)      -> Union[str, Any]:
   _a						: Dict    =			["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

   _a						: int    =			{}
   for i, token in enumerate(UpperCAmelCase__							):
     _a						: Any    =			i
   _a						: Tuple    =			WordpieceTokenizer(vocab=UpperCAmelCase__							,		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	:		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							)      -> 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	:		List[Any]							)      -> Optional[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 _lowercase			(  self	:		str							)      -> List[Any]:
   _a						: Tuple    =			self.get_tokenizer()
   _a						: Union[str, Any]    =			self.get_rust_tokenizer()

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

   self.assertListEqual(
       [rust_tokenizer.tokenize(UpperCAmelCase__							) for t in ["""Test""", """\xad""", """test"""]]							,		[["""[UNK]"""], [], ["""[UNK]"""]]							)


 @slow
 def _lowercase			(  self	:		List[str]							)      -> str:
   _a						: Union[str, Any]    =			self.tokenizer_class.from_pretrained("""google/electra-base-discriminator"""							)

   _a						: Tuple    =			tokenizer.encode("""sequence builders"""							,		add_special_tokens=UpperCAmelCase__							)
   _a						: Dict    =			tokenizer.encode("""multi-sequence build"""							,		add_special_tokens=UpperCAmelCase__							)

   _a						: Optional[int]    =			tokenizer.build_inputs_with_special_tokens(UpperCAmelCase__							)
   _a						: str    =			tokenizer.build_inputs_with_special_tokens(UpperCAmelCase__							,		UpperCAmelCase__							)

   assert encoded_sentence == [101] + text + [102]
   assert encoded_pair == [101] + text + [102] + text_a + [102]


 def _lowercase			(  self	:		List[str]							)      -> Optional[int]:
   for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
     with self.subTest(f"""{tokenizer.__class__.__name__} ({pretrained_name})"""							):
       _a						: Union[str, Any]    =			self.rust_tokenizer_class.from_pretrained(UpperCAmelCase__							,		**UpperCAmelCase__							)

       _a						: Tuple    =			f"""A, naïve {tokenizer_r.mask_token} AllenNLP sentence."""
       _a						: int    =			tokenizer_r.encode_plus(
           UpperCAmelCase__							,		return_attention_mask=UpperCAmelCase__							,		return_token_type_ids=UpperCAmelCase__							,		return_offsets_mapping=UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							,		)

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

       _a						: Any    =			tokenizer_p.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
       _a						: Any    =			tokenizer_r.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)

       _a						: Any    =			tokenizer_r.convert_ids_to_tokens(UpperCAmelCase__							)
       _a						: int    =			tokenizer_p.convert_ids_to_tokens(UpperCAmelCase__							)

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

       _a						: Dict    =			False
       _a						: List[Any]    =			self.rust_tokenizer_class.from_pretrained(UpperCAmelCase__							,		**UpperCAmelCase__							)
       _a						: Tuple    =			self.tokenizer_class.from_pretrained(UpperCAmelCase__							,		**UpperCAmelCase__							)

       _a						: Union[str, Any]    =			tokenizer_r.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)
       _a						: List[Any]    =			tokenizer_p.encode(UpperCAmelCase__							,		add_special_tokens=UpperCAmelCase__							)

       _a						: Optional[Any]    =			tokenizer_r.convert_ids_to_tokens(UpperCAmelCase__							)
       _a						: Any    =			tokenizer_p.convert_ids_to_tokens(UpperCAmelCase__							)

       # it is expected that only the first Chinese character is not preceded by "##".
       _a						: int    =			[
           f"""##{token}""" if idx != 0 else token for idx, token in enumerate(UpperCAmelCase__							)
       ]
       self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)
       self.assertListEqual(UpperCAmelCase__							,		UpperCAmelCase__							)









"""simple docstring"""




import json
import os
from dataclasses import dataclass

import numpy as np
import pyarrow as pa

import datasets
from utils import get_duration


_snake_case					   =     1000_0000_0000
_snake_case					   =     1_0000

_snake_case   ,				_snake_case					   =     os.path.split(__file__)
_snake_case					   =     os.path.join(RESULTS_BASEPATH, 'results', RESULTS_FILENAME.replace('.py', '.json'))

def     lowerCAmelCase__  (  UpperCamelCase__  ,     UpperCamelCase__       ):



  '''simple docstring'''
  _a						: Union[str, Any]    =			pa.Table.from_pydict({"""col""": [0] * chunk_size}       )
  _a						: Dict    =			pa.concat_tables([table] * (num_examples // chunk_size)       )
  return datasets.Dataset(UpperCamelCase__  ,     fingerprint="""table_100B"""       )






@dataclass
class  UpperCamelCase    :
 UpperCamelCase      : int
 UpperCamelCase      : int
 UpperCamelCase      : int
 UpperCamelCase      : int


 def _lowercase			(  self	:		Optional[int]							)      -> Dict:
   _a						: Tuple    =			np.random.default_rng(self.seed							)
   _a						: Dict    =			rng.integers(low=self.low							,		high=self.high							,		size=self.size							).tolist()


 def __iter__(  self	:		List[Any]							)      -> Optional[Any]:
   return iter(self._sampled_values							)







 def __len__(  self	:		Dict							)      -> Optional[Any]:
   return self.size







@get_duration
def     lowerCAmelCase__  (  UpperCamelCase__       ):



  '''simple docstring'''
  _a						: Union[str, Any]    =			dataset[0]

@get_duration
def     lowerCAmelCase__  (  UpperCamelCase__       ):



  '''simple docstring'''
  _a						: Optional[Any]    =			dataset[-1]

@get_duration
def     lowerCAmelCase__  (  UpperCamelCase__       ):



  '''simple docstring'''
  _a						: str    =			dataset[range(len(UpperCamelCase__       ) // 2  ,     len(UpperCamelCase__       ) // 2 + 1_0_2_4       )]

@get_duration
def     lowerCAmelCase__  (  UpperCamelCase__       ):



  '''simple docstring'''
  _a						: Tuple    =			dataset[RandIter(0  ,     len(UpperCamelCase__       )  ,     1_0_2_4  ,     seed=4_2       )]

def     lowerCAmelCase__  (  ):



  '''simple docstring'''
  _a						: str    =			{"""num examples""": SPEED_TEST_N_EXAMPLES}
  _a						: Dict    =			(get_first_row, get_last_row, get_batch_of_aaaa_rows, get_batch_of_aaaa_random_rows)
  print("""generating dataset"""       )
  _a						: List[str]    =			generate_100B_dataset(num_examples=UpperCamelCase__  ,     chunk_size=UpperCamelCase__       )
  print("""Functions"""       )
  for func in functions:
    print(func.__name__       )
    _a						: Optional[Any]    =			func(UpperCamelCase__       )

  with open(UpperCamelCase__  ,     """wb"""       ) as f:
    f.write(json.dumps(UpperCamelCase__       ).encode("""utf-8"""       )       )


if __name__ == "__main__":  # useful to run the profiler
 benchmark_table_100B()

