



'''simple docstring'''

import os
import sys
import unittest


_A				:    Tuple     						=os.path.abspath(os.path.dirname(os.path.dirname(os.path.dirname(__file__))))
sys.path.append(os.path.join(git_repo_path, '''utils'''))

import get_test_info  # noqa: E402
from get_test_info import (  # noqa: E402
    get_model_to_test_mapping,
    get_model_to_tester_mapping,
    get_test_to_tester_mapping,
)


_A				:    Union[str, Any]     						=os.path.join('''tests''', '''models''', '''bert''', '''test_modeling_bert.py''')
_A				:    Dict     						=os.path.join('''tests''', '''models''', '''blip''', '''test_modeling_blip.py''')



class 		lowerCamelCase__      (						unittest.TestCase     ):




					'''simple docstring'''



					def 						__UpperCAmelCase     (  self				:	List[Any]					)  ->       str:



						'''simple docstring'''





						_lowercase					:       Union[str, Any]       =		get_test_to_tester_mapping(UpperCamelCase_					)
						_lowercase					:       Optional[int]       =		get_test_to_tester_mapping(UpperCamelCase_					)

						_lowercase					:       str       =		{'BertModelTest': 'BertModelTester'}

						_lowercase					:       List[Any]       =		{
						    'BlipModelTest': 'BlipModelTester',
						    'BlipTextImageModelTest': 'BlipTextImageModelsModelTester',
						    'BlipTextModelTest': 'BlipTextModelTester',
						    'BlipTextRetrievalModelTest': 'BlipTextRetrievalModelTester',
						    'BlipVQAModelTest': 'BlipVQAModelTester',
						    'BlipVisionModelTest': 'BlipVisionModelTester',
						}

						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)
						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)




					def 						__UpperCAmelCase     (  self				:	Dict					)  ->       List[str]:



						'''simple docstring'''





						_lowercase					:       Union[str, Any]       =		get_model_to_test_mapping(UpperCamelCase_					)
						_lowercase					:       int       =		get_model_to_test_mapping(UpperCamelCase_					)

						_lowercase					:       Union[str, Any]       =		{
						    'BertForMaskedLM': ['BertModelTest'],
						    'BertForMultipleChoice': ['BertModelTest'],
						    'BertForNextSentencePrediction': ['BertModelTest'],
						    'BertForPreTraining': ['BertModelTest'],
						    'BertForQuestionAnswering': ['BertModelTest'],
						    'BertForSequenceClassification': ['BertModelTest'],
						    'BertForTokenClassification': ['BertModelTest'],
						    'BertLMHeadModel': ['BertModelTest'],
						    'BertModel': ['BertModelTest'],
						}

						_lowercase					:       Optional[int]       =		{
						    'BlipForConditionalGeneration': ['BlipTextImageModelTest'],
						    'BlipForImageTextRetrieval': ['BlipTextRetrievalModelTest'],
						    'BlipForQuestionAnswering': ['BlipVQAModelTest'],
						    'BlipModel': ['BlipModelTest'],
						    'BlipTextModel': ['BlipTextModelTest'],
						    'BlipVisionModel': ['BlipVisionModelTest'],
						}

						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)
						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)




					def 						__UpperCAmelCase     (  self				:	int					)  ->       str:



						'''simple docstring'''





						_lowercase					:       Dict       =		get_model_to_tester_mapping(UpperCamelCase_					)
						_lowercase					:       Union[str, Any]       =		get_model_to_tester_mapping(UpperCamelCase_					)

						_lowercase					:       int       =		{
						    'BertForMaskedLM': ['BertModelTester'],
						    'BertForMultipleChoice': ['BertModelTester'],
						    'BertForNextSentencePrediction': ['BertModelTester'],
						    'BertForPreTraining': ['BertModelTester'],
						    'BertForQuestionAnswering': ['BertModelTester'],
						    'BertForSequenceClassification': ['BertModelTester'],
						    'BertForTokenClassification': ['BertModelTester'],
						    'BertLMHeadModel': ['BertModelTester'],
						    'BertModel': ['BertModelTester'],
						}

						_lowercase					:       Union[str, Any]       =		{
						    'BlipForConditionalGeneration': ['BlipTextImageModelsModelTester'],
						    'BlipForImageTextRetrieval': ['BlipTextRetrievalModelTester'],
						    'BlipForQuestionAnswering': ['BlipVQAModelTester'],
						    'BlipModel': ['BlipModelTester'],
						    'BlipTextModel': ['BlipTextModelTester'],
						    'BlipVisionModel': ['BlipVisionModelTester'],
						}

						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)
						self.assertEqual(get_test_info.to_json(UpperCamelCase_					)     ,  UpperCamelCase_					)





