

'''simple docstring'''
import dataclasses
import re
import string
from typing import Any, Dict, Iterator, List, Mapping, Optional, Sequence, Tuple

import numpy as np

from . import residue_constants


lowerCamelCase :      int					 =     Mapping[str, np.ndarray]
lowerCamelCase :      Union[str, Any]					 =     Mapping[str, Any]  # Is a nested dict.
lowerCamelCase :      Dict					 =     0.0_1







@dataclasses.dataclass(frozen=A__  )
class 							A__    :
  A__		 =				42  # [num_res, num_atom_type, 3]

  # Amino-acid type for each residue represented as an integer between 0 and
  # 20, where 20 is 'X'.
  A__		 =				42  # [num_res]

  # Binary float mask to indicate presence of a particular atom. 1.0 if an atom
  # is present and 0.0 if not. This should be used for loss masking.
  A__		 =				42  # [num_res, num_atom_type]

  # Residue index as used in PDB. It is not necessarily continuous or 0-indexed.
  A__		 =				42  # [num_res]

  # B-factors, or temperature factors, of each residue (in sq. angstroms units),
  # representing the displacement of the residue from its ground truth mean
  # value.
  A__		 =				42  # [num_res, num_atom_type]

  # Chain indices for multi-chain predictions
  A__		 =				None

  # Optional remark about the protein. Included as a comment in output PDB
  # files
  A__		 =				None

  # Templates used to generate this protein (prediction-only)
  A__		 =				None

  # Chain corresponding to each parent
  A__		 =				None


