
import re




def 							__SCREAMING_SNAKE_CASE       (		SCREAMING_SNAKE_CASE	:				int      )					->						Union[str, Any]:
				def docstring_decorator(SCREAMING_SNAKE_CASE	:				Union[str, Any]      ):
								__lowercase								= fn.__doc__
								__lowercase								= func_doc.split('\n'      )
								__lowercase								= 0
								while i < len(SCREAMING_SNAKE_CASE      ) and re.search(R'^\s*Examples?:\s*$'   ,							lines[i]      ) is None:
												i += 1
								if i < len(SCREAMING_SNAKE_CASE      ):
												__lowercase								= example_docstring
												__lowercase								= '\n'.join(SCREAMING_SNAKE_CASE      )
								else:
												raise ValueError(
												    F"""The function {fn} should have an empty 'Examples:' in its docstring as placeholder, """
												    F"""current docstring is:\n{func_doc}"""      )
								__lowercase								= func_doc
								return fn

				return docstring_decorator







import unittest

from transformers import load_tool
from transformers.tools.agent_types import AGENT_TYPE_MAPPING

from .test_tools_common import ToolTesterMixin, output_types
class      A__       (						unittest.TestCase   ,						lowerCAmelCase__    ):





							def       a__			(			self  :       Any    )						->		Dict:



											"""simple docstring"""


											__lowercase								= load_tool('translation'    )
											self.tool.setup()
											__lowercase								= load_tool('translation'					,			remote=_UpperCAmelCase    )





							def       a__			(			self  :       int    )						->		Optional[Any]:



											"""simple docstring"""


											__lowercase								= self.tool('Hey, what\'s up?'					,			src_lang='English'					,			tgt_lang='French'    )
											self.assertEqual(_UpperCAmelCase					,			'- Hé, comment ça va?'    )





							def       a__			(			self  :       Tuple    )						->		Optional[int]:



											"""simple docstring"""


											__lowercase								= self.remote_tool('Hey, what\'s up?'					,			src_lang='English'					,			tgt_lang='French'    )
											self.assertEqual(_UpperCAmelCase					,			'- Hé, comment ça va?'    )





							def       a__			(			self  :       Union[str, Any]    )						->		Tuple:



											"""simple docstring"""


											__lowercase								= self.tool(text='Hey, what\'s up?'					,			src_lang='English'					,			tgt_lang='French'    )
											self.assertEqual(_UpperCAmelCase					,			'- Hé, comment ça va?'    )





							def       a__			(			self  :       Tuple    )						->		Optional[Any]:



											"""simple docstring"""


											__lowercase								= self.remote_tool(text='Hey, what\'s up?'					,			src_lang='English'					,			tgt_lang='French'    )
											self.assertEqual(_UpperCAmelCase					,			'- Hé, comment ça va?'    )





							def       a__			(			self  :       List[Any]    )						->		int:



											"""simple docstring"""


											__lowercase								= ['Hey, what\'s up?', 'English', 'Spanish']
											__lowercase								= self.tool(*_UpperCAmelCase    )

											# There is a single output
											if len(self.tool.outputs    ) == 1:
															__lowercase								= [outputs]

											self.assertListEqual(output_types(_UpperCAmelCase    )					,			self.tool.outputs    )





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



											"""simple docstring"""


											__lowercase								= ['Hey, what\'s up?', 'English', 'Spanish']
											__lowercase								= self.tool(*_UpperCAmelCase    )

											if not isinstance(_UpperCAmelCase					,			_UpperCAmelCase    ):
															__lowercase								= [outputs]

											self.assertEqual(len(_UpperCAmelCase    )					,			len(self.tool.outputs    )    )

											for output, output_type in zip(_UpperCAmelCase					,			self.tool.outputs    ):
															__lowercase								= AGENT_TYPE_MAPPING[output_type]
															self.assertTrue(isinstance(_UpperCAmelCase					,			_UpperCAmelCase    )    )





							def       a__			(			self  :       Tuple    )						->		List[Any]:



											"""simple docstring"""


											__lowercase								= ['Hey, what\'s up?', 'English', 'Spanish']

											__lowercase								= []

											for _input, input_type in zip(_UpperCAmelCase					,			self.tool.inputs    ):
															if isinstance(_UpperCAmelCase					,			_UpperCAmelCase    ):
																			_inputs.append([AGENT_TYPE_MAPPING[_input_type](_input    ) for _input_type in input_type]    )
															else:
																			_inputs.append(AGENT_TYPE_MAPPING[input_type](_input    )    )

        # Should not raise an error
											__lowercase								= self.tool(*_UpperCAmelCase    )

											if not isinstance(_UpperCAmelCase					,			_UpperCAmelCase    ):
															__lowercase								= [outputs]

											self.assertEqual(len(_UpperCAmelCase    )					,			len(self.tool.outputs    )    )





