
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  UpperCamelCase__		(    SCREAMING_SNAKE_CASE_,			unittest.TestCase   ):







	'''simple docstring'''
	_lowerCAmelCase							       =				RoCBertTokenizer
	_lowerCAmelCase							       =				None
	_lowerCAmelCase							       =				False
	_lowerCAmelCase							       =				True
	_lowerCAmelCase							       =				filter_non_english





	def   __snake_case				(			self		):
					super().setUp()

					A__  :	Optional[Any]         =							['''[UNK]''', '''[CLS]''', '''[SEP]''', '''[PAD]''', '''[MASK]''', '''你''', '''好''', '''是''', '''谁''', '''a''', '''b''', '''c''', '''d''']
					A__  :	int         =							{}
					A__  :	Dict         =							{}
					for i, value in enumerate(UpperCamelCase__		):
									A__  :	int         =							i
									A__  :	Any         =							i
					A__  :	Tuple         =							os.path.join(self.tmpdirname				,       VOCAB_FILES_NAMES['''vocab_file''']		)
					A__  :	str         =							os.path.join(self.tmpdirname				,       VOCAB_FILES_NAMES['''word_shape_file''']		)
					A__  :	Dict         =							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   __snake_case				(			self		):
					A__  :	Optional[Any]         =							self.tokenizer_class(self.vocab_file				,       self.word_shape_file				,       self.word_pronunciation_file		)

					A__  :	Tuple         =							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   __snake_case				(			self		):
					A__  :	Dict         =							RoCBertBasicTokenizer()

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





	def   __snake_case				(			self		):
					A__  :	List[str]         =							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   __snake_case				(			self		):
					A__  :	str         =							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   __snake_case				(			self		):
					A__  :	str         =							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   __snake_case				(			self		):
					A__  :	int         =							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   __snake_case				(			self		):
					A__  :	List[Any]         =							RoCBertBasicTokenizer(do_lower_case=UpperCamelCase__		)

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





	def   __snake_case				(			self		):
					A__  :	Any         =							RoCBertBasicTokenizer(do_lower_case=UpperCamelCase__				,       strip_accents=UpperCamelCase__		)

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





	def   __snake_case				(			self		):
					A__  :	Optional[Any]         =							RoCBertBasicTokenizer(do_lower_case=UpperCamelCase__				,       strip_accents=UpperCamelCase__		)

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





	def   __snake_case				(			self		):
					A__  :	Dict         =							RoCBertBasicTokenizer(do_lower_case=UpperCamelCase__				,       never_split=['''[UNK]''']		)

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





	def   __snake_case				(			self		):
					A__  :	str         =							['''[UNK]''', '''[CLS]''', '''[SEP]''', '''want''', '''##want''', '''##ed''', '''wa''', '''un''', '''runn''', '''##ing''']

					A__  :	Union[str, Any]         =							{}
					for i, token in enumerate(UpperCamelCase__		):
									A__  :	List[str]         =							i
					A__  :	Dict         =							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   __snake_case				(			self		):
					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   __snake_case				(			self		):
					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   __snake_case				(			self		):
					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   __snake_case				(			self		):
					A__  :	Tuple         =							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:
									A__  :	Optional[int]         =							self.get_rust_tokenizer()
									self.assertListEqual(
									    [rust_tokenizer.tokenize(UpperCamelCase__		) for t in ['''Test''', '''\xad''', '''test''']]				,       [['''[UNK]'''], [], ['''[UNK]''']]		)





	def   __snake_case				(			self		):
					for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
									with self.subTest(F"{tokenizer.__class__.__name__} ({pretrained_name})"		):
													A__  :	Optional[int]         =							self.rust_tokenizer_class.from_pretrained(UpperCamelCase__				,       **UpperCamelCase__		)

													A__  :	Optional[int]         =							F"A, naïve {tokenizer_r.mask_token} AllenNLP sentence."
													A__  :	Dict         =							tokenizer_r.encode_plus(
													    UpperCamelCase__				,       return_attention_mask=UpperCamelCase__				,       return_token_type_ids=UpperCamelCase__				,       return_offsets_mapping=UpperCamelCase__				,       add_special_tokens=UpperCamelCase__				,       )

													A__  :	Tuple         =							tokenizer_r.do_lower_case if hasattr(UpperCamelCase__				,       '''do_lower_case'''		) else False
													A__  :	Optional[Any]         =							(
													    [
													        ((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   __snake_case				(			self		):
					A__  :	int         =							['''的''', '''人''', '''有''']
					A__  :	Tuple         =							''''''.join(UpperCamelCase__		)
					for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
									with self.subTest(F"{tokenizer.__class__.__name__} ({pretrained_name})"		):
													A__  :	List[str]         =							True
													A__  :	List[str]         =							self.tokenizer_class.from_pretrained(UpperCamelCase__				,       **UpperCamelCase__		)
													A__  :	Any         =							self.rust_tokenizer_class.from_pretrained(UpperCamelCase__				,       **UpperCamelCase__		)

													A__  :	List[str]         =							tokenizer_p.encode(UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)
													A__  :	str         =							tokenizer_r.encode(UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)

													A__  :	Union[str, Any]         =							tokenizer_r.convert_ids_to_tokens(UpperCamelCase__		)
													A__  :	int         =							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__		)

													A__  :	Optional[int]         =							False
													A__  :	Optional[Any]         =							self.rust_tokenizer_class.from_pretrained(UpperCamelCase__				,       **UpperCamelCase__		)
													A__  :	Tuple         =							self.tokenizer_class.from_pretrained(UpperCamelCase__				,       **UpperCamelCase__		)

													A__  :	Any         =							tokenizer_r.encode(UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)
													A__  :	Dict         =							tokenizer_p.encode(UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)

													A__  :	Dict         =							tokenizer_r.convert_ids_to_tokens(UpperCamelCase__		)
													A__  :	str         =							tokenizer_p.convert_ids_to_tokens(UpperCamelCase__		)

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





	@slow
	def   __snake_case				(			self		):
					A__  :	int         =							self.tokenizer_class(self.vocab_file				,       self.word_shape_file				,       self.word_pronunciation_file		)

					A__  :	str         =							tokenizer.encode('''你好'''				,       add_special_tokens=UpperCamelCase__		)
					A__  :	List[str]         =							tokenizer.encode('''你是谁'''				,       add_special_tokens=UpperCamelCase__		)

					A__  :	Dict         =							tokenizer.build_inputs_with_special_tokens(UpperCamelCase__		)
					A__  :	Optional[int]         =							tokenizer.build_inputs_with_special_tokens(UpperCamelCase__				,       UpperCamelCase__		)

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





	def   __snake_case				(			self		):
					A__  :	Tuple         =							self.get_tokenizers(do_lower_case=UpperCamelCase__		)
					for tokenizer in tokenizers:
									with self.subTest(F"{tokenizer.__class__.__name__}"		):
													A__  :	Union[str, Any]         =							'''你好，你是谁'''
													A__  :	Optional[int]         =							tokenizer.tokenize(UpperCamelCase__		)
													A__  :	Optional[Any]         =							tokenizer.convert_tokens_to_ids(UpperCamelCase__		)
													A__  :	int         =							tokenizer.convert_tokens_to_shape_ids(UpperCamelCase__		)
													A__  :	Any         =							tokenizer.convert_tokens_to_pronunciation_ids(UpperCamelCase__		)
													A__  :	Dict         =							tokenizer.prepare_for_model(
													    UpperCamelCase__				,       UpperCamelCase__				,       UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)

													A__  :	Any         =							tokenizer.encode_plus(UpperCamelCase__				,       add_special_tokens=UpperCamelCase__		)

													self.assertEqual(UpperCamelCase__				,       UpperCamelCase__		)

from ...utils import (
    OptionalDependencyNotAvailable,
    is_torch_available,
    is_transformers_available,
    is_transformers_version,
)


try:
					if not (is_transformers_available() and is_torch_available()):
										raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
					from ...utils.dummy_torch_and_transformers_objects import KandinskyPipeline, KandinskyPriorPipeline
else:
					from .pipeline_kandinsky import KandinskyPipeline
					from .pipeline_kandinsky_imgaimg import KandinskyImgaImgPipeline
					from .pipeline_kandinsky_inpaint import KandinskyInpaintPipeline
					from .pipeline_kandinsky_prior import KandinskyPriorPipeline, KandinskyPriorPipelineOutput
					from .text_encoder import MultilingualCLIP