def test_from_argparse_ast_empty(self) -> None:
"""
Tests `argparse_ast` empty condition
"""
self.assertEqual(
emit.to_code(
emit.argparse_function(
parse.argparse_ast(
FunctionDef(
body=[],
arguments_args=None,
identifier_name=None,
stmt=None,
)),
emit_default_doc=True,
)).rstrip("\n"),
"def set_cli_args(argument_parser):\n{tab}{body}".format(
tab=tab,
body=tab.join((
'"""\n',
"Set CLI arguments\n\n",
":param argument_parser: argument parser\n",
":type argument_parser: ```ArgumentParser```\n\n",
":returns: argument_parser\n",
":rtype: ```ArgumentParser```\n",
'"""\n',
"argument_parser.description = ''\n",
"return argument_parser",
)),
),
)
def test_to_argparse(self) -> None:
"""
Tests whether `to_argparse` produces `argparse_func_ast` given `class_ast`
"""
run_ast_test(
self,
emit.argparse_function(
parse.class_(class_ast),
emit_default_doc=False,
emit_default_doc_in_return=False,
),
gold=argparse_func_ast,
)
def test_from_torch_ir_to_argparse(self) -> None:
""" Tests if emission of class from torch IR is as expected """
func = emit.argparse_function(
deepcopy(class_torch_nn_l1loss_ir),
emit_default_doc=False,
emit_default_doc_in_return=False,
)
run_ast_test(
self,
func,
argparse_func_torch_nn_l1loss_ast,
)
def test_to_argparse_func_nargs(self) -> None:
"""
Tests whether an argparse function is generated with `action="append"` set properly
"""
run_ast_test(
self,
gen_ast=emit.argparse_function(
parse.class_(class_nargs_ast),
emit_default_doc=False,
emit_default_doc_in_return=False,
function_name="set_cli_action_append",
),
gold=argparse_func_action_append_ast,
)
def test_from_argparse_with_extra_body_to_argparse_with_extra_body(self) -> None:
""" Tests if this can make the roundtrip from a full argparse function to a argparse full function """
ir = parse.argparse_ast(argparse_func_with_body_ast)
func = emit.argparse_function(
ir,
emit_default_doc=False,
emit_default_doc_in_return=False,
)
run_ast_test(
self,
func,
argparse_func_with_body_ast,
)
def test_to_argparse_google_tf_tensorboard(self) -> None:
"""
Tests whether `to_argparse` produces `argparse_function_google_tf_tensorboard_ast`
given `class_google_tf_tensorboard_ast`
"""
run_ast_test(
self,
emit.argparse_function(
parse.class_(
class_google_tf_tensorboard_ast, merge_inner_function="__init__"
),
emit_default_doc=False,
emit_default_doc_in_return=False,
),
gold=argparse_function_google_tf_tensorboard_ast,
)