




"""simple docstring"""







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__					(					_a						,				unittest.TestCase					):
						_a    						=						RoCBertTokenizer
						_a    						=						None
						_a    						=						False
						_a    						=						True
						_a    						=						filter_non_english


						def   __lowercase (							self				:			Optional[Any]					):
										super().setUp()

										lowerCAmelCase  =				["""[UNK]""", """[CLS]""", """[SEP]""", """[PAD]""", """[MASK]""", """你""", """好""", """是""", """谁""", """a""", """b""", """c""", """d"""]
										lowerCAmelCase  =				{}
										lowerCAmelCase  =				{}
										for i, value in enumerate(lowerCAmelCase					):
														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(lowerCAmelCase						,						lowerCAmelCase						,						ensure_ascii=lowerCAmelCase					)
										with open(self.word_pronunciation_file						,						"""w"""						,						encoding="""utf-8"""					) as word_pronunciation_writer:
														json.dump(lowerCAmelCase						,						lowerCAmelCase						,						ensure_ascii=lowerCAmelCase					)


						def   __lowercase (							self				:			List[str]					):
										lowerCAmelCase  =				self.tokenizer_class(self.vocab_file						,						self.word_shape_file						,						self.word_pronunciation_file					)

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


						def   __lowercase (							self				:			int					):
										lowerCAmelCase  =				RoCBertBasicTokenizer()

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


						def   __lowercase (							self				:			List[str]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase					)

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHeLLo!how  \n Are yoU?  """					)						,						["""hello""", """!""", """how""", """are""", """you""", """?"""]					)
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""					)						,						["""hello"""]					)


						def   __lowercase (							self				:			Optional[Any]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase						,						strip_accents=lowerCAmelCase					)

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """					)						,						["""hällo""", """!""", """how""", """are""", """you""", """?"""]					)
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""					)						,						["""h\u00E9llo"""]					)


						def   __lowercase (							self				:			Optional[int]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase						,						strip_accents=lowerCAmelCase					)

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """					)						,						["""hallo""", """!""", """how""", """are""", """you""", """?"""]					)
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""					)						,						["""hello"""]					)


						def   __lowercase (							self				:			Union[str, Any]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase					)

										self.assertListEqual(
										    tokenizer.tokenize(""" \tHäLLo!how  \n Are yoU?  """					)						,						["""hallo""", """!""", """how""", """are""", """you""", """?"""]					)
										self.assertListEqual(tokenizer.tokenize("""H\u00E9llo"""					)						,						["""hello"""]					)


						def   __lowercase (							self				:			Dict					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase					)

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


						def   __lowercase (							self				:			int					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase						,						strip_accents=lowerCAmelCase					)

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


						def   __lowercase (							self				:			List[str]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase						,						strip_accents=lowerCAmelCase					)

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


						def   __lowercase (							self				:			List[str]					):
										lowerCAmelCase  =				RoCBertBasicTokenizer(do_lower_case=lowerCAmelCase						,						never_split=["""[UNK]"""]					)

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


						def   __lowercase (							self				:			int					):
										lowerCAmelCase  =				["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

										lowerCAmelCase  =				{}
										for i, token in enumerate(lowerCAmelCase					):
														lowerCAmelCase  =				i
										lowerCAmelCase  =				RoCBertWordpieceTokenizer(vocab=lowerCAmelCase						,						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   __lowercase (							self				:			str					):
										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   __lowercase (							self				:			List[Any]					):
										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   __lowercase (							self				:			Union[str, Any]					):
										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   __lowercase (							self				:			Optional[Any]					):
										lowerCAmelCase  =				self.get_tokenizer()

										# Example taken from the issue https://github.com/huggingface/tokenizers/issues/340
										self.assertListEqual([tokenizer.tokenize(lowerCAmelCase					) for t in ["""Test""", """\xad""", """test"""]]						,						[["""[UNK]"""], [], ["""[UNK]"""]]					)

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


						def   __lowercase (							self				:			Optional[int]					):
										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  =				f'''A, naïve {tokenizer_r.mask_token} AllenNLP sentence.'''
																		lowerCAmelCase  =				tokenizer_r.encode_plus(
																		    lowerCAmelCase						,						return_attention_mask=lowerCAmelCase						,						return_token_type_ids=lowerCAmelCase						,						return_offsets_mapping=lowerCAmelCase						,						add_special_tokens=lowerCAmelCase						,						)

																		lowerCAmelCase  =				tokenizer_r.do_lower_case if hasattr(lowerCAmelCase						,						"""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   __lowercase (							self				:			List[Any]					):
										lowerCAmelCase  =				["""的""", """人""", """有"""]
										lowerCAmelCase  =				"""""".join(lowerCAmelCase					)
										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(lowerCAmelCase						,						**lowerCAmelCase					)
																		lowerCAmelCase  =				self.rust_tokenizer_class.from_pretrained(lowerCAmelCase						,						**lowerCAmelCase					)

																		lowerCAmelCase  =				tokenizer_p.encode(lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer_r.encode(lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)

																		lowerCAmelCase  =				tokenizer_r.convert_ids_to_tokens(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer_p.convert_ids_to_tokens(lowerCAmelCase					)

																		# it is expected that each Chinese character is not preceded by "##"
																		self.assertListEqual(lowerCAmelCase						,						lowerCAmelCase					)
																		self.assertListEqual(lowerCAmelCase						,						lowerCAmelCase					)

																		lowerCAmelCase  =				False
																		lowerCAmelCase  =				self.rust_tokenizer_class.from_pretrained(lowerCAmelCase						,						**lowerCAmelCase					)
																		lowerCAmelCase  =				self.tokenizer_class.from_pretrained(lowerCAmelCase						,						**lowerCAmelCase					)

																		lowerCAmelCase  =				tokenizer_r.encode(lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer_p.encode(lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)

																		lowerCAmelCase  =				tokenizer_r.convert_ids_to_tokens(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer_p.convert_ids_to_tokens(lowerCAmelCase					)

																		# 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(lowerCAmelCase					)
																		]
																		self.assertListEqual(lowerCAmelCase						,						lowerCAmelCase					)
																		self.assertListEqual(lowerCAmelCase						,						lowerCAmelCase					)


						@slow
						def   __lowercase (							self				:			Any					):
										lowerCAmelCase  =				self.tokenizer_class(self.vocab_file						,						self.word_shape_file						,						self.word_pronunciation_file					)

										lowerCAmelCase  =				tokenizer.encode("""你好"""						,						add_special_tokens=lowerCAmelCase					)
										lowerCAmelCase  =				tokenizer.encode("""你是谁"""						,						add_special_tokens=lowerCAmelCase					)

										lowerCAmelCase  =				tokenizer.build_inputs_with_special_tokens(lowerCAmelCase					)
										lowerCAmelCase  =				tokenizer.build_inputs_with_special_tokens(lowerCAmelCase						,						lowerCAmelCase					)

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




						def   __lowercase (							self				:			Dict					):
										lowerCAmelCase  =				self.get_tokenizers(do_lower_case=lowerCAmelCase					)
										for tokenizer in tokenizers:
														with self.subTest(f'''{tokenizer.__class__.__name__}'''					):
																		lowerCAmelCase  =				"""你好，你是谁"""
																		lowerCAmelCase  =				tokenizer.tokenize(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer.convert_tokens_to_ids(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer.convert_tokens_to_shape_ids(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer.convert_tokens_to_pronunciation_ids(lowerCAmelCase					)
																		lowerCAmelCase  =				tokenizer.prepare_for_model(
																		    lowerCAmelCase						,						lowerCAmelCase						,						lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)

																		lowerCAmelCase  =				tokenizer.encode_plus(lowerCAmelCase						,						add_special_tokens=lowerCAmelCase					)

																		self.assertEqual(lowerCAmelCase						,						lowerCAmelCase					)











"""simple docstring"""







from typing import TYPE_CHECKING

from ...utils import (
    OptionalDependencyNotAvailable,
    _LazyModule,
    is_flax_available,
    is_tf_available,
    is_tokenizers_available,
    is_torch_available,
    is_vision_available,
)


a      =							{
    'configuration_owlvit': [
        'OWLVIT_PRETRAINED_CONFIG_ARCHIVE_MAP',
        'OwlViTConfig',
        'OwlViTOnnxConfig',
        'OwlViTTextConfig',
        'OwlViTVisionConfig',
    ],
    'processing_owlvit': ['OwlViTProcessor'],
}


try:
		if not is_vision_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		a      =							['OwlViTFeatureExtractor']
		a      =							['OwlViTImageProcessor']

try:
		if not is_torch_available():
				raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
		pass
else:
		a      =							[
		    'OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST',
		    'OwlViTModel',
		    'OwlViTPreTrainedModel',
		    'OwlViTTextModel',
		    'OwlViTVisionModel',
		    'OwlViTForObjectDetection',
		]

if TYPE_CHECKING:
		from .configuration_owlvit import (
		    OWLVIT_PRETRAINED_CONFIG_ARCHIVE_MAP,
		    OwlViTConfig,
		    OwlViTOnnxConfig,
		    OwlViTTextConfig,
		    OwlViTVisionConfig,
		)
		from .processing_owlvit import OwlViTProcessor

		try:
				if not is_vision_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .feature_extraction_owlvit import OwlViTFeatureExtractor
				from .image_processing_owlvit import OwlViTImageProcessor

		try:
				if not is_torch_available():
						raise OptionalDependencyNotAvailable()
		except OptionalDependencyNotAvailable:
				pass
		else:
				from .modeling_owlvit import (
				    OWLVIT_PRETRAINED_MODEL_ARCHIVE_LIST,
				    OwlViTForObjectDetection,
				    OwlViTModel,
				    OwlViTPreTrainedModel,
				    OwlViTTextModel,
				    OwlViTVisionModel,
				)

else:
		import sys

		a      =							_LazyModule(__name__, globals()['__file__'], _import_structure, module_spec=__spec__)





