
import html

from ...feature_extraction_utils import BatchFeature, FeatureExtractionMixin
from ...utils import is_bsa_available, logging, requires_backends


if is_bsa_available():
						import bsa
						from bsa import BeautifulSoup


_snake_case      :							Any     	=    logging.get_logger(__name__)





class   UpperCamelCase_      (      __snake_case		):







						'''simple docstring'''



						def __init__(			self							:Optional[Any]		,    **lowerCAmelCase__							:List[str]     )  ->str:
										requires_backends(self		,    ["bs4"]     )
										super().__init__(**lowerCAmelCase__     )





						def        SCREAMING_SNAKE_CASE(			self							:Optional[int]		,    lowerCAmelCase__							:str     )  ->Dict:
										lowercase         =      []
										lowercase         =      []
										lowercase         =      element if element.name else element.parent
										for parent in child.parents:  # type: bs4.element.Tag
														lowercase         =      parent.find_all(child.name		,    recursive=lowerCAmelCase__     )
														xpath_tags.append(child.name     )
														xpath_subscripts.append(
														    0 if 1 == len(lowerCAmelCase__     ) else next(i for i, s in enumerate(lowerCAmelCase__		,    1     ) if s is child     )     )
														lowercase         =      parent
										xpath_tags.reverse()
										xpath_subscripts.reverse()
										return xpath_tags, xpath_subscripts





						def        SCREAMING_SNAKE_CASE(			self							:List[Any]		,    lowerCAmelCase__							:str     )  ->Tuple:
										lowercase         =      BeautifulSoup(lowerCAmelCase__		,    "html.parser"     )

										lowercase         =      []
										lowercase         =      []
										lowercase         =      []

										for element in html_code.descendants:
														if type(lowerCAmelCase__     ) == bsa.element.NavigableString:
																		if type(element.parent     ) != bsa.element.Tag:
																						continue

																		lowercase         =      html.unescape(lowerCAmelCase__     ).strip()
																		if not text_in_this_tag:
																						continue

																		all_doc_strings.append(lowerCAmelCase__     )

																		lowercase     ,	lowercase         =      self.xpath_soup(lowerCAmelCase__     )
																		stringaxtag_seq.append(lowerCAmelCase__     )
																		stringaxsubs_seq.append(lowerCAmelCase__     )

										if len(lowerCAmelCase__     ) != len(lowerCAmelCase__     ):
														raise ValueError("Number of doc strings and xtags does not correspond"     )
										if len(lowerCAmelCase__     ) != len(lowerCAmelCase__     ):
														raise ValueError("Number of doc strings and xsubs does not correspond"     )

										return all_doc_strings, stringaxtag_seq, stringaxsubs_seq





						def        SCREAMING_SNAKE_CASE(			self							:Optional[int]		,    lowerCAmelCase__							:Any		,    lowerCAmelCase__							:Union[str, Any]     )  ->int:
										lowercase         =      ""
										for tagname, subs in zip(lowerCAmelCase__		,    lowerCAmelCase__     ):
														xpath += F'''/{tagname}'''
														if subs != 0:
																		xpath += F'''[{subs}]'''
										return xpath





						def __call__(			self							:Union[str, Any]		,    lowerCAmelCase__							:List[Any]     )  ->List[Any]:
										lowercase         =      False

										# Check that strings has a valid type
										if isinstance(lowerCAmelCase__		,    lowerCAmelCase__     ):
														lowercase         =      True
										elif isinstance(lowerCAmelCase__		,    (list, tuple)     ):
														if len(lowerCAmelCase__     ) == 0 or isinstance(html_strings[0]		,    lowerCAmelCase__     ):
																		lowercase         =      True

										if not valid_strings:
														raise ValueError(
														    "HTML strings must of type `str`, `List[str]` (batch of examples), "
														    F'''but is of type {type(lowerCAmelCase__     )}.'''     )

										lowercase         =      bool(isinstance(lowerCAmelCase__		,    (list, tuple)     ) and (isinstance(html_strings[0]		,    lowerCAmelCase__     ))     )

										if not is_batched:
														lowercase         =      [html_strings]

										# Get nodes + xpaths
										lowercase         =      []
										lowercase         =      []
										for html_string in html_strings:
														lowercase     ,	lowercase     ,	lowercase         =      self.get_three_from_single(lowerCAmelCase__     )
														nodes.append(lowerCAmelCase__     )
														lowercase         =      []
														for node, tag_list, sub_list in zip(lowerCAmelCase__		,    lowerCAmelCase__		,    lowerCAmelCase__     ):
																		lowercase         =      self.construct_xpath(lowerCAmelCase__		,    lowerCAmelCase__     )
																		xpath_strings.append(lowerCAmelCase__     )
														xpaths.append(lowerCAmelCase__     )

										# return as Dict
										lowercase         =      {"nodes": nodes, "xpaths": xpaths}
										lowercase         =      BatchFeature(data=lowerCAmelCase__		,    tensor_type=lowerCAmelCase__     )

										return encoded_inputs




