






"""simple docstring"""







import os
import unittest

from transformers import MobileBertTokenizer, MobileBertTokenizerFast
from transformers.models.bert.tokenization_bert import (
    VOCAB_FILES_NAMES,
    BasicTokenizer,
    WordpieceTokenizer,
    _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							):





							"""simple docstring"""





							_SCREAMING_SNAKE_CASE    =MobileBertTokenizer
							_SCREAMING_SNAKE_CASE    =MobileBertTokenizerFast
							_SCREAMING_SNAKE_CASE    =True
							_SCREAMING_SNAKE_CASE    =True
							_SCREAMING_SNAKE_CASE    =filter_non_english
							_SCREAMING_SNAKE_CASE    ='google/mobilebert-uncased'




							def  lowercase		(   self:     Tuple    ):

												'''simple docstring'''

												super().setUp()

												a__          =  [
												    """[UNK]""",
												    """[CLS]""",
												    """[SEP]""",
												    """[PAD]""",
												    """[MASK]""",
												    """want""",
												    """##want""",
												    """##ed""",
												    """wa""",
												    """un""",
												    """runn""",
												    """##ing""",
												    """,""",
												    """low""",
												    """lowest""",
												]
												a__          =  os.path.join(self.tmpdirname				,		VOCAB_FILES_NAMES['''vocab_file''']    )
												with open(self.vocab_file				,		'''w'''				,		encoding='''utf-8'''    ) as vocab_writer:
																	vocab_writer.write(''''''.join([x + '''\n''' for x in vocab_tokens]    )    )

												a__          =  [
												    (tokenizer_def[0], self.pre_trained_model_path, tokenizer_def[2])  # else the 'google/' prefix is stripped
												    for tokenizer_def in self.tokenizers_list
												]




							def  lowercase		(   self:     Union[str, Any]				,		__A:     int    ):

												'''simple docstring'''

												a__          =  """UNwant\u00E9d,running"""
												a__          =  """unwanted, running"""
												return input_text, output_text




							def  lowercase		(   self:     str    ):

												'''simple docstring'''

												a__          =  self.tokenizer_class(self.vocab_file    )

												a__          =  tokenizer.tokenize('''UNwant\u00E9d,running'''    )
												self.assertListEqual(lowerCamelCase_				,		['''un''', '''##want''', '''##ed''', ''',''', '''runn''', '''##ing''']    )
												self.assertListEqual(tokenizer.convert_tokens_to_ids(lowerCamelCase_    )				,		[9, 6, 7, 12, 10, 11]    )




							def  lowercase		(   self:     int    ):

												'''simple docstring'''

												if not self.test_rust_tokenizer:
																	return

												a__          =  self.get_tokenizer()
												a__          =  self.get_rust_tokenizer()

												a__          =  """UNwant\u00E9d,running"""

												a__          =  tokenizer.tokenize(lowerCamelCase_    )
												a__          =  rust_tokenizer.tokenize(lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )

												a__          =  tokenizer.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
												a__          =  rust_tokenizer.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )

												a__          =  self.get_rust_tokenizer()
												a__          =  tokenizer.encode(lowerCamelCase_    )
												a__          =  rust_tokenizer.encode(lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )

												# With lower casing
												a__          =  self.get_tokenizer(do_lower_case=lowerCamelCase_    )
												a__          =  self.get_rust_tokenizer(do_lower_case=lowerCamelCase_    )

												a__          =  """UNwant\u00E9d,running"""

												a__          =  tokenizer.tokenize(lowerCamelCase_    )
												a__          =  rust_tokenizer.tokenize(lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )

												a__          =  tokenizer.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
												a__          =  rust_tokenizer.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )

												a__          =  self.get_rust_tokenizer()
												a__          =  tokenizer.encode(lowerCamelCase_    )
												a__          =  rust_tokenizer.encode(lowerCamelCase_    )
												self.assertListEqual(lowerCamelCase_				,		lowerCamelCase_    )




							def  lowercase		(   self:     Optional[int]    ):

												'''simple docstring'''

												a__          =  BasicTokenizer()

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




							def  lowercase		(   self:     Tuple    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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:     Dict    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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:     Tuple    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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:     Optional[int]    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(do_lower_case=lowerCamelCase_    )

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




							def  lowercase		(   self:     Union[str, Any]    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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:     Union[str, Any]    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(do_lower_case=lowerCamelCase_				,		strip_accents=lowerCamelCase_    )

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




							def  lowercase		(   self:     Optional[int]    ):

												'''simple docstring'''

												a__          =  BasicTokenizer(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:     Dict    ):

												'''simple docstring'''

												a__          =  ["""[UNK]""", """[CLS]""", """[SEP]""", """want""", """##want""", """##ed""", """wa""", """un""", """runn""", """##ing"""]

												a__          =  {}
												for i, token in enumerate(lowerCamelCase_    ):
																	a__          =  i
												a__          =  WordpieceTokenizer(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:     Dict    ):

												'''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  lowercase		(   self:     Tuple    ):

												'''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  lowercase		(   self:     Any    ):

												'''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  lowercase		(   self:     List[Any]    ):

												'''simple docstring'''

												a__          =  self.get_tokenizer()
												a__          =  self.get_rust_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]''']]    )

												self.assertListEqual(
												    [rust_tokenizer.tokenize(lowerCamelCase_    ) for t in ['''Test''', '''\xad''', '''test''']]				,		[['''[UNK]'''], [], ['''[UNK]''']]    )




							@slow
							def  lowercase		(   self:     Optional[int]    ):

												'''simple docstring'''

												a__          =  self.tokenizer_class.from_pretrained('''google/mobilebert-uncased'''    )

												a__          =  tokenizer.encode('''sequence builders'''				,		add_special_tokens=lowerCamelCase_    )
												a__          =  tokenizer.encode('''multi-sequence build'''				,		add_special_tokens=lowerCamelCase_    )

												a__          =  tokenizer.build_inputs_with_special_tokens(lowerCamelCase_    )
												a__          =  tokenizer.build_inputs_with_special_tokens(lowerCamelCase_				,		lowerCamelCase_    )

												assert encoded_sentence == [101] + text + [102]
												assert encoded_pair == [101] + text + [102] + text_a + [102]




							def  lowercase		(   self:     Tuple    ):

												'''simple docstring'''

												for tokenizer, pretrained_name, kwargs in self.tokenizers_list:
																	with self.subTest(F'{tokenizer.__class__.__name__} ({pretrained_name})'    ):
																						a__          =  self.rust_tokenizer_class.from_pretrained(lowerCamelCase_				,		**lowerCamelCase_    )

																						a__          =  F'A, naïve {tokenizer_r.mask_token} AllenNLP sentence.'
																						a__          =  tokenizer_r.encode_plus(
																						    lowerCamelCase_				,		return_attention_mask=lowerCamelCase_				,		return_token_type_ids=lowerCamelCase_				,		return_offsets_mapping=lowerCamelCase_				,		add_special_tokens=lowerCamelCase_				,		)

																						a__          =  tokenizer_r.do_lower_case if hasattr(lowerCamelCase_				,		'''do_lower_case'''    ) else False
																						a__          =  (
																						    [
																						        ((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:     Any    ):

												'''simple docstring'''

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

																						a__          =  tokenizer_p.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
																						a__          =  tokenizer_r.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )

																						a__          =  tokenizer_r.convert_ids_to_tokens(lowerCamelCase_    )
																						a__          =  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_    )

																						a__          =  False
																						a__          =  self.rust_tokenizer_class.from_pretrained(lowerCamelCase_				,		**lowerCamelCase_    )
																						a__          =  self.tokenizer_class.from_pretrained(lowerCamelCase_				,		**lowerCamelCase_    )

																						a__          =  tokenizer_r.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )
																						a__          =  tokenizer_p.encode(lowerCamelCase_				,		add_special_tokens=lowerCamelCase_    )

																						a__          =  tokenizer_r.convert_ids_to_tokens(lowerCamelCase_    )
																						a__          =  tokenizer_p.convert_ids_to_tokens(lowerCamelCase_    )

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










