

import unittest

from transformers import ErnieMTokenizer
from transformers.testing_utils import get_tests_dir, require_sentencepiece, require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin


lowerCAmelCase_	     =      get_tests_dir('''fixtures/spiece.model''')





@require_sentencepiece
@require_tokenizers
class      snake_case_       (		lowercase_     ,	unittest.TestCase ):







			'''simple docstring'''

			SCREAMING_SNAKE_CASE					:       Tuple						      =    ErnieMTokenizer
			SCREAMING_SNAKE_CASE					:       Tuple						      =    False
			SCREAMING_SNAKE_CASE					:       int						      =    True
			SCREAMING_SNAKE_CASE					:       List[Any]						      =    False
			SCREAMING_SNAKE_CASE					:       List[Any]						      =    False







			def     snake_case__(					self			:    Optional[Any]							)				->Union[str, Any]:
									super().setUp()

									# We have a SentencePiece fixture for testing
									snake_case_       =     ErnieMTokenizer(UpperCamelCase_							,       unk_token='''<unk>'''							,       pad_token='''<pad>'''							)
									tokenizer.save_pretrained(self.tmpdirname							)







			def     snake_case__(					self			:    Optional[Any]							,       _UpperCamelCase			:    Tuple							)				->List[str]:
									snake_case_       =     '''this is a test'''
									snake_case_       =     '''this is a test'''
									return input_text, output_text







			def     snake_case__(					self			:    Optional[int]							)				->List[str]:
									snake_case_       =     '''<pad>'''
									snake_case_       =     0

									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			:    Tuple							)				->Any:
									snake_case_       =     list(self.get_tokenizer().get_vocab().keys()							)

									self.assertEqual(vocab_keys[0]							,       '''<pad>'''							)
									self.assertEqual(vocab_keys[1]							,       '''<unk>'''							)
									self.assertEqual(vocab_keys[-1]							,       '''▁eloquent'''							)
									self.assertEqual(len(UpperCamelCase_							)							,       3_0_0_0_0							)







			def     snake_case__(					self			:    Tuple							)				->str:
									self.assertEqual(self.get_tokenizer().vocab_size							,       3_0_0_0_0							)







			def     snake_case__(					self			:    Union[str, Any]							)				->Tuple:



									if not self.test_rust_tokenizer:
															return

									snake_case_       =     self.get_tokenizer()
									snake_case_       =     self.get_rust_tokenizer()

									snake_case_       =     '''I was born in 92000, and this is falsé.'''

									snake_case_       =     tokenizer.tokenize(UpperCamelCase_							)
									snake_case_       =     rust_tokenizer.tokenize(UpperCamelCase_							)
									self.assertListEqual(UpperCamelCase_							,       UpperCamelCase_							)

									snake_case_       =     tokenizer.encode(UpperCamelCase_							,       add_special_tokens=UpperCamelCase_							)
									snake_case_       =     rust_tokenizer.encode(UpperCamelCase_							,       add_special_tokens=UpperCamelCase_							)
									self.assertListEqual(UpperCamelCase_							,       UpperCamelCase_							)

									snake_case_       =     self.get_rust_tokenizer()
									snake_case_       =     tokenizer.encode(UpperCamelCase_							)
									snake_case_       =     rust_tokenizer.encode(UpperCamelCase_							)
									self.assertListEqual(UpperCamelCase_							,       UpperCamelCase_							)







			def     snake_case__(					self			:    Union[str, Any]							)				->List[Any]:
									snake_case_       =     ErnieMTokenizer(UpperCamelCase_							,       do_lower_case=UpperCamelCase_							,       unk_token='''<unk>'''							,       pad_token='''<pad>'''							)

									snake_case_       =     tokenizer.tokenize('''This is a test'''							)
									self.assertListEqual(UpperCamelCase_							,       ['''▁this''', '''▁is''', '''▁a''', '''▁test''']							)

									self.assertListEqual(tokenizer.convert_tokens_to_ids(UpperCamelCase_							)							,       [4_8, 2_5, 2_1, 1_2_8_9]							)

									snake_case_       =     tokenizer.tokenize('''I was born in 92000, and this is falsé.'''							)
									# ErnieMTokenizer(paddlenlp implementation) outputs '9' instead of '_9' so to mimic that '_9' is changed to '9'
									self.assertListEqual(
									    UpperCamelCase_							,       ['''▁i''', '''▁was''', '''▁born''', '''▁in''', '''9''', '''2000''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁fal''', '''s''', '''é''', '''.''']							)
									snake_case_       =     tokenizer.convert_tokens_to_ids(UpperCamelCase_							)
									self.assertListEqual(UpperCamelCase_							,       [3_1, 2_3, 3_8_6, 1_9, 5_1_8, 3_0_5_0, 1_5, 1_7, 4_8, 2_5, 8_2_5_6, 1_8, 1, 9]							)

									snake_case_       =     tokenizer.convert_ids_to_tokens(UpperCamelCase_							)
									self.assertListEqual(
									    UpperCamelCase_							,       ['''▁i''', '''▁was''', '''▁born''', '''▁in''', '''9''', '''2000''', ''',''', '''▁and''', '''▁this''', '''▁is''', '''▁fal''', '''s''', '''<unk>''', '''.''']							,       )







			def     snake_case__(					self			:    List[str]							)				->List[Any]:
									snake_case_       =     ErnieMTokenizer(UpperCamelCase_							,       unk_token='''<unk>'''							,       pad_token='''<pad>'''							)

									snake_case_       =     tokenizer.encode('''sequence builders'''							)
									snake_case_       =     tokenizer.encode('''multi-sequence build'''							)

									snake_case_       =     tokenizer.build_inputs_with_special_tokens(UpperCamelCase_							)
									snake_case_       =     tokenizer.build_inputs_with_special_tokens(UpperCamelCase_							,       UpperCamelCase_							)

									assert encoded_sentence == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id]
									assert encoded_pair == [tokenizer.cls_token_id] + text + [tokenizer.sep_token_id] + [
									    tokenizer.sep_token_id
									] + text_a + [tokenizer.sep_token_id]







			@slow
			def     snake_case__(					self			:    List[str]							)				->int:
									snake_case_       =     {'''input_ids''': [[0, 1_1_0_6_2, 8_2_7_7_2, 7, 1_5, 8_2_7_7_2, 5_3_8, 5_1_5_2_9, 2_3_7, 1_7_1_9_8, 1_2_9_0, 2_0_6, 9, 2_1_5_1_7_5, 1_3_1_4, 1_3_6, 1_7_1_9_8, 1_2_9_0, 2_0_6, 9, 5_6_3_5_9, 4_2, 1_2_2_0_0_9, 9, 1_6_4_6_6, 1_6, 8_7_3_4_4, 4_5_3_7, 9, 4_7_1_7, 7_8_3_8_1, 6, 1_5_9_9_5_8, 7, 1_5, 2_4_4_8_0, 6_1_8, 4, 5_2_7, 2_2_6_9_3, 9, 3_0_4, 4, 2_7_7_7, 2_4_4_8_0, 9_8_7_4, 4, 4_3_5_2_3, 5_9_4, 4, 8_0_3, 1_8_3_9_2, 3_3_1_8_9, 1_8, 4, 4_3_5_2_3, 2_4_4_4_7, 5, 5, 5, 1_6, 1_0_0, 2_4_9_5_5, 8_3_6_5_8, 9_6_2_6, 1_4_4_0_5_7, 1_5, 8_3_9, 2_2_3_3_5, 1_6, 1_3_6, 2_4_9_5_5, 8_3_6_5_8, 8_3_4_7_9, 1_5, 3_9_1_0_2, 7_2_4, 1_6, 6_7_8, 6_4_5, 6_4_6_0, 1_3_2_8, 4_5_8_9, 4_2, 1_2_2_0_0_9, 1_1_5_7_7_4, 2_3, 3_5_5_9, 1_3_2_8, 4_6_8_7_6, 7, 1_3_6, 5_3_8_9_4, 1_9_4_0, 4_2_2_2_7, 4_1_1_5_9, 1_7_7_2_1, 8_2_3, 4_2_5, 4, 2_7_5_1_2, 9_8_7_2_2, 2_0_6, 1_3_6, 5_5_3_1, 4_9_7_0, 9_1_9, 1_7_3_3_6, 5, 2], [0, 2_0_0_8_0, 6_1_8, 8_3, 8_2_7_7_5, 4_7, 4_7_9, 9, 1_5_1_7, 7_3, 5_3_8_9_4, 3_3_3, 8_0_5_8_1, 1_1_0_1_1_7, 1_8_8_1_1, 5_2_5_6, 1_2_9_5, 5_1, 1_5_2_5_2_6, 2_9_7, 7_9_8_6, 3_9_0, 1_2_4_4_1_6, 5_3_8, 3_5_4_3_1, 2_1_4, 9_8, 1_5_0_4_4, 2_5_7_3_7, 1_3_6, 7_1_0_8, 4_3_7_0_1, 2_3, 7_5_6, 1_3_5_3_5_5, 7, 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], [0, 5_8_1, 6_3_7_7_3, 1_1_9_4_5_5, 6, 1_4_7_7_9_7, 8_8_2_0_3, 7, 6_4_5, 7_0, 2_1, 3_2_8_5, 1_0_2_6_9, 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]]}
									# fmt: on

									self.tokenizer_integration_test_util(
									    expected_encoding=UpperCamelCase_							,       model_name='''susnato/ernie-m-base_pytorch'''							,       sequences=[
									        '''Transformers (formerly known as pytorch-transformers and pytorch-pretrained-bert) provides '''
									        '''general-purpose architectures (BERT, GPT-2, RoBERTa, XLM, DistilBert, XLNet...) for Natural '''
									        '''Language Understanding (NLU) and Natural Language Generation (NLG) with over32+ pretrained '''
									        '''models in100+ languages and deep interoperability between Jax, PyTorch and TensorFlow.''',
									        '''BERT is designed to pre-train deep bidirectional representations from unlabeled text by jointly '''
									        '''conditioning on both left and right context in all layers.''',
									        '''The quick brown fox jumps over the lazy dog.''',
									    ]							,       )


