






import json
import os
import unittest

from transformers.models.roc_bert.tokenization_roc_bert import (
    VOCAB_FILES_NAMES,
    RoCBertBasicTokenizer,
    RoCBertTokenizer,
    RoCBertWordpieceTokenizer,
    _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       __SCREAMING_SNAKE_CASE   (				__lowercase						,   unittest.TestCase):
						_SCREAMING_SNAKE_CASE		:    List[str]						=   RoCBertTokenizer
						_SCREAMING_SNAKE_CASE		:    str						=   None
						_SCREAMING_SNAKE_CASE		:    Tuple						=   False
						_SCREAMING_SNAKE_CASE		:    List[Any]						=   True
						_SCREAMING_SNAKE_CASE		:    Union[str, Any]						=   filter_non_english






						def  UpperCamelCase__ (		self				):






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

												lowerCAmelCase__        =  ['[UNK]', '[CLS]', '[SEP]', '[PAD]', '[MASK]', '你', '好', '是', '谁', 'a', 'b', 'c', 'd']
												lowerCAmelCase__        =  {}
												lowerCAmelCase__        =  {}
												for i, value in enumerate(_UpperCamelCase				):
																		lowerCAmelCase__        =  i
																		lowerCAmelCase__        =  i
												lowerCAmelCase__        =  os.path.join(self.tmpdirname       ,  VOCAB_FILES_NAMES['vocab_file']				)
												lowerCAmelCase__        =  os.path.join(self.tmpdirname       ,  VOCAB_FILES_NAMES['word_shape_file']				)
												lowerCAmelCase__        =  os.path.join(self.tmpdirname       ,  VOCAB_FILES_NAMES['word_pronunciation_file']				)
												with open(self.vocab_file       ,  'w'       ,  encoding='utf-8'				) as vocab_writer:
																		vocab_writer.write(''.join([x + '\n' for x in vocab_tokens]				)				)
												with open(self.word_shape_file       ,  'w'       ,  encoding='utf-8'				) as word_shape_writer:
																		json.dump(_UpperCamelCase       ,  _UpperCamelCase       ,  ensure_ascii=_UpperCamelCase				)
												with open(self.word_pronunciation_file       ,  'w'       ,  encoding='utf-8'				) as word_pronunciation_writer:
																		json.dump(_UpperCamelCase       ,  _UpperCamelCase       ,  ensure_ascii=_UpperCamelCase				)






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  self.tokenizer_class(self.vocab_file       ,  self.word_shape_file       ,  self.word_pronunciation_file				)

												lowerCAmelCase__        =  tokenizer.tokenize('你好[SEP]你是谁'				)
												self.assertListEqual(_UpperCamelCase       ,  ['你', '好', '[SEP]', '你', '是', '谁']				)
												self.assertListEqual(tokenizer.convert_tokens_to_ids(_UpperCamelCase				)       ,  [5, 6, 2, 5, 7, 8]				)
												self.assertListEqual(tokenizer.convert_tokens_to_shape_ids(_UpperCamelCase				)       ,  [5, 6, 2, 5, 7, 8]				)
												self.assertListEqual(tokenizer.convert_tokens_to_pronunciation_ids(_UpperCamelCase				)       ,  [5, 6, 2, 5, 7, 8]				)






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer()

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






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(do_lower_case=_UpperCamelCase				)

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






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(do_lower_case=_UpperCamelCase       ,  strip_accents=_UpperCamelCase				)

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






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(do_lower_case=_UpperCamelCase       ,  strip_accents=_UpperCamelCase				)

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






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  RoCBertBasicTokenizer(do_lower_case=_UpperCamelCase       ,  never_split=['[UNK]']				)

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






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  ['[UNK]', '[CLS]', '[SEP]', 'want', '##want', '##ed', 'wa', 'un', 'runn', '##ing']

												lowerCAmelCase__        =  {}
												for i, token in enumerate(_UpperCamelCase				):
																		lowerCAmelCase__        =  i
												lowerCAmelCase__        =  RoCBertWordpieceTokenizer(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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  self.get_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]']]				)

												if self.test_rust_tokenizer:
																		lowerCAmelCase__        =  self.get_rust_tokenizer()
																		self.assertListEqual(
																		    [rust_tokenizer.tokenize(_UpperCamelCase				) for t in ['Test', '\xad', 'test']]       ,  [['[UNK]'], [], ['[UNK]']]				)






						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												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(_UpperCamelCase       ,  **_UpperCamelCase				)

																								lowerCAmelCase__        =  F"A, naïve {tokenizer_r.mask_token} AllenNLP sentence."
																								lowerCAmelCase__        =  tokenizer_r.encode_plus(
																								    _UpperCamelCase       ,  return_attention_mask=_UpperCamelCase       ,  return_token_type_ids=_UpperCamelCase       ,  return_offsets_mapping=_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase       ,  )

																								lowerCAmelCase__        =  tokenizer_r.do_lower_case if hasattr(_UpperCamelCase       ,  'do_lower_case'				) else False
																								lowerCAmelCase__        =  (
																								    [
																								        ((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  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  ['的', '人', '有']
												lowerCAmelCase__        =  ''.join(_UpperCamelCase				)
												for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
																		with self.subTest(F"{tokenizer.__class__.__name__} ({pretrained_name})"				):
																								lowerCAmelCase__        =  True
																								lowerCAmelCase__        =  self.tokenizer_class.from_pretrained(_UpperCamelCase       ,  **_UpperCamelCase				)
																								lowerCAmelCase__        =  self.rust_tokenizer_class.from_pretrained(_UpperCamelCase       ,  **_UpperCamelCase				)

																								lowerCAmelCase__        =  tokenizer_p.encode(_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer_r.encode(_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)

																								lowerCAmelCase__        =  tokenizer_r.convert_ids_to_tokens(_UpperCamelCase				)
																								lowerCAmelCase__        =  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				)

																								lowerCAmelCase__        =  False
																								lowerCAmelCase__        =  self.rust_tokenizer_class.from_pretrained(_UpperCamelCase       ,  **_UpperCamelCase				)
																								lowerCAmelCase__        =  self.tokenizer_class.from_pretrained(_UpperCamelCase       ,  **_UpperCamelCase				)

																								lowerCAmelCase__        =  tokenizer_r.encode(_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer_p.encode(_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)

																								lowerCAmelCase__        =  tokenizer_r.convert_ids_to_tokens(_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer_p.convert_ids_to_tokens(_UpperCamelCase				)

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






						@slow
						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  self.tokenizer_class(self.vocab_file       ,  self.word_shape_file       ,  self.word_pronunciation_file				)

												lowerCAmelCase__        =  tokenizer.encode('你好'       ,  add_special_tokens=_UpperCamelCase				)
												lowerCAmelCase__        =  tokenizer.encode('你是谁'       ,  add_special_tokens=_UpperCamelCase				)

												lowerCAmelCase__        =  tokenizer.build_inputs_with_special_tokens(_UpperCamelCase				)
												lowerCAmelCase__        =  tokenizer.build_inputs_with_special_tokens(_UpperCamelCase       ,  _UpperCamelCase				)

												assert encoded_sentence == [1] + text + [2]
												assert encoded_pair == [1] + text + [2] + text_a + [2]







						def  UpperCamelCase__ (		self				):






												"""simple docstring"""
												lowerCAmelCase__        =  self.get_tokenizers(do_lower_case=_UpperCamelCase				)
												for tokenizer in tokenizers:
																		with self.subTest(F"{tokenizer.__class__.__name__}"				):
																								lowerCAmelCase__        =  '你好，你是谁'
																								lowerCAmelCase__        =  tokenizer.tokenize(_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer.convert_tokens_to_ids(_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer.convert_tokens_to_shape_ids(_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer.convert_tokens_to_pronunciation_ids(_UpperCamelCase				)
																								lowerCAmelCase__        =  tokenizer.prepare_for_model(
																								    _UpperCamelCase       ,  _UpperCamelCase       ,  _UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)

																								lowerCAmelCase__        =  tokenizer.encode_plus(_UpperCamelCase       ,  add_special_tokens=_UpperCamelCase				)

																								self.assertEqual(_UpperCamelCase       ,  _UpperCamelCase				)













def 		_UpperCamelCase				(      UpperCamelCase_    :   str						)					->				str:

						"""simple docstring"""



						return " ".join(input_str.split()[::-1]						)


if __name__ == "__main__":
					import doctest

					doctest.testmod()





