





"""simple docstring"""





import unittest
from pathlib import Path
from shutil import copyfile

from transformers import SPIECE_UNDERLINE, is_sentencepiece_available
from transformers.models.speech_to_text import SpeechaTextTokenizer
from transformers.models.speech_to_text.tokenization_speech_to_text import VOCAB_FILES_NAMES, save_json
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin


_UpperCamelCase							=      get_tests_dir('fixtures/test_sentencepiece.model')

if is_sentencepiece_available():
							import sentencepiece as sp


_UpperCamelCase							=      5
_UpperCamelCase							=      10
@require_sentencepiece
@require_tokenizers
class 		SCREAMING_SNAKE_CASE_							(    UpperCamelCase_       ,  unittest.TestCase   ):
				__snake_case   :       Any	 =		SpeechaTextTokenizer
				__snake_case   :       Dict	 =		False
				__snake_case   :       List[Any]	 =		True




				def 					__lowercase  (     self  :Dict			):
											super().setUp()

											__lowerCamelCase      :      Optional[int]         =sp.SentencePieceProcessor()
											spm_model.Load(__lowercase			)
											__lowerCamelCase      :      str         =['''<s>''', '''<pad>''', '''</s>''', '''<unk>''']

											vocab += [spm_model.IdToPiece(id_			) for id_ in range(len(__lowercase			)			)]
											__lowerCamelCase      :      Dict         =dict(zip(__lowercase						,    range(len(__lowercase			)			)			)			)

											__lowerCamelCase      :      Dict         =Path(self.tmpdirname			)
											save_json(__lowercase						,    save_dir / VOCAB_FILES_NAMES['''vocab_file''']			)
											if not (save_dir / VOCAB_FILES_NAMES["spm_file"]).exists():
																		copyfile(__lowercase						,    save_dir / VOCAB_FILES_NAMES['''spm_file''']			)

											__lowerCamelCase      :      List[str]         =SpeechaTextTokenizer.from_pretrained(self.tmpdirname			)
											tokenizer.save_pretrained(self.tmpdirname			)




				def 					__lowercase  (     self  :Optional[Any]			):
											__lowerCamelCase      :      Union[str, Any]         ='''<pad>'''
											__lowerCamelCase      :      Union[str, Any]         =1

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




				def 					__lowercase  (     self  :int			):
											__lowerCamelCase      :      str         =list(self.get_tokenizer().get_vocab().keys()			)

											self.assertEqual(vocab_keys[0]						,    '''<s>'''			)
											self.assertEqual(vocab_keys[1]						,    '''<pad>'''			)
											self.assertEqual(vocab_keys[-1]						,    '''j'''			)
											self.assertEqual(len(__lowercase			)						,    1001			)




				def 					__lowercase  (     self  :str			):
											self.assertEqual(self.get_tokenizer().vocab_size						,    1001			)




				def 					__lowercase  (     self  :Optional[Any]			):
											__lowerCamelCase      :      Optional[int]         =SpeechaTextTokenizer.from_pretrained(self.tmpdirname			)

											__lowerCamelCase      :      Dict         =tokenizer.tokenize('''This is a test'''			)
											self.assertListEqual(__lowercase						,    ['''▁This''', '''▁is''', '''▁a''', '''▁t''', '''est''']			)

											self.assertListEqual(
											    tokenizer.convert_tokens_to_ids(__lowercase			)						,    [289, 50, 14, 174, 386]						,    )

											__lowerCamelCase      :      Optional[int]         =tokenizer.tokenize('''I was born in 92000, and this is falsé.'''			)
											self.assertListEqual(
											    __lowercase						,    [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      :      Optional[Any]         =tokenizer.convert_tokens_to_ids(__lowercase			)
											self.assertListEqual(__lowercase						,    [12, 25, 88, 59, 28, 23, 11, 4, 606, 351, 351, 351, 7, 16, 70, 50, 76, 84, 10, 4, 8]			)

											__lowerCamelCase      :      Union[str, Any]         =tokenizer.convert_ids_to_tokens(__lowercase			)
											self.assertListEqual(
											    __lowercase						,    [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>''', '''.''']						,    )




				@slow
				def 					__lowercase  (     self  :Optional[Any]			):
											# fmt: off
											__lowerCamelCase      :      Optional[Any]         ={'''input_ids''': [[3791, 797, 31, 11, 64, 797, 31, 2429, 433, 12, 1176, 12, 20, 786, 915, 142, 2413, 240, 37, 3238, 797, 31, 11, 35, 93, 915, 142, 2413, 240, 37, 5540, 567, 1276, 93, 37, 610, 40, 62, 455, 657, 1042, 123, 780, 177, 37, 309, 241, 1298, 514, 20, 292, 2737, 114, 2469, 241, 85, 64, 302, 548, 528, 423, 4, 509, 406, 423, 37, 601, 4, 777, 302, 548, 528, 423, 284, 4, 3388, 511, 459, 4, 3555, 40, 321, 302, 705, 4, 3388, 511, 583, 326, 5, 5, 5, 62, 3310, 560, 177, 2680, 217, 1508, 32, 31, 853, 418, 64, 583, 511, 1605, 62, 35, 93, 560, 177, 2680, 217, 1508, 1521, 64, 583, 511, 519, 62, 20, 1515, 764, 20, 149, 261, 5625, 7972, 20, 5540, 567, 1276, 93, 3925, 1675, 11, 15, 802, 7972, 576, 217, 1508, 11, 35, 93, 1253, 2441, 15, 289, 652, 31, 416, 321, 3842, 115, 40, 911, 8, 476, 619, 4, 380, 142, 423, 335, 240, 35, 93, 264, 8, 11, 335, 569, 420, 163, 5, 2], [260, 548, 528, 423, 20, 451, 20, 2681, 1153, 3434, 20, 5540, 37, 567, 126, 1253, 2441, 3376, 449, 210, 431, 1563, 177, 767, 5540, 11, 1203, 472, 11, 2953, 685, 285, 364, 706, 1153, 20, 6799, 20, 2869, 20, 4464, 126, 40, 2429, 20, 1040, 866, 2664, 418, 20, 318, 20, 1726, 186, 20, 265, 522, 35, 93, 2191, 4634, 20, 1040, 12, 6799, 15, 228, 2356, 142, 31, 11, 5, 2, 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], [2575, 2666, 684, 1582, 1176, 12, 627, 149, 619, 20, 4902, 563, 11, 20, 149, 261, 3420, 2356, 174, 142, 4714, 131, 5, 2, 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, 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]], '''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, 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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 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, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 0, 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=__lowercase						,    model_name='''facebook/s2t-small-mustc-en-de-st'''						,    revision='''a14f04cf0776c02f62a8cb800cf7909e15ea23ad'''						,    )

@require_sentencepiece
class 		SCREAMING_SNAKE_CASE_							(    unittest.TestCase   ):
				__snake_case   :       Tuple	 =		"""valhalla/s2t_mustc_multilinguial_medium"""

				__snake_case   :       List[Any]	 =		"""C'est trop cool"""
				__snake_case   :       Dict	 =		"""Esto es genial"""




				@classmethod
				def 					__lowercase  (     cls  :int			):
											__lowerCamelCase      :      SpeechaTextTokenizer         =SpeechaTextTokenizer.from_pretrained(cls.checkpoint_name			)
											return cls




				def 					__lowercase  (     self  :int			):
											self.assertEqual(self.tokenizer.lang_code_to_id['''pt''']						,    4			)
											self.assertEqual(self.tokenizer.lang_code_to_id['''ru''']						,    6			)
											self.assertEqual(self.tokenizer.lang_code_to_id['''it''']						,    9			)
											self.assertEqual(self.tokenizer.lang_code_to_id['''de''']						,    11			)




				def 					__lowercase  (     self  :List[Any]			):
											self.assertEqual(self.tokenizer.vocab_size						,    1_0000			)




				def 					__lowercase  (     self  :Optional[Any]			):
											self.assertIn(__lowercase						,    self.tokenizer.all_special_ids			)
											__lowerCamelCase      :      int         =[ES_CODE, 4, 1601, 47, 7647, 2]
											__lowerCamelCase      :      str         =self.tokenizer.decode(__lowercase						,    skip_special_tokens=__lowercase			)
											__lowerCamelCase      :      Any         =self.tokenizer.decode(generated_ids[1:]						,    skip_special_tokens=__lowercase			)
											self.assertEqual(__lowercase						,    __lowercase			)
											self.assertNotIn(self.tokenizer.eos_token						,    __lowercase			)




				def 					__lowercase  (     self  :Union[str, Any]			):
											__lowerCamelCase      :      Tuple         ='''fr'''
											__lowerCamelCase      :      int         =self.tokenizer(self.french_text			).input_ids
											self.assertEqual(encoded[0]						,    __lowercase			)
											self.assertEqual(encoded[-1]						,    self.tokenizer.eos_token_id			)




				def 					__lowercase  (     self  :List[Any]			):
											__lowerCamelCase      :      Tuple         ='''fr'''
											self.assertListEqual(self.tokenizer.prefix_tokens						,    [FR_CODE]			)

											__lowerCamelCase      :      Optional[Any]         ='''es'''
											self.assertListEqual(self.tokenizer.prefix_tokens						,    [ES_CODE]			)











"""simple docstring"""





from .glue import GlueDataset, GlueDataTrainingArguments
from .language_modeling import (
    LineByLineTextDataset,
    LineByLineWithRefDataset,
    LineByLineWithSOPTextDataset,
    TextDataset,
    TextDatasetForNextSentencePrediction,
)
from .squad import SquadDataset, SquadDataTrainingArguments




