




'''simple docstring'''

import unittest

from transformers import RoFormerTokenizer, RoFormerTokenizerFast
from transformers.testing_utils import require_rjieba, require_tokenizers

from ...test_tokenization_common import TokenizerTesterMixin






@require_rjieba
@require_tokenizers
class 						a				(   _lowerCamelCase   ,     unittest.TestCase     ):
				snake_case_      =		RoFormerTokenizer
				snake_case_      =		RoFormerTokenizerFast
				snake_case_      =		True
				snake_case_      =		True


				def   A_ (		self    :					Tuple					):
									super().setUp()


				def   A_ (		self    :					List[str]       ,				**lowercase_    :					str					):
									return self.tokenizer_class.from_pretrained('''junnyu/roformer_chinese_base'''       ,				**lowercase_					)


				def   A_ (		self    :					int       ,				**lowercase_    :					int					):
									return self.rust_tokenizer_class.from_pretrained('''junnyu/roformer_chinese_base'''       ,				**lowercase_					)


				def   A_ (		self    :					Optional[int]					):
									snake_case_									=				'''永和服装饰品有限公司,今天天气非常好'''
									snake_case_									=				'''永和 服装 饰品 有限公司 , 今 天 天 气 非常 好'''
									return input_text, output_text


				def   A_ (		self    :					List[Any]					):
									snake_case_									=				self.get_tokenizer()
									snake_case_			,snake_case_									=				self.get_chinese_input_output_texts()
									snake_case_									=				tokenizer.tokenize(lowercase_					)

									self.assertListEqual(lowercase_       ,				output_text.split()					)

									snake_case_									=				tokens + [tokenizer.unk_token]
									snake_case_									=				[2_2943, 2_1332, 3_4431, 4_5904, 117, 306, 1231, 1231, 2653, 3_3994, 1266, 100]
									self.assertListEqual(tokenizer.convert_tokens_to_ids(lowercase_					)       ,				lowercase_					)


				def   A_ (		self    :					int					):
									snake_case_									=				self.get_rust_tokenizer()
									snake_case_			,snake_case_									=				self.get_chinese_input_output_texts()
									snake_case_									=				tokenizer.tokenize(lowercase_					)
									self.assertListEqual(lowercase_       ,				output_text.split()					)
									snake_case_									=				tokens + [tokenizer.unk_token]
									snake_case_									=				[2_2943, 2_1332, 3_4431, 4_5904, 117, 306, 1231, 1231, 2653, 3_3994, 1266, 100]
									self.assertListEqual(tokenizer.convert_tokens_to_ids(lowercase_					)       ,				lowercase_					)


				def   A_ (		self    :					int					):
									pass


				def   A_ (		self    :					List[Any]					):
									pass


				def   A_ (		self    :					Optional[int]					):
									pass








'''simple docstring'''

import contextlib
import importlib
import io
import unittest

import transformers

# Try to import everything from transformers to ensure every object can be loaded.
from transformers import *  # noqa F406
from transformers.testing_utils import DUMMY_UNKNOWN_IDENTIFIER, require_flax, require_tf, require_torch
from transformers.utils import ContextManagers, find_labels, is_flax_available, is_tf_available, is_torch_available


if is_torch_available():
  from transformers import BertForPreTraining, BertForQuestionAnswering, BertForSequenceClassification

if is_tf_available():
  from transformers import TFBertForPreTraining, TFBertForQuestionAnswering, TFBertForSequenceClassification

if is_flax_available():
  from transformers import FlaxBertForPreTraining, FlaxBertForQuestionAnswering, FlaxBertForSequenceClassification


__lowerCAmelCase							:	Dict			=DUMMY_UNKNOWN_IDENTIFIER
# An actual model hosted on huggingface.co

__lowerCAmelCase							:	List[Any]			="main"
# Default branch name
__lowerCAmelCase							:	int			="f2c752cfc5c0ab6f4bdec59acea69eefbee381c2"
# One particular commit (not the top of `main`)
__lowerCAmelCase							:	List[Any]			="aaaaaaa"
# This commit does not exist, so we should 404.

