Fix standalone tests in hyperopt and optimize tests

This commit is contained in:
Matthias Voppichler 2018-03-11 15:21:29 +01:00
parent 8c74104801
commit 423f9f7351
2 changed files with 9 additions and 2 deletions

View File

@ -7,7 +7,7 @@ import pandas as pd
from freqtrade.optimize.hyperopt import calculate_loss, TARGET_TRADES, EXPECTED_MAX_PROFIT, start, \
log_results, save_trials, read_trials, generate_roi_table, has_space
from freqtrade.strategy.strategy import Strategy
import freqtrade.optimize.hyperopt as hyperopt
@ -72,6 +72,7 @@ def test_start_calls_fmin(mocker):
args = mocker.Mock(epochs=1, config='config.json.example', mongodb=False,
timerange=None, spaces='all')
Strategy().init({'strategy': 'default_strategy'})
start(args)
mock_fmin.assert_called_once()
@ -86,6 +87,7 @@ def test_start_uses_mongotrials(mocker):
args = mocker.Mock(epochs=1, config='config.json.example', mongodb=True,
timerange=None, spaces='all')
Strategy().init({'strategy': 'default_strategy'})
start(args)
mock_mongotrials.assert_called_once()
@ -149,6 +151,7 @@ def test_fmin_best_results(mocker, caplog):
args = mocker.Mock(epochs=1, config='config.json.example',
timerange=None, spaces='all')
Strategy().init({'strategy': 'default_strategy'})
start(args)
exists = [
@ -166,6 +169,7 @@ def test_fmin_best_results(mocker, caplog):
def test_fmin_throw_value_error(mocker, caplog):
caplog.set_level(logging.INFO)
Strategy().init({'strategy': 'default_strategy'})
mocker.patch('freqtrade.optimize.hyperopt.MongoTrials', return_value=create_trials(mocker))
mocker.patch('freqtrade.optimize.tickerdata_to_dataframe')
mocker.patch('freqtrade.optimize.load_data')
@ -173,6 +177,7 @@ def test_fmin_throw_value_error(mocker, caplog):
args = mocker.Mock(epochs=1, config='config.json.example',
timerange=None, spaces='all')
Strategy().init({'strategy': 'default_strategy'})
start(args)
exists = [
@ -209,7 +214,7 @@ def test_resuming_previous_hyperopt_results_succeeds(mocker):
mongodb=False,
timerange=None,
spaces='all')
Strategy().init({'strategy': 'default_strategy'})
start(args)
mock_read.assert_called_once()

View File

@ -10,6 +10,7 @@ from freqtrade.exchange import Bittrex
from freqtrade.optimize.__init__ import make_testdata_path, download_pairs,\
download_backtesting_testdata, load_tickerdata_file, trim_tickerlist, file_dump_json
from freqtrade.tests.conftest import log_has
from freqtrade.strategy.strategy import Strategy
# Change this if modifying BTC_UNITEST testdatafile
_BTC_UNITTEST_LENGTH = 13681
@ -218,6 +219,7 @@ def test_init(default_conf, mocker):
def test_tickerdata_to_dataframe():
Strategy().init({'strategy': 'default_strategy'})
timerange = ((None, 'line'), None, -100)
tick = load_tickerdata_file(None, 'BTC_UNITEST', 1, timerange=timerange)
tickerlist = {'BTC_UNITEST': tick}