def        _lowerCAmelCase					(						_UpperCamelCase						:     str				)      ->			Protein:
     """simple docstring"""




     _SCREAMING_SNAKE_CASE							    =r'(\[[A-Z]+\]\n)'
     _SCREAMING_SNAKE_CASE							    =[tag.strip() for tag in re.split(_UpperCamelCase							, _UpperCamelCase				) if len(_UpperCamelCase				) > 0]
     _SCREAMING_SNAKE_CASE							    =zip(tags[0::2]							, [l.split('\n'				) for l in tags[1::2]]				)

     _SCREAMING_SNAKE_CASE							    =["N", "CA", "C"]
     _SCREAMING_SNAKE_CASE							    =None
     _SCREAMING_SNAKE_CASE							    =None
     _SCREAMING_SNAKE_CASE							    =None
     for g in groups:
          if "[PRIMARY]" == g[0]:
               _SCREAMING_SNAKE_CASE							    =g[1][0].strip()
               for i in range(len(_UpperCamelCase				)				):
                    if seq[i] not in residue_constants.restypes:
                         _SCREAMING_SNAKE_CASE							    ='X'  # FIXME: strings are immutable
               _SCREAMING_SNAKE_CASE							    =np.array(
                   [residue_constants.restype_order.get(_UpperCamelCase							, residue_constants.restype_num				) for res_symbol in seq]				)
          elif "[TERTIARY]" == g[0]:
               _SCREAMING_SNAKE_CASE							    =[]
               for axis in range(3				):
                    tertiary.append(list(map(_UpperCamelCase							, g[1][axis].split()				)				)				)
               _SCREAMING_SNAKE_CASE							    =np.array(_UpperCamelCase				)
               _SCREAMING_SNAKE_CASE							    =np.zeros((len(tertiary[0]				) // 3, residue_constants.atom_type_num, 3)				).astype(np.floataa				)
               for i, atom in enumerate(_UpperCamelCase				):
                    _SCREAMING_SNAKE_CASE							    =np.transpose(tertiary_np[:, i::3]				)
               atom_positions *= PICO_TO_ANGSTROM
          elif "[MASK]" == g[0]:
               _SCREAMING_SNAKE_CASE							    =np.array(list(map({'-': 0, '+': 1}.get							, g[1][0].strip()				)				)				)
               _SCREAMING_SNAKE_CASE							    =np.zeros(
                   (
                       len(_UpperCamelCase				),
                       residue_constants.atom_type_num,
                   )				).astype(np.floataa				)
               for i, atom in enumerate(_UpperCamelCase				):
                    _SCREAMING_SNAKE_CASE							    =1
               atom_mask *= mask[..., None]

     assert aatype is not None

     return Protein(
         atom_positions=_UpperCamelCase							, atom_mask=_UpperCamelCase							, aatype=_UpperCamelCase							, residue_index=np.arange(len(_UpperCamelCase				)				)							, b_factors=_UpperCamelCase							, )


def        _lowerCAmelCase					(						_UpperCamelCase						:     Protein							, _UpperCamelCase						:     int = 0				)      ->			List[str]:
     """simple docstring"""




     _SCREAMING_SNAKE_CASE							    =[]

     _SCREAMING_SNAKE_CASE							    =prot.remark
     if remark is not None:
          pdb_headers.append(f"REMARK {remark}"				)

     _SCREAMING_SNAKE_CASE							    =prot.parents
     _SCREAMING_SNAKE_CASE							    =prot.parents_chain_index
     if parents is not None and parents_chain_index is not None:
          _SCREAMING_SNAKE_CASE							    =[p for i, p in zip(_UpperCamelCase							, _UpperCamelCase				) if i == chain_id]

     if parents is None or len(_UpperCamelCase				) == 0:
          _SCREAMING_SNAKE_CASE							    =['N/A']

     pdb_headers.append(f"PARENT {' '.join(_UpperCamelCase				)}"				)

     return pdb_headers


def        _lowerCAmelCase					(						_UpperCamelCase						:     Protein							, _UpperCamelCase						:     str				)      ->			str:
     """simple docstring"""




     _SCREAMING_SNAKE_CASE							    =[]
     _SCREAMING_SNAKE_CASE							    =pdb_str.split('\n'				)

     _SCREAMING_SNAKE_CASE							    =prot.remark
     if remark is not None:
          out_pdb_lines.append(f"REMARK {remark}"				)

     _SCREAMING_SNAKE_CASE							    =42
     if prot.parents is not None and len(prot.parents				) > 0:
          _SCREAMING_SNAKE_CASE							    =[]
          if prot.parents_chain_index is not None:
               _SCREAMING_SNAKE_CASE							    ={}
               for p, i in zip(prot.parents							, prot.parents_chain_index				):
                    parent_dict.setdefault(str(_UpperCamelCase				)							, []				)
                    parent_dict[str(_UpperCamelCase				)].append(_UpperCamelCase				)

               _SCREAMING_SNAKE_CASE							    =max([int(_UpperCamelCase				) for chain_idx in parent_dict]				)
               for i in range(max_idx + 1				):
                    _SCREAMING_SNAKE_CASE							    =parent_dict.get(str(_UpperCamelCase				)							, ['N/A']				)
                    parents_per_chain.append(_UpperCamelCase				)
          else:
               parents_per_chain.append(list(prot.parents				)				)
     else:
          _SCREAMING_SNAKE_CASE							    =[['N/A']]

     def make_parent_line(_UpperCamelCase						:     Sequence[str]				) -> str:
          return f"PARENT {' '.join(_UpperCamelCase				)}"

     out_pdb_lines.append(make_parent_line(parents_per_chain[0]				)				)

     _SCREAMING_SNAKE_CASE							    =0
     for i, l in enumerate(_UpperCamelCase				):
          if "PARENT" not in l and "REMARK" not in l:
               out_pdb_lines.append(_UpperCamelCase				)
          if "TER" in l and "END" not in lines[i + 1]:
               chain_counter += 1
               if not chain_counter >= len(_UpperCamelCase				):
                    _SCREAMING_SNAKE_CASE							    =parents_per_chain[chain_counter]
               else:
                    _SCREAMING_SNAKE_CASE							    =['N/A']

               out_pdb_lines.append(make_parent_line(_UpperCamelCase				)				)

     return "\n".join(_UpperCamelCase				)


def        _lowerCAmelCase					(						_UpperCamelCase						:     Protein				)      ->			str:
     """simple docstring"""




     _SCREAMING_SNAKE_CASE							    =residue_constants.restypes + ['X']

     def res_atoa(_UpperCamelCase						:     int				) -> str:
          return residue_constants.restype_atoa.get(restypes[r]							, 'UNK'				)

     _SCREAMING_SNAKE_CASE							    =residue_constants.atom_types

     _SCREAMING_SNAKE_CASE							    =[]

     _SCREAMING_SNAKE_CASE							    =prot.atom_mask
     _SCREAMING_SNAKE_CASE							    =prot.aatype
     _SCREAMING_SNAKE_CASE							    =prot.atom_positions
     _SCREAMING_SNAKE_CASE							    =prot.residue_index.astype(np.intaa				)
     _SCREAMING_SNAKE_CASE							    =prot.b_factors
     _SCREAMING_SNAKE_CASE							    =prot.chain_index

     if np.any(aatype > residue_constants.restype_num				):
          raise ValueError('Invalid aatypes.'				)

     _SCREAMING_SNAKE_CASE							    =get_pdb_headers(_UpperCamelCase				)
     if len(_UpperCamelCase				) > 0:
          pdb_lines.extend(_UpperCamelCase				)

     _SCREAMING_SNAKE_CASE							    =aatype.shape[0]
     _SCREAMING_SNAKE_CASE							    =1
     _SCREAMING_SNAKE_CASE							    =0
     _SCREAMING_SNAKE_CASE							    =string.ascii_uppercase
     _SCREAMING_SNAKE_CASE							    =None
     # Add all atom sites.
     for i in range(_UpperCamelCase				):
          _SCREAMING_SNAKE_CASE							    =res_atoa(aatype[i]				)
          for atom_name, pos, mask, b_factor in zip(_UpperCamelCase							, atom_positions[i]							, atom_mask[i]							, b_factors[i]				):
               if mask < 0.5:
                    continue

               _SCREAMING_SNAKE_CASE							    ='ATOM'
               _SCREAMING_SNAKE_CASE							    =atom_name if len(_UpperCamelCase				) == 4 else f" {atom_name}"
               _SCREAMING_SNAKE_CASE							    =''
               _SCREAMING_SNAKE_CASE							    =''
               _SCREAMING_SNAKE_CASE							    =1.00
               _SCREAMING_SNAKE_CASE							    =atom_name[0]  # Protein supports only C, N, O, S, this works.
               _SCREAMING_SNAKE_CASE							    =''

               _SCREAMING_SNAKE_CASE							    ='A'
               if chain_index is not None:
                    _SCREAMING_SNAKE_CASE							    =chain_tags[chain_index[i]]

               # PDB is a columnar format, every space matters here!
               _SCREAMING_SNAKE_CASE							    =(
                   f"{record_type:<6}{atom_index:>5} {name:<4}{alt_loc:>1}"
                   f"{res_name_a:>3} {chain_tag:>1}"
                   f"{residue_index[i]:>4}{insertion_code:>1}   "
                   f"{pos[0]:>8.3f}{pos[1]:>8.3f}{pos[2]:>8.3f}"
                   f"{occupancy:>6.2f}{b_factor:>6.2f}          "
                   f"{element:>2}{charge:>2}"
               )
               pdb_lines.append(_UpperCamelCase				)
               atom_index += 1

          _SCREAMING_SNAKE_CASE							    =i == n - 1
          if chain_index is not None:
               if i != n - 1 and chain_index[i + 1] != prev_chain_index:
                    _SCREAMING_SNAKE_CASE							    =True
                    _SCREAMING_SNAKE_CASE							    =chain_index[i + 1]

          if should_terminate:
               # Close the chain.
               _SCREAMING_SNAKE_CASE							    ='TER'
               _SCREAMING_SNAKE_CASE							    =(
                   f"{chain_end:<6}{atom_index:>5}      {res_atoa(aatype[i]				):>3} {chain_tag:>1}{residue_index[i]:>4}"
               )
               pdb_lines.append(_UpperCamelCase				)
               atom_index += 1

               if i != n - 1:
                    # "prev" is a misnomer here. This happens at the beginning of
                    # each new chain.
                    pdb_lines.extend(get_pdb_headers(_UpperCamelCase							, _UpperCamelCase				)				)

     pdb_lines.append('END'				)
     pdb_lines.append(''				)
     return "\n".join(_UpperCamelCase				)


def        _lowerCAmelCase					(						_UpperCamelCase						:     Protein				)      ->			np.ndarray:
     """simple docstring"""




     return residue_constants.STANDARD_ATOM_MASK[prot.aatype]







def        _lowerCAmelCase					(						_UpperCamelCase						:     FeatureDict							, _UpperCamelCase						:     ModelOutput							, _UpperCamelCase						:     Optional[np.ndarray] = None							, _UpperCamelCase						:     Optional[np.ndarray] = None							, _UpperCamelCase						:     Optional[str] = None							, _UpperCamelCase						:     Optional[Sequence[str]] = None							, _UpperCamelCase						:     Optional[Sequence[int]] = None							, )      ->			Protein:
     """simple docstring"""




     return Protein(
         aatype=features['aatype']							, atom_positions=result['final_atom_positions']							, atom_mask=result['final_atom_mask']							, residue_index=features['residue_index'] + 1							, b_factors=b_factors if b_factors is not None else np.zeros_like(result['final_atom_mask']				)							, chain_index=_UpperCamelCase							, remark=_UpperCamelCase							, parents=_UpperCamelCase							, parents_chain_index=_UpperCamelCase							, )

'''simple docstring'''




import argparse
import json
import os
import sys
import tempfile
import unittest
from argparse import Namespace
from dataclasses import dataclass, field
from enum import Enum
from pathlib import Path
from typing import List, Literal, Optional

import yaml

from transformers import HfArgumentParser, TrainingArguments
from transformers.hf_argparser import make_choice_type_function, string_to_bool


# Since Python 3.10, we can use the builtin `|` operator for Union types
# See PEP 604: https://peps.python.org/pep-0604
_lowercase						: int						=						sys.version_info >= (3, 10)




def 							lowerCamelCase	(      UpperCAmelCase__       :					Tuple=None      ,      UpperCAmelCase__       :					Union[str, Any]=None )			->      Tuple:
							return field(default_factory=lambda: default      ,      metadata=UpperCAmelCase__ )





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       42
			UpperCamelCase__				=       42
			UpperCamelCase__				=       42
			UpperCamelCase__				=       42





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       42
			UpperCamelCase__				=       field(default='''toto''',     metadata={'''help''': '''help message'''})





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       False
			UpperCamelCase__				=       True
			UpperCamelCase__				=       None





class     __magic_name__						(    _UpperCAmelCase):
			UpperCamelCase__				=       '''titi'''
			UpperCamelCase__				=       '''toto'''





class     __magic_name__						(    _UpperCAmelCase):
			UpperCamelCase__				=       '''titi'''
			UpperCamelCase__				=       '''toto'''
			UpperCamelCase__				=       42





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       "toto"







			def 	SCREAMING_SNAKE_CASE_     (	self				:  int    ):
										lowercase_					:			Optional[int]										=       BasicEnum(self.foo    )





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       "toto"







			def 	SCREAMING_SNAKE_CASE_     (	self				:  int    ):
										lowercase_					:			Optional[int]										=       MixedTypeEnum(self.foo    )





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       None
			UpperCamelCase__				=       field(default=_UpperCAmelCase,     metadata={'''help''': '''help message'''})
			UpperCamelCase__				=       None
			UpperCamelCase__				=       list_field(default=[])
			UpperCamelCase__				=       list_field(default=[])





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       list_field(default=[])
			UpperCamelCase__				=       list_field(default=[1, 2, 3])
			UpperCamelCase__				=       list_field(default=['''Hallo''', '''Bonjour''', '''Hello'''])
			UpperCamelCase__				=       list_field(default=[0.1, 0.2, 0.3])





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       field()
			UpperCamelCase__				=       field()
			UpperCamelCase__				=       field()







			def 	SCREAMING_SNAKE_CASE_     (	self				:  List[str]    ):
										lowercase_					:			List[Any]										=       BasicEnum(self.required_enum    )





@dataclass
class     __magic_name__						:
			UpperCamelCase__				=       42
			UpperCamelCase__				=       field()
			UpperCamelCase__				=       None
			UpperCamelCase__				=       field(default='''toto''',     metadata={'''help''': '''help message'''})
			UpperCamelCase__				=       list_field(default=['''Hallo''', '''Bonjour''', '''Hello'''])


if is_python_no_less_than_3_10:





						@dataclass
						class     __magic_name__						:
									UpperCamelCase__				=       False
									UpperCamelCase__				=       True
									UpperCamelCase__				=       None





						@dataclass
						class     __magic_name__						:
									UpperCamelCase__				=       None
									UpperCamelCase__				=       field(default=_UpperCAmelCase,     metadata={'''help''': '''help message'''})
									UpperCamelCase__				=       None
									UpperCamelCase__				=       list_field(default=[])
									UpperCamelCase__				=       list_field(default=[])





class     __magic_name__						(    unittest.TestCase):







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Dict				,			lowercase_				:  argparse.ArgumentParser				,			lowercase_				:  argparse.ArgumentParser    ):
										self.assertEqual(len(a._actions    )				,			len(b._actions    )    )
										for x, y in zip(a._actions				,			b._actions    ):
																	lowercase_					:			List[str]										=       {k: v for k, v in vars(lowercase_    ).items() if k != """container"""}
																	lowercase_					:			Any										=       {k: v for k, v in vars(lowercase_    ).items() if k != """container"""}

																	# Choices with mixed type have custom function as "type"
																	# So we need to compare results directly for equality
																	if xx.get("""choices"""				,			lowercase_    ) and yy.get("""choices"""				,			lowercase_    ):
																								for expected_choice in yy["choices"] + xx["choices"]:
																															self.assertEqual(xx["""type"""](lowercase_    )				,			yy["""type"""](lowercase_    )    )
																								del xx["type"], yy["type"]

																	self.assertEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Tuple    ):
										lowercase_					:			List[str]										=       HfArgumentParser(lowercase_    )

										lowercase_					:			str										=       argparse.ArgumentParser()
										expected.add_argument("""--foo"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument("""--bar"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument("""--baz"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument("""--flag"""				,			type=lowercase_				,			default=lowercase_				,			const=lowercase_				,			nargs="""?"""    )
										self.argparsersEqual(lowercase_				,			lowercase_    )

										lowercase_					:			List[str]										=       ["""--foo""", """1""", """--baz""", """quux""", """--bar""", """0.5"""]
										((lowercase_)						,						)					:			Optional[Any]										=       parser.parse_args_into_dataclasses(lowercase_				,			look_for_args_file=lowercase_    )
										self.assertFalse(example.flag    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Optional[int]    ):
										lowercase_					:			Any										=       HfArgumentParser(lowercase_    )

										lowercase_					:			str										=       argparse.ArgumentParser()
										expected.add_argument("""--foo"""				,			default=42				,			type=lowercase_    )
										expected.add_argument("""--baz"""				,			default="""toto"""				,			type=lowercase_				,			help="""help message"""    )
										self.argparsersEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  int    ):
										lowercase_					:			Optional[int]										=       argparse.ArgumentParser()
										expected.add_argument("""--foo"""				,			type=lowercase_				,			default=lowercase_				,			const=lowercase_				,			nargs="""?"""    )
										expected.add_argument("""--baz"""				,			type=lowercase_				,			default=lowercase_				,			const=lowercase_				,			nargs="""?"""    )
										# A boolean no_* argument always has to come after its "default: True" regular counter-part
										# and its default must be set to False
										expected.add_argument("""--no_baz"""				,			action="""store_false"""				,			default=lowercase_				,			dest="""baz"""    )
										expected.add_argument("""--opt"""				,			type=lowercase_				,			default=lowercase_    )

										lowercase_					:			List[Any]										=       [WithDefaultBoolExample]
										if is_python_no_less_than_3_10:
																	dataclass_types.append(lowercase_    )

										for dataclass_type in dataclass_types:
																	lowercase_					:			List[str]										=       HfArgumentParser(lowercase_    )
																	self.argparsersEqual(lowercase_				,			lowercase_    )

																	lowercase_					:			str										=       parser.parse_args([]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			baz=lowercase_				,			opt=lowercase_    )    )

																	lowercase_					:			Optional[Any]										=       parser.parse_args(["""--foo""", """--no_baz"""]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			baz=lowercase_				,			opt=lowercase_    )    )

																	lowercase_					:			Optional[Any]										=       parser.parse_args(["""--foo""", """--baz"""]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			baz=lowercase_				,			opt=lowercase_    )    )

																	lowercase_					:			Dict										=       parser.parse_args(["""--foo""", """True""", """--baz""", """True""", """--opt""", """True"""]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			baz=lowercase_				,			opt=lowercase_    )    )

																	lowercase_					:			int										=       parser.parse_args(["""--foo""", """False""", """--baz""", """False""", """--opt""", """False"""]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			baz=lowercase_				,			opt=lowercase_    )    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Union[str, Any]    ):
										lowercase_					:			str										=       HfArgumentParser(lowercase_    )

										lowercase_					:			Tuple										=       argparse.ArgumentParser()
										expected.add_argument(
										    """--foo"""				,			default="""toto"""				,			choices=["""titi""", """toto""", 42]				,			type=make_choice_type_function(["""titi""", """toto""", 42]    )				,			)
										self.argparsersEqual(lowercase_				,			lowercase_    )

										lowercase_					:			Optional[int]										=       parser.parse_args([]    )
										self.assertEqual(args.foo				,			"""toto"""    )
										lowercase_					:			Union[str, Any]										=       parser.parse_args_into_dataclasses([]    )[0]
										self.assertEqual(enum_ex.foo				,			MixedTypeEnum.toto    )

										lowercase_					:			int										=       parser.parse_args(["""--foo""", """titi"""]    )
										self.assertEqual(args.foo				,			"""titi"""    )
										lowercase_					:			int										=       parser.parse_args_into_dataclasses(["""--foo""", """titi"""]    )[0]
										self.assertEqual(enum_ex.foo				,			MixedTypeEnum.titi    )

										lowercase_					:			List[str]										=       parser.parse_args(["""--foo""", """42"""]    )
										self.assertEqual(args.foo				,			42    )
										lowercase_					:			List[str]										=       parser.parse_args_into_dataclasses(["""--foo""", """42"""]    )[0]
										self.assertEqual(enum_ex.foo				,			MixedTypeEnum.fourtytwo    )







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





										@dataclass
										class     __magic_name__						:
													UpperCamelCase__				=       "toto"




										lowercase_					:			Optional[int]										=       HfArgumentParser(lowercase_    )

										lowercase_					:			List[str]										=       argparse.ArgumentParser()
										expected.add_argument(
										    """--foo"""				,			default="""toto"""				,			choices=("""titi""", """toto""", 42)				,			type=make_choice_type_function(["""titi""", """toto""", 42]    )				,			)
										self.argparsersEqual(lowercase_				,			lowercase_    )

										lowercase_					:			List[str]										=       parser.parse_args([]    )
										self.assertEqual(args.foo				,			"""toto"""    )

										lowercase_					:			Optional[int]										=       parser.parse_args(["""--foo""", """titi"""]    )
										self.assertEqual(args.foo				,			"""titi"""    )

										lowercase_					:			List[Any]										=       parser.parse_args(["""--foo""", """42"""]    )
										self.assertEqual(args.foo				,			42    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Dict    ):
										lowercase_					:			int										=       HfArgumentParser(lowercase_    )

										lowercase_					:			Dict										=       argparse.ArgumentParser()
										expected.add_argument("""--foo_int"""				,			nargs="""+"""				,			default=[]				,			type=lowercase_    )
										expected.add_argument("""--bar_int"""				,			nargs="""+"""				,			default=[1, 2, 3]				,			type=lowercase_    )
										expected.add_argument("""--foo_str"""				,			nargs="""+"""				,			default=["""Hallo""", """Bonjour""", """Hello"""]				,			type=lowercase_    )
										expected.add_argument("""--foo_float"""				,			nargs="""+"""				,			default=[0.1, 0.2, 0.3]				,			type=lowercase_    )

										self.argparsersEqual(lowercase_				,			lowercase_    )

										lowercase_					:			int										=       parser.parse_args([]    )
										self.assertEqual(
										    lowercase_				,			Namespace(foo_int=[]				,			bar_int=[1, 2, 3]				,			foo_str=["""Hallo""", """Bonjour""", """Hello"""]				,			foo_float=[0.1, 0.2, 0.3]    )				,			)

										lowercase_					:			Optional[int]										=       parser.parse_args("""--foo_int 1 --bar_int 2 3 --foo_str a b c --foo_float 0.1 0.7""".split()    )
										self.assertEqual(lowercase_				,			Namespace(foo_int=[1]				,			bar_int=[2, 3]				,			foo_str=["""a""", """b""", """c"""]				,			foo_float=[0.1, 0.7]    )    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Tuple    ):
										lowercase_					:			Tuple										=       argparse.ArgumentParser()
										expected.add_argument("""--foo"""				,			default=lowercase_				,			type=lowercase_    )
										expected.add_argument("""--bar"""				,			default=lowercase_				,			type=lowercase_				,			help="""help message"""    )
										expected.add_argument("""--baz"""				,			default=lowercase_				,			type=lowercase_    )
										expected.add_argument("""--ces"""				,			nargs="""+"""				,			default=[]				,			type=lowercase_    )
										expected.add_argument("""--des"""				,			nargs="""+"""				,			default=[]				,			type=lowercase_    )

										lowercase_					:			Optional[int]										=       [OptionalExample]
										if is_python_no_less_than_3_10:
																	dataclass_types.append(lowercase_    )

										for dataclass_type in dataclass_types:
																	lowercase_					:			Tuple										=       HfArgumentParser(lowercase_    )

																	self.argparsersEqual(lowercase_				,			lowercase_    )

																	lowercase_					:			Union[str, Any]										=       parser.parse_args([]    )
																	self.assertEqual(lowercase_				,			Namespace(foo=lowercase_				,			bar=lowercase_				,			baz=lowercase_				,			ces=[]				,			des=[]    )    )

																	lowercase_					:			List[Any]										=       parser.parse_args("""--foo 12 --bar 3.14 --baz 42 --ces a b c --des 1 2 3""".split()    )
																	self.assertEqual(lowercase_				,			Namespace(foo=12				,			bar=3.14				,			baz="""42"""				,			ces=["""a""", """b""", """c"""]				,			des=[1, 2, 3]    )    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Union[str, Any]    ):
										lowercase_					:			Dict										=       HfArgumentParser(lowercase_    )

										lowercase_					:			int										=       argparse.ArgumentParser()
										expected.add_argument("""--required_list"""				,			nargs="""+"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument("""--required_str"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument(
										    """--required_enum"""				,			type=make_choice_type_function(["""titi""", """toto"""]    )				,			choices=["""titi""", """toto"""]				,			required=lowercase_				,			)
										self.argparsersEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Optional[Any]    ):
										lowercase_					:			Dict										=       HfArgumentParser(lowercase_    )

										lowercase_					:			List[str]										=       argparse.ArgumentParser()
										expected.add_argument("""--foo"""				,			type=lowercase_				,			required=lowercase_    )
										expected.add_argument(
										    """--required_enum"""				,			type=make_choice_type_function(["""titi""", """toto"""]    )				,			choices=["""titi""", """toto"""]				,			required=lowercase_				,			)
										expected.add_argument("""--opt"""				,			type=lowercase_				,			default=lowercase_    )
										expected.add_argument("""--baz"""				,			default="""toto"""				,			type=lowercase_				,			help="""help message"""    )
										expected.add_argument("""--foo_str"""				,			nargs="""+"""				,			default=["""Hallo""", """Bonjour""", """Hello"""]				,			type=lowercase_    )
										self.argparsersEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  List[Any]    ):
										lowercase_					:			str										=       HfArgumentParser(lowercase_    )

										lowercase_					:			Optional[int]										=       {
										    """foo""": 12,
										    """bar""": 3.14,
										    """baz""": """42""",
										    """flag""": True,
										}

										lowercase_					:			Optional[Any]										=       parser.parse_dict(lowercase_    )[0]
										lowercase_					:			Dict										=       BasicExample(**lowercase_    )
										self.assertEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Optional[Any]    ):
										lowercase_					:			Dict										=       HfArgumentParser(lowercase_    )

										lowercase_					:			Optional[int]										=       {
										    """foo""": 12,
										    """bar""": 3.14,
										    """baz""": """42""",
										    """flag""": True,
										    """extra""": 42,
										}

										self.assertRaises(lowercase_				,			parser.parse_dict				,			lowercase_				,			allow_extra_keys=lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Any    ):
										lowercase_					:			List[Any]										=       HfArgumentParser(lowercase_    )

										lowercase_					:			int										=       {
										    """foo""": 12,
										    """bar""": 3.14,
										    """baz""": """42""",
										    """flag""": True,
										}
										with tempfile.TemporaryDirectory() as tmp_dir:
																	lowercase_					:			Optional[int]										=       os.path.join(lowercase_				,			"""temp_json"""    )
																	os.mkdir(lowercase_    )
																	with open(temp_local_path + """.json"""				,			"""w+"""    ) as f:
																								json.dump(lowercase_				,			lowercase_    )
																	lowercase_					:			str										=       parser.parse_yaml_file(Path(temp_local_path + """.json"""    )    )[0]

										lowercase_					:			Tuple										=       BasicExample(**lowercase_    )
										self.assertEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  List[Any]    ):
										lowercase_					:			str										=       HfArgumentParser(lowercase_    )

										lowercase_					:			List[Any]										=       {
										    """foo""": 12,
										    """bar""": 3.14,
										    """baz""": """42""",
										    """flag""": True,
										}
										with tempfile.TemporaryDirectory() as tmp_dir:
																	lowercase_					:			Tuple										=       os.path.join(lowercase_				,			"""temp_yaml"""    )
																	os.mkdir(lowercase_    )
																	with open(temp_local_path + """.yaml"""				,			"""w+"""    ) as f:
																								yaml.dump(lowercase_				,			lowercase_    )
																	lowercase_					:			List[str]										=       parser.parse_yaml_file(Path(temp_local_path + """.yaml"""    )    )[0]
										lowercase_					:			Any										=       BasicExample(**lowercase_    )
										self.assertEqual(lowercase_				,			lowercase_    )







			def 	SCREAMING_SNAKE_CASE_     (	self				:  Union[str, Any]    ):
										lowercase_					:			str										=       HfArgumentParser(lowercase_    )
										self.assertIsNotNone(lowercase_    )





