def test_filter_signature_with_PARAMS_TO_REMOVE(self):
def test_method(self, credentials: Optional, sync, init_param: str):
pass
filtered_signature = utils.filter_signature(signature(test_method))
expected_output_str = "<Signature (init_param: str)>"
self.assertEqual(repr(filtered_signature), expected_output_str)
def test_filter_signature_with_resouce_name(self):
def test_method(model_name: str):
pass
param_map = {}
filtered_signature = utils.filter_signature(
signature=signature(test_method),
is_init_signature=True,
self_type=str,
component_param_name_to_mb_sdk_param_name=param_map)
expected_output_str = "<Signature (model: str)>"
self.assertEqual(repr(filtered_signature), expected_output_str)
def test_generate_docstring_with_PARAMS_TO_REMOVE(self):
args_dict = {'input': 'input_name', 'input_param': 'input_param'}
param_map = {}
converted_method_signature = utils.filter_signature(
signature(self._test_method),
is_init_signature=True,
component_param_name_to_mb_sdk_param_name=param_map)
results = utils.generate_docstring(
args_dict=args_dict,
signature=converted_method_signature,
method_docstring=self._test_method.__doc__)
expected_results = "".join("Test short description.\n"
"Long descirption\nArgs:\n"
" input:\n"
" input_name\n"
" input_param:\n"
" input_param\n")
self.assertEqual(results, expected_results)
