def test_recommendations_single(self):
"""Test whether recommended args work for non-group."""
parser = ParlaiParser()
parser.add_argument(
'-bs',
'--batchsize',
default=1,
type=int,
help='batch size for minibatch training schemes',
recommended="10",
)
with testing_utils.capture_output() as _:
parser.parse_args()
help_str = parser.format_help()
assert re.search(r'--batchsize[^\n]*\n[^\n]*\(recommended: 10\)', help_str)
def test_recommendations_single(self):
"""
Test whether recommended args work for non-group.
"""
parser = ParlaiParser(False, False)
parser.add_argument(
'-bs',
'--batchsize',
default=1,
type=int,
help='batch size for minibatch training schemes',
recommended=1337,
)
parser.parse_args([])
help_str = parser.format_help()
assert 'recommended:' in help_str
assert '1337' in help_str
def test_recommendations_group(self):
"""Test whether recommended args work for a group."""
parser = ParlaiParser()
parser_grp = parser.add_argument_group('Test Group')
parser_grp.add_argument(
'-bs',
'--batchsize',
default=1,
type=int,
help='batch size for minibatch training schemes',
recommended=[5, 10, 15],
)
with testing_utils.capture_output() as _:
parser.parse_args()
help_str = parser.format_help()
assert re.search(r'Test Group:\n', help_str)
assert re.search(
r'--batchsize[^\n]*\n[^\n]*\(recommended: \[5, 10, 15\]\)', help_str
)
def test_recommendations_group(self):
"""
Test whether recommended args work for a group.
"""
parser = ParlaiParser(False, False)
parser_grp = parser.add_argument_group('Test Group')
parser_grp.add_argument(
'-bs',
'--batchsize',
default=1,
type=int,
help='batch size for minibatch training schemes',
recommended=1337,
)
parser.parse_args([])
help_str = parser.format_help()
assert 'Test Group' in help_str
_, latter = help_str.split('Test Group')
assert 'recommended:' in latter
assert '1337' in latter
