diff --git a/tests/optimize/test_hyperopt_tools.py b/tests/optimize/test_hyperopt_tools.py index cbcb13384..e845b2a77 100644 --- a/tests/optimize/test_hyperopt_tools.py +++ b/tests/optimize/test_hyperopt_tools.py @@ -64,34 +64,53 @@ def test_load_previous_results2(mocker, testdatadir, caplog) -> None: @pytest.mark.parametrize("spaces, expected_results", [ (['buy'], - {'buy': True, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': False}), + {'buy': True, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': False, + 'protection': False}), (['sell'], - {'buy': False, 'sell': True, 'roi': False, 'stoploss': False, 'trailing': False}), + {'buy': False, 'sell': True, 'roi': False, 'stoploss': False, 'trailing': False, + 'protection': False}), (['roi'], - {'buy': False, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False}), + {'buy': False, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False, + 'protection': False}), (['stoploss'], - {'buy': False, 'sell': False, 'roi': False, 'stoploss': True, 'trailing': False}), + {'buy': False, 'sell': False, 'roi': False, 'stoploss': True, 'trailing': False, + 'protection': False}), (['trailing'], - {'buy': False, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': True}), + {'buy': False, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': True, + 'protection': False}), (['buy', 'sell', 'roi', 'stoploss'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False, + 'protection': False}), (['buy', 'sell', 'roi', 'stoploss', 'trailing'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True, + 'protection': False}), (['buy', 'roi'], - {'buy': True, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False}), + {'buy': True, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False, + 'protection': False}), (['all'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True, + 'protection': True}), (['default'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False, + 'protection': False}), (['default', 'trailing'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True, + 'protection': False}), (['all', 'buy'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True, + 'protection': True}), (['default', 'buy'], - {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}), + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False, + 'protection': False}), + (['all'], + {'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True, + 'protection': True}), + (['protection'], + {'buy': False, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': False, + 'protection': True}), ]) def test_has_space(hyperopt_conf, spaces, expected_results): - for s in ['buy', 'sell', 'roi', 'stoploss', 'trailing']: + for s in ['buy', 'sell', 'roi', 'stoploss', 'trailing', 'protection']: hyperopt_conf.update({'spaces': spaces}) assert HyperoptTools.has_space(hyperopt_conf, s) == expected_results[s]