def test_should_set_channels_to_none_by_default(self):
args = model_args_parser().parse_args([])
assert args.channels is None
def test_should_allow_all_base_loss_options(self):
for base_loss in ALL_BASE_LOSS:
args = model_args_parser().parse_args(['--base_loss', base_loss])
assert args.base_loss == base_loss
def test_should_parse_channels(self):
args = model_args_parser().parse_args(['--channels', 'a,b,c'])
assert args.channels == ['a', 'b', 'c']