






"""simple docstring"""
import enum
import shutil
import sys


lowerCamelCase_			,	lowerCamelCase_       			=   shutil.get_terminal_size()

lowerCamelCase_       			=   {'''UP''': '''A''', '''DOWN''': '''B''', '''RIGHT''': '''C''', '''LEFT''': '''D'''}







class     UpperCamelCase_ (enum.Enum  ):
	__magic_name__			   =       0
	__magic_name__			   =       1





def        snake_case   (     A__			,A__=""		):
			sys.stdout.write(str(A__		) + end		)
			sys.stdout.flush()

def        snake_case   (     A__			,A__			,A__=""		):
			forceWrite(F"""\u001b[{color}m{content}\u001b[0m"""			,A__		)

def        snake_case   (     ):
			forceWrite("\r"		)

def        snake_case   (     A__			,A__		):
			forceWrite(F"""\033[{num_lines}{CURSOR_TO_CHAR[direction.upper()]}"""		)

def        snake_case   (     ):
			forceWrite(" " * TERMINAL_WIDTH		)
			reset_cursor()




def        snake_case   (     ):
			reset_cursor()
			forceWrite("-" * TERMINAL_WIDTH		)











"""simple docstring"""
import copy
import json
import os
import tempfile

from transformers import is_torch_available

from .test_configuration_utils import config_common_kwargs







