






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


__snake_case			:			Optional[int]			   =   get_tests_dir("""fixtures/spiece.model""")


@require_sentencepiece
@require_tokenizers
class 			__SCREAMING_SNAKE_CASE    (      __lowercase						,    unittest.TestCase):
						_SCREAMING_SNAKE_CASE		:				Union[str, Any]            =    ErnieMTokenizer
						_SCREAMING_SNAKE_CASE		:				Dict            =    False
						_SCREAMING_SNAKE_CASE		:				str            =    True
						_SCREAMING_SNAKE_CASE		:				str            =    False
						_SCREAMING_SNAKE_CASE		:				Optional[int]            =    False






						def 					UpperCamelCase__					(		self      ):






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

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






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  'this is a test'
												lowerCAmelCase__			  =  'this is a test'
												return input_text, output_text






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  '<pad>'
												lowerCAmelCase__			  =  0

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






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  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_00_00      )






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												self.assertEqual(self.get_tokenizer().vocab_size   ,  3_00_00      )






						def 					UpperCamelCase__					(		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 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  ErnieMTokenizer(_UpperCamelCase   ,  do_lower_case=_UpperCamelCase   ,  unk_token='<unk>'   ,  pad_token='<pad>'      )

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

												self.assertListEqual(tokenizer.convert_tokens_to_ids(_UpperCamelCase      )   ,  [48, 25, 21, 12_89]      )

												lowerCAmelCase__			  =  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', 'é', '.']      )
												lowerCAmelCase__			  =  tokenizer.convert_tokens_to_ids(_UpperCamelCase      )
												self.assertListEqual(_UpperCamelCase   ,  [31, 23, 3_86, 19, 5_18, 30_50, 15, 17, 48, 25, 82_56, 18, 1, 9]      )

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






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  ErnieMTokenizer(_UpperCamelCase   ,  unk_token='<unk>'   ,  pad_token='<pad>'      )

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

												lowerCAmelCase__			  =  tokenizer.build_inputs_with_special_tokens(_UpperCamelCase      )
												lowerCAmelCase__			  =  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 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												# fmt: off
												lowerCAmelCase__			  =  {'input_ids': [[0, 1_10_62, 8_27_72, 7, 15, 8_27_72, 5_38, 5_15_29, 2_37, 1_71_98, 12_90, 2_06, 9, 21_51_75, 13_14, 1_36, 1_71_98, 12_90, 2_06, 9, 5_63_59, 42, 12_20_09, 9, 1_64_66, 16, 8_73_44, 45_37, 9, 47_17, 7_83_81, 6, 15_99_58, 7, 15, 2_44_80, 6_18, 4, 5_27, 2_26_93, 9, 3_04, 4, 27_77, 2_44_80, 98_74, 4, 4_35_23, 5_94, 4, 8_03, 1_83_92, 3_31_89, 18, 4, 4_35_23, 2_44_47, 5, 5, 5, 16, 1_00, 2_49_55, 8_36_58, 96_26, 14_40_57, 15, 8_39, 2_23_35, 16, 1_36, 2_49_55, 8_36_58, 8_34_79, 15, 3_91_02, 7_24, 16, 6_78, 6_45, 64_60, 13_28, 45_89, 42, 12_20_09, 11_57_74, 23, 35_59, 13_28, 4_68_76, 7, 1_36, 5_38_94, 19_40, 4_22_27, 4_11_59, 1_77_21, 8_23, 4_25, 4, 2_75_12, 9_87_22, 2_06, 1_36, 55_31, 49_70, 9_19, 1_73_36, 5, 2], [0, 2_00_80, 6_18, 83, 8_27_75, 47, 4_79, 9, 15_17, 73, 5_38_94, 3_33, 8_05_81, 11_01_17, 1_88_11, 52_56, 12_95, 51, 15_25_26, 2_97, 79_86, 3_90, 12_44_16, 5_38, 3_54_31, 2_14, 98, 1_50_44, 2_57_37, 1_36, 71_08, 4_37_01, 23, 7_56, 13_53_55, 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_81, 6_37_73, 11_94_55, 6, 14_77_97, 8_82_03, 7, 6_45, 70, 21, 32_85, 1_02_69, 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 csv
import dataclasses
import json
from dataclasses import dataclass
from typing import List, Optional, Union

from ...utils import is_tf_available, is_torch_available, logging


__snake_case			:			Union[str, Any]			   =   logging.get_logger(__name__)


@dataclass
class 			__SCREAMING_SNAKE_CASE    :
						_SCREAMING_SNAKE_CASE		:				str
						_SCREAMING_SNAKE_CASE		:				str
						_SCREAMING_SNAKE_CASE		:				Optional[str]            =    None
						_SCREAMING_SNAKE_CASE		:				Optional[str]            =    None






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												return json.dumps(dataclasses.asdict(self      )   ,  indent=2      ) + "\n"




@dataclass(frozen=__lowercase)
class 			__SCREAMING_SNAKE_CASE    :
						_SCREAMING_SNAKE_CASE		:				List[int]
						_SCREAMING_SNAKE_CASE		:				Optional[List[int]]            =    None
						_SCREAMING_SNAKE_CASE		:				Optional[List[int]]            =    None
						_SCREAMING_SNAKE_CASE		:				Optional[Union[int, float]]            =    None






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												return json.dumps(dataclasses.asdict(self      )      ) + "\n"




class 			__SCREAMING_SNAKE_CASE    :






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												raise NotImplementedError()






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												raise NotImplementedError()






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												raise NotImplementedError()






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												raise NotImplementedError()






						def 					UpperCamelCase__					(		self      ):






												"""simple docstring"""
												raise NotImplementedError()






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												if len(self.get_labels()      ) > 1:
																		lowerCAmelCase__			  =  self.get_labels()[int(example.label      )]
												return example







						@classmethod
						def 					UpperCamelCase__					(		cls   ,  _UpperCamelCase   ,  _UpperCamelCase=None      ):






												"""simple docstring"""
												with open(_UpperCamelCase   ,  'r'   ,  encoding='utf-8-sig'      ) as f:
																		return list(csv.reader(_UpperCamelCase   ,  delimiter='\t'   ,  quotechar=_UpperCamelCase      )      )




class 			__SCREAMING_SNAKE_CASE    (      __lowercase):






						def __init__(		self   ,  _UpperCamelCase=None   ,  _UpperCamelCase=None   ,  _UpperCamelCase="classification"   ,  _UpperCamelCase=False      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  [] if labels is None else labels
												lowerCAmelCase__			  =  [] if examples is None else examples
												lowerCAmelCase__			  =  mode
												lowerCAmelCase__			  =  verbose






						def __len__(		self      ):






												"""simple docstring"""
												return len(self.examples      )






						def __getitem__(		self   ,  _UpperCamelCase      ):






												"""simple docstring"""
												if isinstance(_UpperCamelCase   ,  _UpperCamelCase      ):
																		return SingleSentenceClassificationProcessor(labels=self.labels   ,  examples=self.examples[idx]      )
												return self.examples[idx]






						@classmethod
						def 					UpperCamelCase__					(		cls   ,  _UpperCamelCase   ,  _UpperCamelCase=""   ,  _UpperCamelCase=0   ,  _UpperCamelCase=1   ,  _UpperCamelCase=None   ,  _UpperCamelCase=False   ,  **_UpperCamelCase      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  cls(**_UpperCamelCase      )
												processor.add_examples_from_csv(
												    _UpperCamelCase   ,  split_name=_UpperCamelCase   ,  column_label=_UpperCamelCase   ,  column_text=_UpperCamelCase   ,  column_id=_UpperCamelCase   ,  skip_first_row=_UpperCamelCase   ,  overwrite_labels=_UpperCamelCase   ,  overwrite_examples=_UpperCamelCase   ,  )
												return processor






						@classmethod
						def 					UpperCamelCase__					(		cls   ,  _UpperCamelCase   ,  _UpperCamelCase=None   ,  **_UpperCamelCase      ):






												"""simple docstring"""
												lowerCAmelCase__			  =  cls(**_UpperCamelCase      )
												processor.add_examples(_UpperCamelCase   ,  labels=_UpperCamelCase      )
												return processor






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase   ,  _UpperCamelCase=""   ,  _UpperCamelCase=0   ,  _UpperCamelCase=1   ,  _UpperCamelCase=None   ,  _UpperCamelCase=False   ,  _UpperCamelCase=False   ,  _UpperCamelCase=False   ,  ):






												"""simple docstring"""
												lowerCAmelCase__			  =  self._read_tsv(_UpperCamelCase      )
												if skip_first_row:
																		lowerCAmelCase__			  =  lines[1:]
												lowerCAmelCase__			  =  []
												lowerCAmelCase__			  =  []
												lowerCAmelCase__			  =  []
												for i, line in enumerate(_UpperCamelCase      ):
																		texts.append(line[column_text]      )
																		labels.append(line[column_label]      )
																		if column_id is not None:
																								ids.append(line[column_id]      )
																		else:
																								lowerCAmelCase__			  =  F"{split_name}-{i}" if split_name else str(_UpperCamelCase      )
																								ids.append(_UpperCamelCase      )

												return self.add_examples(
												    _UpperCamelCase   ,  _UpperCamelCase   ,  _UpperCamelCase   ,  overwrite_labels=_UpperCamelCase   ,  overwrite_examples=_UpperCamelCase      )






						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase   ,  _UpperCamelCase=None   ,  _UpperCamelCase=None   ,  _UpperCamelCase=False   ,  _UpperCamelCase=False      ):






												"""simple docstring"""
												if labels is not None and len(_UpperCamelCase      ) != len(_UpperCamelCase      ):
																		raise ValueError(
																		    F"Text and labels have mismatched lengths {len(_UpperCamelCase      )} and {len(_UpperCamelCase      )}"      )
												if ids is not None and len(_UpperCamelCase      ) != len(_UpperCamelCase      ):
																		raise ValueError(F"Text and ids have mismatched lengths {len(_UpperCamelCase      )} and {len(_UpperCamelCase      )}"      )
												if ids is None:
																		lowerCAmelCase__			  =  [None] * len(_UpperCamelCase      )
												if labels is None:
																		lowerCAmelCase__			  =  [None] * len(_UpperCamelCase      )
												lowerCAmelCase__			  =  []
												lowerCAmelCase__			  =  set()
												for text_or_text_and_label, label, guid in zip(_UpperCamelCase   ,  _UpperCamelCase   ,  _UpperCamelCase      ):
																		if isinstance(_UpperCamelCase   ,  (tuple, list)      ) and label is None:
																								lowerCAmelCase__  ,				lowerCAmelCase__			  =  text_or_text_and_label
																		else:
																								lowerCAmelCase__			  =  text_or_text_and_label
																		added_labels.add(_UpperCamelCase      )
																		examples.append(InputExample(guid=_UpperCamelCase   ,  text_a=_UpperCamelCase   ,  text_b=_UpperCamelCase   ,  label=_UpperCamelCase      )      )

												# Update examples
												if overwrite_examples:
																		lowerCAmelCase__			  =  examples
												else:
																		self.examples.extend(_UpperCamelCase      )

												# Update labels
												if overwrite_labels:
																		lowerCAmelCase__			  =  list(_UpperCamelCase      )
												else:
																		lowerCAmelCase__			  =  list(set(self.labels      ).union(_UpperCamelCase      )      )

												return self.examples







						def 					UpperCamelCase__					(		self   ,  _UpperCamelCase   ,  _UpperCamelCase=None   ,  _UpperCamelCase=False   ,  _UpperCamelCase=0   ,  _UpperCamelCase=True   ,  _UpperCamelCase=None   ,  ):






												"""simple docstring"""
												if max_length is None:
																		lowerCAmelCase__			  =  tokenizer.max_len

												lowerCAmelCase__			  =  {label: i for i, label in enumerate(self.labels      )}

												lowerCAmelCase__			  =  []
												for ex_index, example in enumerate(self.examples      ):
																		if ex_index % 1_00_00 == 0:
																								logger.info(F"Tokenizing example {ex_index}"      )

																		lowerCAmelCase__			  =  tokenizer.encode(
																		    example.text_a   ,  add_special_tokens=_UpperCamelCase   ,  max_length=min(_UpperCamelCase   ,  tokenizer.max_len      )   ,  )
																		all_input_ids.append(_UpperCamelCase      )

												lowerCAmelCase__			  =  max(len(_UpperCamelCase      ) for input_ids in all_input_ids      )

												lowerCAmelCase__			  =  []
												for ex_index, (input_ids, example) in enumerate(zip(_UpperCamelCase   ,  self.examples      )      ):
																		if ex_index % 1_00_00 == 0:
																								logger.info(F"Writing example {ex_index}/{len(self.examples      )}"      )
																		# The mask has 1 for real tokens and 0 for padding tokens. Only real
																		# tokens are attended to.
																		lowerCAmelCase__			  =  [1 if mask_padding_with_zero else 0] * len(_UpperCamelCase      )

																		# Zero-pad up to the sequence length.
																		lowerCAmelCase__			  =  batch_length - len(_UpperCamelCase      )
																		if pad_on_left:
																								lowerCAmelCase__			  =  ([pad_token] * padding_length) + input_ids
																								lowerCAmelCase__			  =  ([0 if mask_padding_with_zero else 1] * padding_length) + attention_mask
																		else:
																								lowerCAmelCase__			  =  input_ids + ([pad_token] * padding_length)
																								lowerCAmelCase__			  =  attention_mask + ([0 if mask_padding_with_zero else 1] * padding_length)

																		if len(_UpperCamelCase      ) != batch_length:
																								raise ValueError(F"Error with input length {len(_UpperCamelCase      )} vs {batch_length}"      )
																		if len(_UpperCamelCase      ) != batch_length:
																								raise ValueError(F"Error with input length {len(_UpperCamelCase      )} vs {batch_length}"      )

																		if self.mode == "classification":
																								lowerCAmelCase__			  =  label_map[example.label]
																		elif self.mode == "regression":
																								lowerCAmelCase__			  =  float(example.label      )
																		else:
																								raise ValueError(self.mode      )

																		if ex_index < 5 and self.verbose:
																								logger.info('*** Example ***'      )
																								logger.info(F"guid: {example.guid}"      )
																								logger.info(F"input_ids: {' '.join([str(_UpperCamelCase      ) for x in input_ids]      )}"      )
																								logger.info(F"attention_mask: {' '.join([str(_UpperCamelCase      ) for x in attention_mask]      )}"      )
																								logger.info(F"label: {example.label} (id = {label})"      )

																		features.append(InputFeatures(input_ids=_UpperCamelCase   ,  attention_mask=_UpperCamelCase   ,  label=_UpperCamelCase      )      )

												if return_tensors is None:
																		return features
												elif return_tensors == "tf":
																		if not is_tf_available():
																								raise RuntimeError('return_tensors set to \'tf\' but TensorFlow 2.0 can\'t be imported'      )
																		import tensorflow as tf

																		def gen():
																								for ex in features:
																														yield ({"input_ids": ex.input_ids, "attention_mask": ex.attention_mask}, ex.label)

																		lowerCAmelCase__			  =  tf.data.Dataset.from_generator(
																		    _UpperCamelCase   ,  ({'input_ids': tf.intaa, 'attention_mask': tf.intaa}, tf.intaa)   ,  ({'input_ids': tf.TensorShape([None]      ), 'attention_mask': tf.TensorShape([None]      )}, tf.TensorShape([]      ))   ,  )
																		return dataset
												elif return_tensors == "pt":
																		if not is_torch_available():
																								raise RuntimeError('return_tensors set to \'pt\' but PyTorch can\'t be imported'      )
																		import torch
																		from torch.utils.data import TensorDataset

																		lowerCAmelCase__			  =  torch.tensor([f.input_ids for f in features]   ,  dtype=torch.long      )
																		lowerCAmelCase__			  =  torch.tensor([f.attention_mask for f in features]   ,  dtype=torch.long      )
																		if self.mode == "classification":
																								lowerCAmelCase__			  =  torch.tensor([f.label for f in features]   ,  dtype=torch.long      )
																		elif self.mode == "regression":
																								lowerCAmelCase__			  =  torch.tensor([f.label for f in features]   ,  dtype=torch.float      )

																		lowerCAmelCase__			  =  TensorDataset(_UpperCamelCase   ,  _UpperCamelCase   ,  _UpperCamelCase      )
																		return dataset
												else:
																		raise ValueError('return_tensors should be one of \'tf\' or \'pt\''      )