"""simple docstring"""







from ...configuration_utils import PretrainedConfig
from ...utils import logging


__a       :	Any             = logging.get_logger(__name__)

__a       :	Union[str, Any]             = {
    'caidas/swin2sr-classicalsr-x2-64': (
        'https://huggingface.co/caidas/swin2sr-classicalsr-x2-64/resolve/main/config.json'
    ),
}







class        _SCREAMING_SNAKE_CASE						(  __snake_case							):





							"""simple docstring"""





							_SCREAMING_SNAKE_CASE    ='swin2sr'

							_SCREAMING_SNAKE_CASE    ={
							    'hidden_size': 'embed_dim',
							    'num_attention_heads': 'num_heads',
							    'num_hidden_layers': 'num_layers',
							}




							def __init__(   self:     Union[str, Any]				,		__A:     List[Any]=64				,		__A:     int=1				,		__A:     Dict=3				,		__A:     List[Any]=180				,		__A:     int=[6, 6, 6, 6, 6, 6]				,		__A:     Tuple=[6, 6, 6, 6, 6, 6]				,		__A:     int=8				,		__A:     Optional[int]=2.0				,		__A:     Optional[int]=True				,		__A:     int=0.0				,		__A:     Any=0.0				,		__A:     Optional[Any]=0.1				,		__A:     Optional[Any]="gelu"				,		__A:     Dict=False				,		__A:     List[Any]=0.0_2				,		__A:     List[Any]=1e-5				,		__A:     List[str]=2				,		__A:     int=1.0				,		__A:     Dict="1conv"				,		__A:     Optional[Any]="pixelshuffle"				,		**__A:     Dict				,		):

												'''simple docstring'''

												super().__init__(**__A    )

												a__          =  image_size
												a__          =  patch_size
												a__          =  num_channels
												a__          =  embed_dim
												a__          =  depths
												a__          =  len(__A    )
												a__          =  num_heads
												a__          =  window_size
												a__          =  mlp_ratio
												a__          =  qkv_bias
												a__          =  hidden_dropout_prob
												a__          =  attention_probs_dropout_prob
												a__          =  drop_path_rate
												a__          =  hidden_act
												a__          =  use_absolute_embeddings
												a__          =  layer_norm_eps
												a__          =  initializer_range
												a__          =  upscale
												a__          =  img_range
												a__          =  resi_connection
												a__          =  upsampler