import json
from typing import List, Optional, Tuple

from tokenizers import normalizers

from ...tokenization_utils_fast import PreTrainedTokenizerFast
from ...utils import logging
from .tokenization_splinter import SplinterTokenizer


lowerCAmelCase_	     =     logging.get_logger(__name__)

lowerCAmelCase_	     =     {'''vocab_file''': '''vocab.txt'''}

lowerCAmelCase_	     =     {
    '''vocab_file''': {
        '''tau/splinter-base''': '''https://huggingface.co/tau/splinter-base/resolve/main/vocab.txt''',
        '''tau/splinter-base-qass''': '''https://huggingface.co/tau/splinter-base-qass/resolve/main/vocab.txt''',
        '''tau/splinter-large''': '''https://huggingface.co/tau/splinter-large/resolve/main/vocab.txt''',
        '''tau/splinter-large-qass''': '''https://huggingface.co/tau/splinter-large-qass/resolve/main/vocab.txt''',
    }
}

lowerCAmelCase_	     =     {
    '''tau/splinter-base''': 5_12,
    '''tau/splinter-base-qass''': 5_12,
    '''tau/splinter-large''': 5_12,
    '''tau/splinter-large-qass''': 5_12,
}

lowerCAmelCase_	     =     {
    '''tau/splinter-base''': {'''do_lower_case''': False},
    '''tau/splinter-base-qass''': {'''do_lower_case''': False},
    '''tau/splinter-large''': {'''do_lower_case''': False},
    '''tau/splinter-large-qass''': {'''do_lower_case''': False},
}





