Fix unit tests in test_arguments.py and test_configuration.py

This commit is contained in:
Gerald Lonlas 2018-03-04 00:58:20 -08:00
parent 2001c20426
commit c52e688979
2 changed files with 3 additions and 3 deletions

View File

@ -129,5 +129,5 @@ def test_parse_args_hyperopt_custom() -> None:
assert call_args.epochs == 20 assert call_args.epochs == 20
assert call_args.loglevel == logging.INFO assert call_args.loglevel == logging.INFO
assert call_args.subparser == 'hyperopt' assert call_args.subparser == 'hyperopt'
assert call_args.spaces == 'buy' assert call_args.spaces == ['buy']
assert call_args.func is not None assert call_args.func is not None

View File

@ -281,5 +281,5 @@ def test_hyperopt_space_argument(mocker, default_conf, caplog) -> None:
configuration = Configuration(args) configuration = Configuration(args)
config = configuration.get_config() config = configuration.get_config()
assert 'spaces' in config assert 'spaces' in config
assert config['spaces'] is 'all' assert config['spaces'] == ['all']
assert tt.log_has('Parameter -s/--spaces detected: all', caplog.record_tuples) assert tt.log_has('Parameter -s/--spaces detected: [\'all\']', caplog.record_tuples)