def test_add_predict_subparser(self):
parser = argparse.ArgumentParser(description="Testing")
subparsers = parser.add_subparsers(title='Commands', metavar='')
add_subparser(subparsers, DEFAULT_PREDICTORS)
raw_args = ["predict", # command
"/path/to/archive", # archive
"/dev/null", # input_file
"--output-file", "/dev/null",
"--silent"]
args = parser.parse_args(raw_args)
assert args.func.__name__ == 'predict_inner'
assert args.archive_file == "/path/to/archive"
assert args.silent
def main(raw_args: Sequence[str]) -> None:
parser = argparse.ArgumentParser(description="Run AllenNLP",
usage='%(prog)s [command]')
subparsers = parser.add_subparsers(title='Commands', metavar='')
# Add sub-commands
predict.add_subparser(subparsers)
train.add_subparser(subparsers)
serve.add_subparser(subparsers)
evaluate.add_subparser(subparsers)
args = parser.parse_args(raw_args)
# If a subparser is triggered, it adds its work as `args.func`.
# So if no such attribute has been added, no subparser was triggered,
# so give the user some help.
if 'func' in dir(args):
args.func(args)
else:
parser.print_help()
def test_add_predict_subparser(self):
parser = argparse.ArgumentParser(description="Testing")
subparsers = parser.add_subparsers(title='Commands', metavar='')
add_subparser(subparsers)
raw_args = [
"predict", # command
"/path/to/model", # model
"/path/to/data", # input_file
"--output-file",
"outfile",
"--print"
]
args = parser.parse_args(raw_args)
assert args.func == predict
assert args.config_file == "/path/to/model"
assert args.input_file == "/path/to/data"
assert args.output_file == "outfile"
assert args.print