def test_command_usage6():
cmd = Command('train', options=[
Option(name='a', type=str, default="A"),
Option(name='b', type=str, default="B"),
Option(name='c', type=str, default="C"),
])
assert cmd.usage() == USAGE_6
def test_command_usage5():
cmd = Command('train',
options=[
Option(name='optimizer', type=str),
Option(name='learning-rate', default=0.0001, short='l')
])
assert cmd.usage() == USAGE_5
def test_command_usage9():
cmd = Command('predict', options=[
Option(name='<file>', type='Optional[str]', descr="The file to use when predicting."),
Option(name='a', type=str, default="A"),
Option(name='b', type=str, default="B"),
Option(name='c', type=str, default="C"),
])
assert cmd.usage() == USAGE_9
def test_command_usage8():
cmd = Command('predict', options=[
Option(name='<file>', type='Optional[str]'),
Option(name='a', type=str, default="A"),
Option(name='b', type=str, default="B"),
Option(name='c', type=str, default="C"),
])
assert cmd.usage() == USAGE_8
def test_command_usage11():
cmd = Command('predict',
options=[
Option(name='@AIs', type="List[AI]"),
Option(name='threshold',
default=0.2,
descr="Prediction Threshold.")
])
assert cmd.usage() == USAGE_11
def test_command_usage4():
cmd = Command('predict',
options=[
Option(name='@AI'),
Option(name='threshold',
default=0.2,
descr="Prediction Threshold.")
])
assert cmd.usage() == USAGE_4
def test_command_usage13():
cmd = Command('predict',
long_descr="Make a prediction.",
examples="ml @skynet predict",
options=[
Option(name='@AIs', type="List[AI]"),
Option(name='threshold',
default=0.2,
descr="Prediction Threshold.")
])
assert cmd.usage() == USAGE_13
def test_command_usage2():
cmd = Command('train',
options=[Option(name='learning-rate', default=0.0001)])
assert cmd.usage() == USAGE_2
def test_command_usage1():
cmd = Command('new', options=[Option(name='<project-name>')])
assert cmd.usage() == USAGE_1
