






'''simple docstring'''







import itertools
import json
import os
import unittest

from transformers import AddedToken, RobertaTokenizer, RobertaTokenizerFast
from transformers.models.roberta.tokenization_roberta import VOCAB_FILES_NAMES
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin




@require_tokenizers
class 		UpperCamelCase__     (a ,unittest.TestCase      ):




		'''simple docstring'''





		_UpperCamelCase						=       RobertaTokenizer
		_UpperCamelCase						=       RobertaTokenizerFast
		_UpperCamelCase						=       True
		_UpperCamelCase						=       {'cls_token': '<s>'}






		def       UpperCamelCase_       (   self							):
								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(_lowerCAmelCase ,range(len(_lowerCAmelCase							)							)							)							)
								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(_lowerCAmelCase							) + """\n"""							)
								with open(self.merges_file ,"""w""" ,encoding="""utf-8"""							) as fp:
														fp.write("""\n""".join(_lowerCAmelCase							)							)






		def       UpperCamelCase_       (   self ,**_lowerCAmelCase							):
								kwargs.update(self.special_tokens_map							)
								return self.tokenizer_class.from_pretrained(self.tmpdirname ,**_lowerCAmelCase							)






		def       UpperCamelCase_       (   self ,**_lowerCAmelCase							):
								kwargs.update(self.special_tokens_map							)
								return RobertaTokenizerFast.from_pretrained(self.tmpdirname ,**_lowerCAmelCase							)






		def       UpperCamelCase_       (   self ,_lowerCAmelCase							):
								lowerCamelCase__								=	"""lower newer"""
								lowerCamelCase__								=	"""lower newer"""
								return input_text, output_text






		def       UpperCamelCase_       (   self							):
								lowerCamelCase__								=	self.tokenizer_class(self.vocab_file ,self.merges_file ,**self.special_tokens_map							)
								lowerCamelCase__								=	"""lower newer"""
								lowerCamelCase__								=	["""l""", """o""", """w""", """er""", """\u0120""", """n""", """e""", """w""", """er"""]
								lowerCamelCase__								=	tokenizer.tokenize(_lowerCAmelCase							)  # , add_prefix_space=True)
								self.assertListEqual(_lowerCAmelCase ,_lowerCAmelCase							)

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






		def       UpperCamelCase_       (   self							):
								lowerCamelCase__								=	self.get_tokenizer()

								self.assertListEqual(tokenizer.encode("""Hello world!""" ,add_special_tokens=_lowerCAmelCase							) ,[0, 3_14_14, 2_32, 3_28, 2]							)
								self.assertListEqual(
								    tokenizer.encode("""Hello world! cécé herlolip 418""" ,add_special_tokens=_lowerCAmelCase							) ,[0, 3_14_14, 2_32, 3_28, 7_40, 11_40, 1_26_95, 69, 4_60_78, 15_88, 2] ,)






		@slow
		def       UpperCamelCase_       (   self							):
								lowerCamelCase__								=	self.tokenizer_class.from_pretrained("""roberta-base"""							)

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

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

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

								assert encoded_sentence == encoded_text_from_decode
								assert encoded_pair == encoded_pair_from_decode






		def       UpperCamelCase_       (   self							):
								lowerCamelCase__								=	self.get_tokenizer()

								lowerCamelCase__								=	"""Encode this sequence."""
								lowerCamelCase__								=	tokenizer.byte_encoder[""" """.encode("""utf-8"""							)[0]]

								# Testing encoder arguments
								lowerCamelCase__								=	tokenizer.encode(_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase							)
								lowerCamelCase__								=	tokenizer.convert_ids_to_tokens(encoded[0]							)[0]
								self.assertNotEqual(_lowerCAmelCase ,_lowerCAmelCase							)

								lowerCamelCase__								=	tokenizer.encode(_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase							)
								lowerCamelCase__								=	tokenizer.convert_ids_to_tokens(encoded[0]							)[0]
								self.assertEqual(_lowerCAmelCase ,_lowerCAmelCase							)

								tokenizer.add_special_tokens({"""bos_token""": """<s>"""}							)
								lowerCamelCase__								=	tokenizer.encode(_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
								lowerCamelCase__								=	tokenizer.convert_ids_to_tokens(encoded[1]							)[0]
								self.assertNotEqual(_lowerCAmelCase ,_lowerCAmelCase							)

								# Testing spaces after special tokens
								lowerCamelCase__								=	"""<mask>"""
								tokenizer.add_special_tokens(
								    {"""mask_token""": AddedToken(_lowerCAmelCase ,lstrip=_lowerCAmelCase ,rstrip=_lowerCAmelCase							)}							)  # mask token has a left space
								lowerCamelCase__								=	tokenizer.convert_tokens_to_ids(_lowerCAmelCase							)

								lowerCamelCase__								=	"""Encode <mask> sequence"""
								lowerCamelCase__								=	"""Encode <mask>sequence"""

								lowerCamelCase__								=	tokenizer.encode(_lowerCAmelCase							)
								lowerCamelCase__								=	encoded.index(_lowerCAmelCase							)
								lowerCamelCase__								=	tokenizer.convert_ids_to_tokens(encoded[mask_loc + 1]							)[0]
								self.assertEqual(_lowerCAmelCase ,_lowerCAmelCase							)

								lowerCamelCase__								=	tokenizer.encode(_lowerCAmelCase							)
								lowerCamelCase__								=	encoded.index(_lowerCAmelCase							)
								lowerCamelCase__								=	tokenizer.convert_ids_to_tokens(encoded[mask_loc + 1]							)[0]
								self.assertNotEqual(_lowerCAmelCase ,_lowerCAmelCase							)






		def       UpperCamelCase_       (   self							):
								pass






		def       UpperCamelCase_       (   self							):
								for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
														with self.subTest(F'''{tokenizer.__class__.__name__} ({pretrained_name})'''							):
																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(_lowerCAmelCase ,**_lowerCAmelCase							)
																				lowerCamelCase__								=	self.tokenizer_class.from_pretrained(_lowerCAmelCase ,**_lowerCAmelCase							)
																				lowerCamelCase__								=	"""A, <mask> AllenNLP sentence."""
																				lowerCamelCase__								=	tokenizer_r.encode_plus(_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase ,return_token_type_ids=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_p.encode_plus(_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase ,return_token_type_ids=_lowerCAmelCase							)

																				# token_type_ids should put 0 everywhere
																				self.assertEqual(sum(tokens_r["""token_type_ids"""]							) ,sum(tokens_p["""token_type_ids"""]							)							)

																				# attention_mask should put 1 everywhere, so sum over length should be 1
																				self.assertEqual(
																				    sum(tokens_r["""attention_mask"""]							) / len(tokens_r["""attention_mask"""]							) ,sum(tokens_p["""attention_mask"""]							) / len(tokens_p["""attention_mask"""]							) ,)

																				lowerCamelCase__								=	tokenizer_r.convert_ids_to_tokens(tokens_r["""input_ids"""]							)
																				lowerCamelCase__								=	tokenizer_p.convert_ids_to_tokens(tokens_p["""input_ids"""]							)

																				# Rust correctly handles the space before the mask while python doesnt
																				self.assertSequenceEqual(tokens_p["""input_ids"""] ,[0, 2_50, 6, 5_02_64, 38_23, 4_87, 2_19_92, 36_45, 4, 2]							)
																				self.assertSequenceEqual(tokens_r["""input_ids"""] ,[0, 2_50, 6, 5_02_64, 38_23, 4_87, 2_19_92, 36_45, 4, 2]							)

																				self.assertSequenceEqual(
																				    _lowerCAmelCase ,["""<s>""", """A""", """,""", """<mask>""", """ĠAllen""", """N""", """LP""", """Ġsentence""", """.""", """</s>"""]							)
																				self.assertSequenceEqual(
																				    _lowerCAmelCase ,["""<s>""", """A""", """,""", """<mask>""", """ĠAllen""", """N""", """LP""", """Ġsentence""", """.""", """</s>"""]							)






		def       UpperCamelCase_       (   self							):
								for trim_offsets, add_prefix_space in itertools.product([True, False] ,repeat=2							):
														lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
														    self.tmpdirname ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)

														lowerCamelCase__								=	json.loads(tokenizer_r.backend_tokenizer.pre_tokenizer.__getstate__()							)
														lowerCamelCase__								=	json.loads(tokenizer_r.backend_tokenizer.post_processor.__getstate__()							)

														self.assertEqual(pre_tokenizer_state["""add_prefix_space"""] ,_lowerCAmelCase							)

														self.assertEqual(post_processor_state["""add_prefix_space"""] ,_lowerCAmelCase							)
														self.assertEqual(post_processor_state["""trim_offsets"""] ,_lowerCAmelCase							)






		def       UpperCamelCase_       (   self							):
								# Test which aims to verify that the offsets are well adapted to the argument `add_prefix_space` and
								# `trim_offsets`
								for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
														with self.subTest(F'''{tokenizer.__class__.__name__} ({pretrained_name})'''							):
																				lowerCamelCase__								=	"""hello"""  # `hello` is a token in the vocabulary of `pretrained_name`
																				lowerCamelCase__								=	F'''{text_of_1_token} {text_of_1_token}'''

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(len(_lowerCAmelCase							) + 1, len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(len(_lowerCAmelCase							) + 1, len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(len(_lowerCAmelCase							), len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(len(_lowerCAmelCase							), len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	F''' {text}'''

																				# tokenizer_r = self.rust_tokenizer_class.from_pretrained(
																				#     pretrained_name, use_fast=True, add_prefix_space=True, trim_offsets=True
																				# )
																				# encoding = tokenizer_r(text, return_offsets_mapping=True, add_special_tokens=False)
																				# self.assertEqual(encoding.offset_mapping[0], (1, 1 + len(text_of_1_token)))
																				# self.assertEqual(
																				#     encoding.offset_mapping[1],
																				#     (1 + len(text_of_1_token) + 1, 1 + len(text_of_1_token) + 1 + len(text_of_1_token)),
																				# )

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(1, 1 + len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(1 + len(_lowerCAmelCase							) + 1, 1 + len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, 1 + len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(1 + len(_lowerCAmelCase							), 1 + len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)

																				lowerCamelCase__								=	self.rust_tokenizer_class.from_pretrained(
																				    _lowerCAmelCase ,use_fast=_lowerCAmelCase ,add_prefix_space=_lowerCAmelCase ,trim_offsets=_lowerCAmelCase							)
																				lowerCamelCase__								=	tokenizer_r(_lowerCAmelCase ,return_offsets_mapping=_lowerCAmelCase ,add_special_tokens=_lowerCAmelCase							)
																				self.assertEqual(encoding.offset_mapping[0] ,(0, 1 + len(_lowerCAmelCase							))							)
																				self.assertEqual(
																				    encoding.offset_mapping[1] ,(1 + len(_lowerCAmelCase							), 1 + len(_lowerCAmelCase							) + 1 + len(_lowerCAmelCase							)) ,)









'''simple docstring'''



import os
from shutil import copyfile
from typing import List, Optional, Tuple

from ...tokenization_utils import AddedToken
from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import is_sentencepiece_available, logging


if is_sentencepiece_available():
      from .tokenization_xlnet import XLNetTokenizer
else:
      SCREAMING_SNAKE_CASE							=						None


SCREAMING_SNAKE_CASE							=						logging.get_logger(__name__)

SCREAMING_SNAKE_CASE							=						{'vocab_file': 'spiece.model', 'tokenizer_file': 'tokenizer.json'}

SCREAMING_SNAKE_CASE							=						{
    'vocab_file': {
        'xlnet-base-cased': 'https://huggingface.co/xlnet-base-cased/resolve/main/spiece.model',
        'xlnet-large-cased': 'https://huggingface.co/xlnet-large-cased/resolve/main/spiece.model',
    },
    'tokenizer_file': {
        'xlnet-base-cased': 'https://huggingface.co/xlnet-base-cased/resolve/main/tokenizer.json',
        'xlnet-large-cased': 'https://huggingface.co/xlnet-large-cased/resolve/main/tokenizer.json',
    },
}

SCREAMING_SNAKE_CASE							=						{
    'xlnet-base-cased': None,
    'xlnet-large-cased': None,
}

SCREAMING_SNAKE_CASE							=						'▁'

# Segments (not really needed)
SCREAMING_SNAKE_CASE							=						0
SCREAMING_SNAKE_CASE							=						1
SCREAMING_SNAKE_CASE							=						2
SCREAMING_SNAKE_CASE							=						3
SCREAMING_SNAKE_CASE							=						4
class UpperCAmelCase_       (     __A					):




      """simple docstring"""




      UpperCamelCase_             =      VOCAB_FILES_NAMES
      UpperCamelCase_             =      PRETRAINED_VOCAB_FILES_MAP
      UpperCamelCase_             =      PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
      UpperCamelCase_             =      '''left'''
      UpperCamelCase_             =      XLNetTokenizer






      def __init__(			self      :		int  ,   UpperCAmelCase      :		Dict=None  ,   UpperCAmelCase      :		str=None  ,   UpperCAmelCase      :		str=False  ,   UpperCAmelCase      :		List[Any]=True  ,   UpperCAmelCase      :		Optional[Any]=False  ,   UpperCAmelCase      :		int="<s>"  ,   UpperCAmelCase      :		Optional[int]="</s>"  ,   UpperCAmelCase      :		str="<unk>"  ,   UpperCAmelCase      :		Optional[Any]="<sep>"  ,   UpperCAmelCase      :		Optional[int]="<pad>"  ,   UpperCAmelCase      :		Optional[Any]="<cls>"  ,   UpperCAmelCase      :		Dict="<mask>"  ,   UpperCAmelCase      :		int=["<eop>", "<eod>"]  ,   **UpperCAmelCase      :		List[Any]  ,   )	->	List[str]:


       '''simple docstring'''

       lowercase       : Dict						=AddedToken(UpperCAmelCase  ,   lstrip=UpperCAmelCase  ,   rstrip=UpperCAmelCase		) if isinstance(UpperCAmelCase  ,   UpperCAmelCase		) else mask_token

       super().__init__(
           vocab_file=UpperCAmelCase  ,   tokenizer_file=UpperCAmelCase  ,   do_lower_case=UpperCAmelCase  ,   remove_space=UpperCAmelCase  ,   keep_accents=UpperCAmelCase  ,   bos_token=UpperCAmelCase  ,   eos_token=UpperCAmelCase  ,   unk_token=UpperCAmelCase  ,   sep_token=UpperCAmelCase  ,   pad_token=UpperCAmelCase  ,   cls_token=UpperCAmelCase  ,   mask_token=UpperCAmelCase  ,   additional_special_tokens=UpperCAmelCase  ,   **UpperCAmelCase  ,   )

       lowercase       : Tuple						=3
       lowercase       : Union[str, Any]						=do_lower_case
       lowercase       : Any						=remove_space
       lowercase       : int						=keep_accents
       lowercase       : int						=vocab_file
       lowercase       : Union[str, Any]						=False if not self.vocab_file else True






      def 					A__   (			self      :		Any  ,   UpperCAmelCase      :		List[int]  ,   UpperCAmelCase      :		Optional[List[int]] = None		)	->	List[int]:


       '''simple docstring'''

       lowercase       : Union[str, Any]						=[self.sep_token_id]
       lowercase       : Optional[Any]						=[self.cls_token_id]
       if token_ids_a is None:
        return token_ids_a + sep + cls
       return token_ids_a + sep + token_ids_a + sep + cls






      def 					A__   (			self      :		str  ,   UpperCAmelCase      :		List[int]  ,   UpperCAmelCase      :		Optional[List[int]] = None		)	->	List[int]:


       '''simple docstring'''

       lowercase       : Optional[int]						=[self.sep_token_id]
       lowercase       : Union[str, Any]						=[2]

       if token_ids_a is None:
        return len(token_ids_a + sep		) * [0] + cls_segment_id
       return len(token_ids_a + sep		) * [0] + len(token_ids_a + sep		) * [1] + cls_segment_id






      def 					A__   (			self      :		str  ,   UpperCAmelCase      :		str  ,   UpperCAmelCase      :		Optional[str] = None		)	->	Tuple[str]:


       '''simple docstring'''

       if not self.can_save_slow_tokenizer:
        raise ValueError(
            '''Your fast tokenizer does not have the necessary information to save the vocabulary for a slow '''
            '''tokenizer.'''		)

       if not os.path.isdir(UpperCAmelCase		):
        logger.error(f'Vocabulary path ({save_directory}) should be a directory'		)
        return
       lowercase       : Dict						=os.path.join(
           UpperCAmelCase  ,   (filename_prefix + '''-''' if filename_prefix else '''''') + VOCAB_FILES_NAMES['''vocab_file''']		)

       if os.path.abspath(self.vocab_file		) != os.path.abspath(UpperCAmelCase		):
        copyfile(self.vocab_file  ,   UpperCAmelCase		)

       return (out_vocab_file,)