"""simple docstring"""



import json
import os
import unittest

from transformers.models.gptsan_japanese.tokenization_gptsan_japanese import (
    VOCAB_FILES_NAMES,
    GPTSanJapaneseTokenizer,
)
from transformers.testing_utils import require_tokenizers, slow

from ...test_tokenization_common import TokenizerTesterMixin







@require_tokenizers
class    __a						( __snake_case					,			unittest.TestCase							):
				lowerCamelCase     :       Any			       =GPTSanJapaneseTokenizer
				lowerCamelCase     :       List[str]			       =False
				lowerCamelCase     :       List[str]			       ={'do_clean_text': False, 'add_prefix_space': False}







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					super().setUp()

					# fmt: off
					lowerCAmelCase_      							=			['''こん''', '''こんに''', '''にちは''', '''ばんは''', '''世界,㔺界''', '''、''', '''。''', '''<BR>''', '''<SP>''', '''<TAB>''', '''<URL>''', '''<EMAIL>''', '''<TEL>''', '''<DATE>''', '''<PRICE>''', '''<BLOCK>''', '''<KIGOU>''', '''<U2000U2BFF>''', '''<|emoji1|>''', '''<unk>''', '''<|bagoftoken|>''', '''<|endoftext|>''']
					# fmt: on
					lowerCAmelCase_      							=			{'''emoji''': {'''\ud83d\ude00''': '''<|emoji1|>'''}, '''emoji_inv''': {'''<|emoji1|>''': '''\ud83d\ude00'''}}  # 😀
					lowerCAmelCase_      							=			{'''unk_token''': '''<unk>'''}

					lowerCAmelCase_      							=			os.path.join(self.tmpdirname      ,	VOCAB_FILES_NAMES['''vocab_file''']   )
					lowerCAmelCase_      							=			os.path.join(self.tmpdirname      ,	VOCAB_FILES_NAMES['''emoji_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.emoji_file      ,	'''w'''   ) as emoji_writer:
						emoji_writer.write(json.dumps(UpperCAmelCase   )   )







				def      lowerCamelCase_     ( self      ,	**UpperCAmelCase   ):

					'''simple docstring'''






					kwargs.update(self.special_tokens_map   )
					return GPTSanJapaneseTokenizer.from_pretrained(self.tmpdirname      ,	**UpperCAmelCase   )







				def      lowerCamelCase_     ( self      ,	UpperCAmelCase   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			'''こんにちは、世界。 \nこんばんは、㔺界。😀'''
					lowerCAmelCase_      							=			'''こんにちは、世界。 \nこんばんは、世界。😀'''
					return input_text, output_text







				def      lowerCamelCase_     ( self      ,	UpperCAmelCase   ):

					'''simple docstring'''






					lowerCAmelCase_							,	lowerCAmelCase_      							=			self.get_input_output_texts(UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.encode(UpperCAmelCase      ,	add_special_tokens=UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.decode(UpperCAmelCase      ,	clean_up_tokenization_spaces=UpperCAmelCase   )
					return text, ids







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					pass  # TODO add if relevant







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					pass  # TODO add if relevant







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					pass  # TODO add if relevant







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.get_tokenizer()

					# Testing tokenization
					lowerCAmelCase_      							=			'''こんにちは、世界。　こんばんは、㔺界。'''
					lowerCAmelCase_      							=			['''こん''', '''にちは''', '''、''', '''世界''', '''。''', '''<SP>''', '''こん''', '''ばんは''', '''、''', '''㔺界''', '''。''']
					lowerCAmelCase_      							=			tokenizer.tokenize(UpperCAmelCase   )
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )

					# Testing conversion to ids without special tokens
					lowerCAmelCase_      							=			[0, 2, 5, 4, 6, 8, 0, 3, 5, 4, 6]
					lowerCAmelCase_      							=			tokenizer.convert_tokens_to_ids(UpperCAmelCase   )
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )

					# Testing conversion to ids with special tokens
					lowerCAmelCase_      							=			tokens + [tokenizer.unk_token]
					lowerCAmelCase_      							=			[0, 2, 5, 4, 6, 8, 0, 3, 5, 4, 6, 19]
					lowerCAmelCase_      							=			tokenizer.convert_tokens_to_ids(UpperCAmelCase   )
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.get_tokenizer()

					# Testing tokenization
					lowerCAmelCase_      							=			'''こんにちは、<|bagoftoken|>世界。こんばんは、<|bagoftoken|>㔺界。'''
					lowerCAmelCase_      							=			'''こんにちは、、、、世界。こんばんは、、、、世界。'''
					lowerCAmelCase_      							=			tokenizer.encode(UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.decode(UpperCAmelCase   )
					self.assertEqual(UpperCAmelCase      ,	UpperCAmelCase   )







				@slow
				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.tokenizer_class.from_pretrained('''Tanrei/GPTSAN-japanese'''   )

					# Testing tokenization
					lowerCAmelCase_      							=			'''こんにちは、世界。'''
					lowerCAmelCase_      							=			'''こんばんは、㔺界。😀'''
					lowerCAmelCase_      							=			'''こんにちは、世界。こんばんは、世界。😀'''
					lowerCAmelCase_      							=			tokenizer.encode(prefix_text + input_text   )
					lowerCAmelCase_      							=			tokenizer.encode(''''''      ,	prefix_text=prefix_text + input_text   )
					lowerCAmelCase_      							=			tokenizer.encode(UpperCAmelCase      ,	prefix_text=UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.decode(UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.decode(UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.decode(UpperCAmelCase   )
					self.assertEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertEqual(UpperCAmelCase      ,	UpperCAmelCase   )







				@slow
				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.tokenizer_class.from_pretrained('''Tanrei/GPTSAN-japanese'''   )

					# Testing tokenization
					lowerCAmelCase_      							=			'''こんにちは、世界。'''
					lowerCAmelCase_      							=			'''こんばんは、㔺界。😀'''

					lowerCAmelCase_      							=			len(tokenizer.encode(UpperCAmelCase   )   ) - 2
					lowerCAmelCase_      							=			len(tokenizer.encode(UpperCAmelCase   )   ) - 2

					lowerCAmelCase_      							=			[1] + [0] * (len_prefix + len_text + 1)
					lowerCAmelCase_      							=			[1] * (len_prefix + len_text + 1) + [0]
					lowerCAmelCase_      							=			[1] + [1] * (len_prefix) + [0] * (len_text + 1)

					lowerCAmelCase_      							=			tokenizer(prefix_text + input_text   ).token_type_ids
					lowerCAmelCase_      							=			tokenizer(''''''      ,	prefix_text=prefix_text + input_text   ).token_type_ids
					lowerCAmelCase_      							=			tokenizer(UpperCAmelCase      ,	prefix_text=UpperCAmelCase   ).token_type_ids
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertListEqual(UpperCAmelCase      ,	UpperCAmelCase   )







				@slow
				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.tokenizer_class.from_pretrained('''Tanrei/GPTSAN-japanese'''   )

					lowerCAmelCase_      							=			tokenizer.encode('''あンいワ'''   )
					lowerCAmelCase_      							=			tokenizer.encode(''''''      ,	prefix_text='''あンいワ'''   )
					lowerCAmelCase_      							=			tokenizer.encode('''いワ'''      ,	prefix_text='''あン'''   )

					self.assertEqual(tokenizer.decode(UpperCAmelCase   )      ,	tokenizer.decode(UpperCAmelCase   )   )
					self.assertEqual(tokenizer.decode(UpperCAmelCase   )      ,	tokenizer.decode(UpperCAmelCase   )   )
					self.assertNotEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertNotEqual(UpperCAmelCase      ,	UpperCAmelCase   )
					self.assertEqual(x_token_a[1]      ,	x_token_a[-1]   )  # SEG token
					self.assertEqual(x_token_a[1]      ,	x_token_a[3]   )  # SEG token







				@slow
				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					lowerCAmelCase_      							=			self.tokenizer_class.from_pretrained('''Tanrei/GPTSAN-japanese'''   )

					lowerCAmelCase_      							=			[['''武田信玄''', '''は、'''], ['''織田信長''', '''の配下の、''']]
					lowerCAmelCase_      							=			tokenizer(UpperCAmelCase      ,	padding=UpperCAmelCase   )
					lowerCAmelCase_      							=			tokenizer.batch_encode_plus(UpperCAmelCase      ,	padding=UpperCAmelCase   )

					# fmt: off
					lowerCAmelCase_      							=			[[3_5993, 8640, 2_5948, 3_5998, 3_0647, 3_5675, 3_5999, 3_5999], [3_5993, 1_0382, 9868, 3_5998, 3_0646, 9459, 3_0646, 3_5675]]
					lowerCAmelCase_      							=			[[1, 1, 1, 0, 0, 0, 0, 0], [1, 1, 1, 0, 0, 0, 0, 0]]
					lowerCAmelCase_      							=			[[1, 1, 1, 1, 1, 1, 0, 0], [1, 1, 1, 1, 1, 1, 1, 1]]
					# fmt: on
					self.assertListEqual(x_token.input_ids      ,	UpperCAmelCase   )
					self.assertListEqual(x_token.token_type_ids      ,	UpperCAmelCase   )
					self.assertListEqual(x_token.attention_mask      ,	UpperCAmelCase   )
					self.assertListEqual(x_token_a.input_ids      ,	UpperCAmelCase   )
					self.assertListEqual(x_token_a.token_type_ids      ,	UpperCAmelCase   )
					self.assertListEqual(x_token_a.attention_mask      ,	UpperCAmelCase   )







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					pass







				def      lowerCamelCase_     ( self   ):

					'''simple docstring'''






					pass