__lowerCAmelCase							:	Optional[int]			="d9e9f15bc825e4b2c9249e9578f884bbcb5e3684"
# Sha-1 of config.json on the top of `main`, for checking purposes
__lowerCAmelCase							:	Dict			="4b243c475af8d0a7754e87d7d096c92e5199ec2fe168a2ee7998e3b8e9bcb1d3"





@contextlib.contextmanager
def 						UpperCamelCase				(							):
    print("Welcome!"       )
    yield
    print("Bye!"       )





@contextlib.contextmanager
def 						UpperCamelCase				(							):
    print("Bonjour!"       )
    yield
    print("Au revoir!"       )






class    UpperCAmelCase	(       unittest.TestCase				):
 def      UpperCAmelCase_							(				self      :str							)->							List[str]:
     # If the spec is missing, importlib would not be able to import the module dynamically.
     assert transformers.__spec__ is not None
     assert importlib.util.find_spec("transformers"							) is not None






class    UpperCAmelCase	(       unittest.TestCase				):
 @unittest.mock.patch("sys.stdout"						,       new_callable=io.StringIO							)
 def      UpperCAmelCase_							(				self      :Union[str, Any]						,       lowercase_      :Union[str, Any]							)->							Any:
     with ContextManagers([]							):
         print("Transformers are awesome!"							)
     # The print statement adds a new line at the end of the output
     self.assertEqual(mock_stdout.getvalue()						,       "Transformers are awesome!\n"							)
 @unittest.mock.patch("sys.stdout"						,       new_callable=io.StringIO							)
 def      UpperCAmelCase_							(				self      :Dict						,       lowercase_      :Optional[Any]							)->							Tuple:
     with ContextManagers([context_en()]							):
         print("Transformers are awesome!"							)
     # The output should be wrapped with an English welcome and goodbye
     self.assertEqual(mock_stdout.getvalue()						,       "Welcome!\nTransformers are awesome!\nBye!\n"							)
 @unittest.mock.patch("sys.stdout"						,       new_callable=io.StringIO							)
 def      UpperCAmelCase_							(				self      :Union[str, Any]						,       lowercase_      :int							)->							Union[str, Any]:
     with ContextManagers([context_fr(), context_en()]							):
         print("Transformers are awesome!"							)
     # The output should be wrapped with an English and French welcome and goodbye
     self.assertEqual(mock_stdout.getvalue()						,       "Bonjour!\nWelcome!\nTransformers are awesome!\nBye!\nAu revoir!\n"							)
 @require_torch
 def      UpperCAmelCase_							(				self      :int							)->							Dict:
     self.assertEqual(find_labels(lowercase_							)						,       ["labels"]							)
     self.assertEqual(find_labels(lowercase_							)						,       ["labels", "next_sentence_label"]							)
     self.assertEqual(find_labels(lowercase_							)						,       ["start_positions", "end_positions"]							)






     class    UpperCAmelCase	(       UpperCamelCase__				):
      pass

     self.assertEqual(find_labels(lowercase_							)						,       ["labels"]							)
 @require_tf
 def      UpperCAmelCase_							(				self      :Union[str, Any]							)->							Union[str, Any]:
     self.assertEqual(find_labels(lowercase_							)						,       ["labels"]							)
     self.assertEqual(find_labels(lowercase_							)						,       ["labels", "next_sentence_label"]							)
     self.assertEqual(find_labels(lowercase_							)						,       ["start_positions", "end_positions"]							)






     class    UpperCAmelCase	(       UpperCamelCase__				):
      pass

     self.assertEqual(find_labels(lowercase_							)						,       ["labels"]							)



 @require_flax
 def      UpperCAmelCase_							(				self      :Dict							)->							str:
     # Flax models don't have labels
     self.assertEqual(find_labels(lowercase_							)						,       []							)
     self.assertEqual(find_labels(lowercase_							)						,       []							)
     self.assertEqual(find_labels(lowercase_							)						,       []							)






     class    UpperCAmelCase	(       UpperCamelCase__				):
      pass

     self.assertEqual(find_labels(lowercase_							)						,       []							)



