Increase Hyperopt() code coverage
This commit is contained in:
		| @@ -3,6 +3,7 @@ import os | |||||||
| from copy import deepcopy | from copy import deepcopy | ||||||
| from unittest.mock import MagicMock | from unittest.mock import MagicMock | ||||||
| import pandas as pd | import pandas as pd | ||||||
|  | from freqtrade.optimize.__init__ import load_tickerdata_file | ||||||
| from freqtrade.optimize.hyperopt import Hyperopt | from freqtrade.optimize.hyperopt import Hyperopt | ||||||
| from freqtrade.tests.conftest import default_conf, log_has | from freqtrade.tests.conftest import default_conf, log_has | ||||||
|  |  | ||||||
| @@ -371,3 +372,43 @@ def test_has_space(): | |||||||
|  |  | ||||||
|     _HYPEROPT.config.update({'spaces': ['all']}) |     _HYPEROPT.config.update({'spaces': ['all']}) | ||||||
|     assert _HYPEROPT.has_space('buy') |     assert _HYPEROPT.has_space('buy') | ||||||
|  |  | ||||||
|  |  | ||||||
|  | def test_populate_indicators() -> None: | ||||||
|  |     """ | ||||||
|  |     Test Hyperopt.populate_indicators() | ||||||
|  |     """ | ||||||
|  |     tick = load_tickerdata_file(None, 'BTC_UNITEST', 1) | ||||||
|  |     tickerlist = {'BTC_UNITEST': tick} | ||||||
|  |     dataframes = _HYPEROPT.tickerdata_to_dataframe(tickerlist) | ||||||
|  |     dataframe = _HYPEROPT.populate_indicators(dataframes['BTC_UNITEST']) | ||||||
|  |  | ||||||
|  |     # Check if some indicators are generated. We will not test all of them | ||||||
|  |     assert 'adx' in dataframe | ||||||
|  |     assert 'ao' in dataframe | ||||||
|  |     assert 'cci' in dataframe | ||||||
|  |  | ||||||
|  |  | ||||||
|  | def test_buy_strategy_generator() -> None: | ||||||
|  |     """ | ||||||
|  |     Test Hyperopt.buy_strategy_generator() | ||||||
|  |     """ | ||||||
|  |     tick = load_tickerdata_file(None, 'BTC_UNITEST', 1) | ||||||
|  |     tickerlist = {'BTC_UNITEST': tick} | ||||||
|  |     dataframes = _HYPEROPT.tickerdata_to_dataframe(tickerlist) | ||||||
|  |     dataframe = _HYPEROPT.populate_indicators(dataframes['BTC_UNITEST']) | ||||||
|  |  | ||||||
|  |     populate_buy_trend = _HYPEROPT.buy_strategy_generator( | ||||||
|  |         { | ||||||
|  |             'adx': { | ||||||
|  |                 'enabled': True, | ||||||
|  |                 'value': 20 | ||||||
|  |             }, | ||||||
|  |             'trigger': { | ||||||
|  |                 'type': 'lower_bb' | ||||||
|  |             } | ||||||
|  |         } | ||||||
|  |     ) | ||||||
|  |     result = populate_buy_trend(dataframe) | ||||||
|  |     # Check if some indicators are generated. We will not test all of them | ||||||
|  |     assert 'adx' in result | ||||||
|   | |||||||
		Reference in New Issue
	
	Block a user