class      snake_case_       (		__A ):







			'''simple docstring'''

			SCREAMING_SNAKE_CASE					:       List[str]						      =    VOCAB_FILES_NAMES
			SCREAMING_SNAKE_CASE					:       List[str]						      =    PRETRAINED_VOCAB_FILES_MAP
			SCREAMING_SNAKE_CASE					:       Optional[int]						      =    PRETRAINED_INIT_CONFIGURATION
			SCREAMING_SNAKE_CASE					:       List[str]						      =    PRETRAINED_POSITIONAL_EMBEDDINGS_SIZES
			SCREAMING_SNAKE_CASE					:       Any						      =    SplinterTokenizer







			def __init__(					self			:    List[str]							,       _UpperCamelCase			:    Dict=None							,       _UpperCamelCase			:    Union[str, Any]=None							,       _UpperCamelCase			:    Union[str, Any]=True							,       _UpperCamelCase			:    List[Any]="[UNK]"							,       _UpperCamelCase			:    List[str]="[SEP]"							,       _UpperCamelCase			:    Optional[int]="[PAD]"							,       _UpperCamelCase			:    Tuple="[CLS]"							,       _UpperCamelCase			:    Optional[int]="[MASK]"							,       _UpperCamelCase			:    str="[QUESTION]"							,       _UpperCamelCase			:    Optional[Any]=True							,       _UpperCamelCase			:    Dict=None							,       **_UpperCamelCase			:    Optional[int]							,       )				->Union[str, Any]:
									super().__init__(
									    _UpperCamelCase							,       tokenizer_file=_UpperCamelCase							,       do_lower_case=_UpperCamelCase							,       unk_token=_UpperCamelCase							,       sep_token=_UpperCamelCase							,       pad_token=_UpperCamelCase							,       cls_token=_UpperCamelCase							,       mask_token=_UpperCamelCase							,       tokenize_chinese_chars=_UpperCamelCase							,       strip_accents=_UpperCamelCase							,       additional_special_tokens=(question_token,)							,       **_UpperCamelCase							,       )

									snake_case_       =     json.loads(self.backend_tokenizer.normalizer.__getstate__()							)
									if (
									    pre_tok_state.get('''lowercase'''							,       _UpperCamelCase							) != do_lower_case
									    or pre_tok_state.get('''strip_accents'''							,       _UpperCamelCase							) != strip_accents
									):
															snake_case_       =     getattr(_UpperCamelCase							,       pre_tok_state.pop('''type'''							)							)
															snake_case_       =     do_lower_case
															snake_case_       =     strip_accents
															snake_case_       =     pre_tok_class(**_UpperCamelCase							)

									snake_case_       =     do_lower_case







			@property
			def     snake_case__(					self			:    Optional[int]							)				->int:
									return self.convert_tokens_to_ids(self.question_token							)







			def     snake_case__(					self			:    Dict							,       _UpperCamelCase			:    List[int]							,       _UpperCamelCase			:    Optional[List[int]] = None							)				->List[int]:
									if token_ids_a is None:
															return [self.cls_token_id] + token_ids_a + [self.sep_token_id]

									snake_case_       =     [self.cls_token_id]
									snake_case_       =     [self.sep_token_id]
									snake_case_       =     [self.question_token_id] + [self.convert_tokens_to_ids('''.'''							)]
									if self.padding_side == "right":
															# Input is question-then-context
															return cls + token_ids_a + question_suffix + sep + token_ids_a + sep
									else:
															# Input is context-then-question
															return cls + token_ids_a + sep + token_ids_a + question_suffix + sep







			def     snake_case__(					self			:    str							,       _UpperCamelCase			:    List[int]							,       _UpperCamelCase			:    Optional[List[int]] = None							)				->List[int]:
									snake_case_       =     [self.sep_token_id]
									snake_case_       =     [self.cls_token_id]
									snake_case_       =     [self.question_token_id] + [self.convert_tokens_to_ids('''.'''							)]
									if token_ids_a is None:
															return len(cls + token_ids_a + sep							) * [0]

									if self.padding_side == "right":
															# Input is question-then-context
															return len(cls + token_ids_a + question_suffix + sep							) * [0] + len(token_ids_a + sep							) * [1]
									else:
															# Input is context-then-question
															return len(cls + token_ids_a + sep							) * [0] + len(token_ids_a + question_suffix + sep							) * [1]







			def     snake_case__(					self			:    Union[str, Any]							,       _UpperCamelCase			:    str							,       _UpperCamelCase			:    Optional[str] = None							)				->Tuple[str]:
									snake_case_       =     self._tokenizer.model.save(_UpperCamelCase							,       name=_UpperCamelCase							)
									return tuple(_UpperCamelCase							)