class     UpperCamelCase_ (__A  ):





	def __init__(   self      :				List[Any]			,			lowerCAmelCase_      :				List[str]			,			lowerCAmelCase_      :				List[Any]=None			,			lowerCAmelCase_      :				List[str]=True			,			lowerCAmelCase_      :				List[Any]=None			,			**lowerCAmelCase_      :				Tuple		)		->    Dict:
				UpperCAmelCase_  :				Tuple      =      parent
				UpperCAmelCase_  :				Optional[int]      =      config_class
				UpperCAmelCase_  :				List[str]      =      has_text_modality
				UpperCAmelCase_  :				Tuple      =      kwargs
				UpperCAmelCase_  :				int      =      common_properties





	def   _SCREAMING_SNAKE_CASE   (   self      :				Union[str, Any]		)		->    Union[str, Any]:
				UpperCAmelCase_  :				Optional[int]      =      self.config_class(**self.inputs_dict		)
				UpperCAmelCase_  :				int      =      (
				    ["hidden_size", "num_attention_heads", "num_hidden_layers"]
				    if self.common_properties is None
				    else self.common_properties
				)

				# Add common fields for text models
				if self.has_text_modality:
							common_properties.extend(["vocab_size"]		)

				# Test that config has the common properties as getters
				for prop in common_properties:
							self.parent.assertTrue(hasattr(lowerCAmelCase_			,			lowerCAmelCase_		)			,			msg=f"""`{prop}` does not exist"""		)

				# Test that config has the common properties as setter
				for idx, name in enumerate(lowerCAmelCase_		):
							try:
										setattr(lowerCAmelCase_			,			lowerCAmelCase_			,			lowerCAmelCase_		)
										self.parent.assertEqual(
										    getattr(lowerCAmelCase_			,			lowerCAmelCase_		)			,			lowerCAmelCase_			,			msg=f"""`{name} value {idx} expected, but was {getattr(lowerCAmelCase_			,			lowerCAmelCase_		)}"""		)
							except NotImplementedError:
										# Some models might not be able to implement setters for common_properties
										# In that case, a NotImplementedError is raised
										pass

        # Test if config class can be called with Config(prop_name=..)
				for idx, name in enumerate(lowerCAmelCase_		):
							try:
										UpperCAmelCase_  :				Optional[Any]      =      self.config_class(**{name: idx}		)
										self.parent.assertEqual(
										    getattr(lowerCAmelCase_			,			lowerCAmelCase_		)			,			lowerCAmelCase_			,			msg=f"""`{name} value {idx} expected, but was {getattr(lowerCAmelCase_			,			lowerCAmelCase_		)}"""		)
							except NotImplementedError:
										# Some models might not be able to implement setters for common_properties
										# In that case, a NotImplementedError is raised
										pass





	def   _SCREAMING_SNAKE_CASE   (   self      :				List[Any]		)		->    Tuple:
				UpperCAmelCase_  :				str      =      self.config_class(**self.inputs_dict		)
				UpperCAmelCase_  :				List[str]      =      json.loads(config.to_json_string()		)
				for key, value in self.inputs_dict.items():
							self.parent.assertEqual(obj[key]			,			lowerCAmelCase_		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				Optional[Any]		)		->    Any:
				UpperCAmelCase_  :				Optional[Any]      =      self.config_class(**self.inputs_dict		)

				with tempfile.TemporaryDirectory() as tmpdirname:
							UpperCAmelCase_  :				List[str]      =      os.path.join(lowerCAmelCase_			,			"config.json"		)
							config_first.to_json_file(lowerCAmelCase_		)
							UpperCAmelCase_  :				Dict      =      self.config_class.from_json_file(lowerCAmelCase_		)

				self.parent.assertEqual(config_second.to_dict()			,			config_first.to_dict()		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				List[Any]		)		->    Dict:
				UpperCAmelCase_  :				Any      =      self.config_class(**self.inputs_dict		)

				with tempfile.TemporaryDirectory() as tmpdirname:
							config_first.save_pretrained(lowerCAmelCase_		)
							UpperCAmelCase_  :				Tuple      =      self.config_class.from_pretrained(lowerCAmelCase_		)

				self.parent.assertEqual(config_second.to_dict()			,			config_first.to_dict()		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				int		)		->    Optional[Any]:
				UpperCAmelCase_  :				Any      =      self.config_class(**self.inputs_dict		)

				UpperCAmelCase_  :				int      =      "test"
				with tempfile.TemporaryDirectory() as tmpdirname:
							UpperCAmelCase_  :				Optional[int]      =      os.path.join(lowerCAmelCase_			,			lowerCAmelCase_		)
							config_first.save_pretrained(lowerCAmelCase_		)
							UpperCAmelCase_  :				Union[str, Any]      =      self.config_class.from_pretrained(lowerCAmelCase_			,			subfolder=lowerCAmelCase_		)

				self.parent.assertEqual(config_second.to_dict()			,			config_first.to_dict()		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				Optional[Any]		)		->    Tuple:
				UpperCAmelCase_  :				List[str]      =      self.config_class(**self.inputs_dict			,			num_labels=5		)
				self.parent.assertEqual(len(config.idalabel		)			,			5		)
				self.parent.assertEqual(len(config.labelaid		)			,			5		)

				UpperCAmelCase_  :				List[Any]      =      3
				self.parent.assertEqual(len(config.idalabel		)			,			3		)
				self.parent.assertEqual(len(config.labelaid		)			,			3		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				Tuple		)		->    Optional[Any]:
				if self.config_class.is_composition:
							return
				UpperCAmelCase_  :				str      =      self.config_class()
				self.parent.assertIsNotNone(lowerCAmelCase_		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				Dict		)		->    List[str]:
				UpperCAmelCase_  :				Optional[int]      =      copy.deepcopy(lowerCAmelCase_		)
				UpperCAmelCase_  :				Optional[int]      =      self.config_class(**lowerCAmelCase_		)
				UpperCAmelCase_  :				Optional[Any]      =      []
				for key, value in config_common_kwargs.items():
							if key == "torch_dtype":
										if not is_torch_available():
													continue
										else:
													import torch

													if config.torch_dtype != torch.floataa:
																wrong_values.append(("torch_dtype", config.torch_dtype, torch.floataa)		)
							elif getattr(lowerCAmelCase_			,			lowerCAmelCase_		) != value:
										wrong_values.append((key, getattr(lowerCAmelCase_			,			lowerCAmelCase_		), value)		)

				if len(lowerCAmelCase_		) > 0:
							UpperCAmelCase_  :				Any      =      "\n".join([f"""- {v[0]}: got {v[1]} instead of {v[2]}""" for v in wrong_values]		)
							raise ValueError(f"""The following keys were not properly set in the config:\n{errors}"""		)





	def   _SCREAMING_SNAKE_CASE   (   self      :				List[Any]		)		->    str:
				self.create_and_test_config_common_properties()
				self.create_and_test_config_to_json_string()
				self.create_and_test_config_to_json_file()
				self.create_and_test_config_from_and_save_pretrained()
				self.create_and_test_config_from_and_save_pretrained_subfolder()
				self.create_and_test_config_with_num_labels()
				self.check_config_can_be_init_without_params()
				self.check_config_arguments_init()