'''simple docstring'''

_A				:    Optional[Any]     						='''ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz0123456789+/'''







def 						__UpperCamelCase				(							_lowercase					) ->			bytes:
	# Make sure the supplied data is a bytes-like object
	if not isinstance(_lowercase,				_lowercase					):
		_lowercase					:       Union[str, Any]       =		f'''a bytes-like object is required, not \'{data.__class__.__name__}\''''
		raise TypeError(_lowercase					)

	_lowercase					:       int       =		''.join(bin(_lowercase					)[2:].zfill(8					) for byte in data					)

	_lowercase					:       Dict       =		len(_lowercase					) % 6 != 0

	if padding_needed:
		# The padding that will be added later
		_lowercase					:       Optional[Any]       =		B'=' * ((6 - len(_lowercase					) % 6) // 2)

		# Append binary_stream with arbitrary binary digits (0's by default) to make its
		# length a multiple of 6.
		binary_stream += "0" * (6 - len(_lowercase					) % 6)
	else:
		_lowercase					:       Optional[int]       =		B''

	# Encode every 6 binary digits to their corresponding Base64 character
	return (
	    "".join(
	        B64_CHARSET[int(binary_stream[index : index + 6],				2					)]
	        for index in range(0,				len(_lowercase					),				6					)					).encode()
	    + padding
	)







def 						__UpperCamelCase				(							_lowercase					) ->			bytes:
	# Make sure encoded_data is either a string or a bytes-like object
	if not isinstance(_lowercase,				_lowercase					) and not isinstance(_lowercase,				_lowercase					):
		_lowercase					:       int       =		(
		    'argument should be a bytes-like object or ASCII string, '
		    f'''not \'{encoded_data.__class__.__name__}\''''
		)
		raise TypeError(_lowercase					)

	# In case encoded_data is a bytes-like object, make sure it contains only
	# ASCII characters so we convert it to a string object
	if isinstance(_lowercase,				_lowercase					):
		try:
			_lowercase					:       Optional[int]       =		encoded_data.decode('utf-8'					)
		except UnicodeDecodeError:
			raise ValueError('base64 encoded data should only contain ASCII characters'					)

	_lowercase					:       Optional[int]       =		encoded_data.count('='					)

	# Check if the encoded string contains non base64 characters
	if padding:
		assert all(
		    char in B64_CHARSET for char in encoded_data[:-padding]					), "Invalid base64 character(s) found."
	else:
		assert all(
		    char in B64_CHARSET for char in encoded_data					), "Invalid base64 character(s) found."

	# Check the padding
	assert len(_lowercase					) % 4 == 0 and padding < 3, "Incorrect padding"

	if padding:
		# Remove padding if there is one
		_lowercase					:       str       =		encoded_data[:-padding]

		_lowercase					:       Tuple       =		''.join(
		    bin(B64_CHARSET.index(_lowercase					)					)[2:].zfill(6					) for char in encoded_data					)[: -padding * 2]
	else:
		_lowercase					:       Union[str, Any]       =		''.join(
		    bin(B64_CHARSET.index(_lowercase					)					)[2:].zfill(6					) for char in encoded_data					)

	_lowercase					:       List[str]       =		[
	    int(binary_stream[index : index + 8],				2					)
	    for index in range(0,				len(_lowercase					),				8					)
	]

	return bytes(_lowercase					)


if __name__ == "__main__":
		import doctest

		doctest.testmod()
