1226 lines
48 KiB
Python
1226 lines
48 KiB
Python
# pragma pylint: disable=missing-docstring,W0212,C0103
|
|
import locale
|
|
from datetime import datetime
|
|
from pathlib import Path
|
|
from typing import Dict, List
|
|
from unittest.mock import MagicMock, PropertyMock
|
|
|
|
import pandas as pd
|
|
import pytest
|
|
from arrow import Arrow
|
|
from filelock import Timeout
|
|
|
|
from freqtrade import constants
|
|
from freqtrade.commands.optimize_commands import (setup_optimize_configuration,
|
|
start_hyperopt)
|
|
from freqtrade.data.history import load_data
|
|
from freqtrade.exceptions import DependencyException, OperationalException
|
|
from freqtrade.optimize.default_hyperopt import DefaultHyperOpt
|
|
from freqtrade.optimize.default_hyperopt_loss import DefaultHyperOptLoss
|
|
from freqtrade.optimize.hyperopt import Hyperopt
|
|
from freqtrade.resolvers.hyperopt_resolver import (HyperOptLossResolver,
|
|
HyperOptResolver)
|
|
from freqtrade.state import RunMode
|
|
from freqtrade.strategy.interface import SellType
|
|
from tests.conftest import (get_args, log_has, log_has_re, patch_exchange,
|
|
patched_configuration_load_config_file)
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def hyperopt(default_conf, mocker):
|
|
default_conf.update({
|
|
'spaces': ['default'],
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
})
|
|
patch_exchange(mocker)
|
|
return Hyperopt(default_conf)
|
|
|
|
|
|
@pytest.fixture(scope='function')
|
|
def hyperopt_results():
|
|
return pd.DataFrame(
|
|
{
|
|
'pair': ['ETH/BTC', 'ETH/BTC', 'ETH/BTC'],
|
|
'profit_percent': [-0.1, 0.2, 0.3],
|
|
'profit_abs': [-0.2, 0.4, 0.6],
|
|
'trade_duration': [10, 30, 10],
|
|
'sell_reason': [SellType.STOP_LOSS, SellType.ROI, SellType.ROI],
|
|
'close_time':
|
|
[
|
|
datetime(2019, 1, 1, 9, 26, 3, 478039),
|
|
datetime(2019, 2, 1, 9, 26, 3, 478039),
|
|
datetime(2019, 3, 1, 9, 26, 3, 478039)
|
|
]
|
|
}
|
|
)
|
|
|
|
|
|
# Functions for recurrent object patching
|
|
def create_trials(mocker, hyperopt, testdatadir) -> List[Dict]:
|
|
"""
|
|
When creating trials, mock the hyperopt Trials so that *by default*
|
|
- we don't create any pickle'd files in the filesystem
|
|
- we might have a pickle'd file so make sure that we return
|
|
false when looking for it
|
|
"""
|
|
hyperopt.trials_file = testdatadir / 'optimize/ut_trials.pickle'
|
|
|
|
mocker.patch.object(Path, "is_file", MagicMock(return_value=False))
|
|
stat_mock = MagicMock()
|
|
stat_mock.st_size = PropertyMock(return_value=1)
|
|
mocker.patch.object(Path, "stat", MagicMock(return_value=False))
|
|
|
|
mocker.patch.object(Path, "unlink", MagicMock(return_value=True))
|
|
mocker.patch('freqtrade.optimize.hyperopt.dump', return_value=None)
|
|
|
|
return [{'loss': 1, 'result': 'foo', 'params': {}}]
|
|
|
|
|
|
def test_setup_hyperopt_configuration_without_arguments(mocker, default_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
]
|
|
|
|
config = setup_optimize_configuration(get_args(args), RunMode.HYPEROPT)
|
|
assert 'max_open_trades' in config
|
|
assert 'stake_currency' in config
|
|
assert 'stake_amount' in config
|
|
assert 'exchange' in config
|
|
assert 'pair_whitelist' in config['exchange']
|
|
assert 'datadir' in config
|
|
assert log_has('Using data directory: {} ...'.format(config['datadir']), caplog)
|
|
assert 'ticker_interval' in config
|
|
assert not log_has_re('Parameter -i/--ticker-interval detected .*', caplog)
|
|
|
|
assert 'position_stacking' not in config
|
|
assert not log_has('Parameter --enable-position-stacking detected ...', caplog)
|
|
|
|
assert 'timerange' not in config
|
|
assert 'runmode' in config
|
|
assert config['runmode'] == RunMode.HYPEROPT
|
|
|
|
|
|
def test_setup_hyperopt_configuration_with_arguments(mocker, default_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
mocker.patch(
|
|
'freqtrade.configuration.configuration.create_datadir',
|
|
lambda c, x: x
|
|
)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
'--datadir', '/foo/bar',
|
|
'--ticker-interval', '1m',
|
|
'--timerange', ':100',
|
|
'--enable-position-stacking',
|
|
'--disable-max-market-positions',
|
|
'--epochs', '1000',
|
|
'--spaces', 'default',
|
|
'--print-all'
|
|
]
|
|
|
|
config = setup_optimize_configuration(get_args(args), RunMode.HYPEROPT)
|
|
assert 'max_open_trades' in config
|
|
assert 'stake_currency' in config
|
|
assert 'stake_amount' in config
|
|
assert 'exchange' in config
|
|
assert 'pair_whitelist' in config['exchange']
|
|
assert 'datadir' in config
|
|
assert config['runmode'] == RunMode.HYPEROPT
|
|
|
|
assert log_has('Using data directory: {} ...'.format(config['datadir']), caplog)
|
|
assert 'ticker_interval' in config
|
|
assert log_has('Parameter -i/--ticker-interval detected ... Using ticker_interval: 1m ...',
|
|
caplog)
|
|
|
|
assert 'position_stacking' in config
|
|
assert log_has('Parameter --enable-position-stacking detected ...', caplog)
|
|
|
|
assert 'use_max_market_positions' in config
|
|
assert log_has('Parameter --disable-max-market-positions detected ...', caplog)
|
|
assert log_has('max_open_trades set to unlimited ...', caplog)
|
|
|
|
assert 'timerange' in config
|
|
assert log_has('Parameter --timerange detected: {} ...'.format(config['timerange']), caplog)
|
|
|
|
assert 'epochs' in config
|
|
assert log_has('Parameter --epochs detected ... Will run Hyperopt with for 1000 epochs ...',
|
|
caplog)
|
|
|
|
assert 'spaces' in config
|
|
assert log_has('Parameter -s/--spaces detected: {}'.format(config['spaces']), caplog)
|
|
assert 'print_all' in config
|
|
assert log_has('Parameter --print-all detected ...', caplog)
|
|
|
|
|
|
def test_setup_hyperopt_configuration_unlimited_stake_amount(mocker, default_conf, caplog) -> None:
|
|
default_conf['stake_amount'] = constants.UNLIMITED_STAKE_AMOUNT
|
|
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
]
|
|
|
|
with pytest.raises(DependencyException, match=r'.`stake_amount`.*'):
|
|
setup_optimize_configuration(get_args(args), RunMode.HYPEROPT)
|
|
|
|
|
|
def test_hyperoptresolver(mocker, default_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
|
|
hyperopt = DefaultHyperOpt
|
|
delattr(hyperopt, 'populate_indicators')
|
|
delattr(hyperopt, 'populate_buy_trend')
|
|
delattr(hyperopt, 'populate_sell_trend')
|
|
mocker.patch(
|
|
'freqtrade.resolvers.hyperopt_resolver.HyperOptResolver.load_object',
|
|
MagicMock(return_value=hyperopt(default_conf))
|
|
)
|
|
default_conf.update({'hyperopt': 'DefaultHyperOpt'})
|
|
x = HyperOptResolver.load_hyperopt(default_conf)
|
|
assert not hasattr(x, 'populate_indicators')
|
|
assert not hasattr(x, 'populate_buy_trend')
|
|
assert not hasattr(x, 'populate_sell_trend')
|
|
assert log_has("Hyperopt class does not provide populate_indicators() method. "
|
|
"Using populate_indicators from the strategy.", caplog)
|
|
assert log_has("Hyperopt class does not provide populate_sell_trend() method. "
|
|
"Using populate_sell_trend from the strategy.", caplog)
|
|
assert log_has("Hyperopt class does not provide populate_buy_trend() method. "
|
|
"Using populate_buy_trend from the strategy.", caplog)
|
|
assert hasattr(x, "ticker_interval")
|
|
|
|
|
|
def test_hyperoptresolver_wrongname(mocker, default_conf, caplog) -> None:
|
|
default_conf.update({'hyperopt': "NonExistingHyperoptClass"})
|
|
|
|
with pytest.raises(OperationalException, match=r'Impossible to load Hyperopt.*'):
|
|
HyperOptResolver.load_hyperopt(default_conf)
|
|
|
|
|
|
def test_hyperoptresolver_noname(default_conf):
|
|
default_conf['hyperopt'] = ''
|
|
with pytest.raises(OperationalException,
|
|
match="No Hyperopt set. Please use `--hyperopt` to specify "
|
|
"the Hyperopt class to use."):
|
|
HyperOptResolver.load_hyperopt(default_conf)
|
|
|
|
|
|
def test_hyperoptlossresolver(mocker, default_conf, caplog) -> None:
|
|
|
|
hl = DefaultHyperOptLoss
|
|
mocker.patch(
|
|
'freqtrade.resolvers.hyperopt_resolver.HyperOptLossResolver.load_object',
|
|
MagicMock(return_value=hl)
|
|
)
|
|
x = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
assert hasattr(x, "hyperopt_loss_function")
|
|
|
|
|
|
def test_hyperoptlossresolver_wrongname(mocker, default_conf, caplog) -> None:
|
|
default_conf.update({'hyperopt_loss': "NonExistingLossClass"})
|
|
|
|
with pytest.raises(OperationalException, match=r'Impossible to load HyperoptLoss.*'):
|
|
HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
|
|
|
|
def test_start_not_installed(mocker, default_conf, caplog, import_fails) -> None:
|
|
start_mock = MagicMock()
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
|
|
mocker.patch('freqtrade.optimize.hyperopt.Hyperopt.start', start_mock)
|
|
patch_exchange(mocker)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
'--epochs', '5'
|
|
]
|
|
pargs = get_args(args)
|
|
|
|
with pytest.raises(OperationalException, match=r"Please ensure that the hyperopt dependencies"):
|
|
start_hyperopt(pargs)
|
|
|
|
|
|
def test_start(mocker, default_conf, caplog) -> None:
|
|
start_mock = MagicMock()
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
mocker.patch('freqtrade.optimize.hyperopt.Hyperopt.start', start_mock)
|
|
patch_exchange(mocker)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
'--epochs', '5'
|
|
]
|
|
pargs = get_args(args)
|
|
start_hyperopt(pargs)
|
|
|
|
assert log_has('Starting freqtrade in Hyperopt mode', caplog)
|
|
assert start_mock.call_count == 1
|
|
|
|
|
|
def test_start_no_data(mocker, default_conf, caplog) -> None:
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
mocker.patch('freqtrade.data.history.load_pair_history', MagicMock(return_value=pd.DataFrame))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
patch_exchange(mocker)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
'--epochs', '5'
|
|
]
|
|
pargs = get_args(args)
|
|
with pytest.raises(OperationalException, match='No data found. Terminating.'):
|
|
start_hyperopt(pargs)
|
|
|
|
|
|
def test_start_filelock(mocker, default_conf, caplog) -> None:
|
|
start_mock = MagicMock(side_effect=Timeout(Hyperopt.get_lock_filename(default_conf)))
|
|
patched_configuration_load_config_file(mocker, default_conf)
|
|
mocker.patch('freqtrade.optimize.hyperopt.Hyperopt.start', start_mock)
|
|
patch_exchange(mocker)
|
|
|
|
args = [
|
|
'hyperopt',
|
|
'--config', 'config.json',
|
|
'--hyperopt', 'DefaultHyperOpt',
|
|
'--epochs', '5'
|
|
]
|
|
pargs = get_args(args)
|
|
start_hyperopt(pargs)
|
|
assert log_has("Another running instance of freqtrade Hyperopt detected.", caplog)
|
|
|
|
|
|
def test_loss_calculation_prefer_correct_trade_count(default_conf, hyperopt_results) -> None:
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, 600,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(hyperopt_results, 600 + 100,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(hyperopt_results, 600 - 100,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over > correct
|
|
assert under > correct
|
|
|
|
|
|
def test_loss_calculation_prefer_shorter_trades(default_conf, hyperopt_results) -> None:
|
|
resultsb = hyperopt_results.copy()
|
|
resultsb.loc[1, 'trade_duration'] = 20
|
|
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
longer = hl.hyperopt_loss_function(hyperopt_results, 100,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
shorter = hl.hyperopt_loss_function(resultsb, 100,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert shorter < longer
|
|
|
|
|
|
def test_loss_calculation_has_limited_profit(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, 600,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, 600,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, 600,
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_sharpe_loss_prefers_higher_profits(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
default_conf.update({'hyperopt_loss': 'SharpeHyperOptLoss'})
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_sharpe_loss_daily_prefers_higher_profits(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
default_conf.update({'hyperopt_loss': 'SharpeHyperOptLossDaily'})
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_sortino_loss_prefers_higher_profits(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
default_conf.update({'hyperopt_loss': 'SortinoHyperOptLoss'})
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_sortino_loss_daily_prefers_higher_profits(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
default_conf.update({'hyperopt_loss': 'SortinoHyperOptLossDaily'})
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_onlyprofit_loss_prefers_higher_profits(default_conf, hyperopt_results) -> None:
|
|
results_over = hyperopt_results.copy()
|
|
results_over['profit_percent'] = hyperopt_results['profit_percent'] * 2
|
|
results_under = hyperopt_results.copy()
|
|
results_under['profit_percent'] = hyperopt_results['profit_percent'] / 2
|
|
|
|
default_conf.update({'hyperopt_loss': 'OnlyProfitHyperOptLoss'})
|
|
hl = HyperOptLossResolver.load_hyperoptloss(default_conf)
|
|
correct = hl.hyperopt_loss_function(hyperopt_results, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
over = hl.hyperopt_loss_function(results_over, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
under = hl.hyperopt_loss_function(results_under, len(hyperopt_results),
|
|
datetime(2019, 1, 1), datetime(2019, 5, 1))
|
|
assert over < correct
|
|
assert under > correct
|
|
|
|
|
|
def test_log_results_if_loss_improves(hyperopt, capsys) -> None:
|
|
hyperopt.current_best_loss = 2
|
|
hyperopt.total_epochs = 2
|
|
|
|
hyperopt.print_results(
|
|
{
|
|
'loss': 1,
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
},
|
|
'total_profit': 0,
|
|
'current_epoch': 2, # This starts from 1 (in a human-friendly manner)
|
|
'is_initial_point': False,
|
|
'is_best': True
|
|
}
|
|
)
|
|
out, err = capsys.readouterr()
|
|
assert all(x in out
|
|
for x in ["Best", "2/2", " 1", "0.10%", "0.00100000 BTC (1.00%)", "20.0 m"])
|
|
|
|
|
|
def test_no_log_if_loss_does_not_improve(hyperopt, caplog) -> None:
|
|
hyperopt.current_best_loss = 2
|
|
hyperopt.print_results(
|
|
{
|
|
'is_best': False,
|
|
'loss': 3,
|
|
'current_epoch': 1,
|
|
}
|
|
)
|
|
assert caplog.record_tuples == []
|
|
|
|
|
|
def test_save_trials_saves_trials(mocker, hyperopt, testdatadir, caplog) -> None:
|
|
trials = create_trials(mocker, hyperopt, testdatadir)
|
|
mock_dump = mocker.patch('freqtrade.optimize.hyperopt.dump', return_value=None)
|
|
trials_file = testdatadir / 'optimize' / 'ut_trials.pickle'
|
|
|
|
hyperopt.trials = trials
|
|
hyperopt.save_trials(final=True)
|
|
assert log_has(f"1 epoch saved to '{trials_file}'.", caplog)
|
|
mock_dump.assert_called_once()
|
|
|
|
hyperopt.trials = trials + trials
|
|
hyperopt.save_trials(final=True)
|
|
assert log_has(f"2 epochs saved to '{trials_file}'.", caplog)
|
|
|
|
|
|
def test_read_trials_returns_trials_file(mocker, hyperopt, testdatadir, caplog) -> None:
|
|
trials = create_trials(mocker, hyperopt, testdatadir)
|
|
mock_load = mocker.patch('freqtrade.optimize.hyperopt.load', return_value=trials)
|
|
trials_file = testdatadir / 'optimize' / 'ut_trials.pickle'
|
|
hyperopt_trial = hyperopt._read_trials(trials_file)
|
|
assert log_has(f"Reading Trials from '{trials_file}'", caplog)
|
|
assert hyperopt_trial == trials
|
|
mock_load.assert_called_once()
|
|
|
|
|
|
def test_roi_table_generation(hyperopt) -> None:
|
|
params = {
|
|
'roi_t1': 5,
|
|
'roi_t2': 10,
|
|
'roi_t3': 15,
|
|
'roi_p1': 1,
|
|
'roi_p2': 2,
|
|
'roi_p3': 3,
|
|
}
|
|
|
|
assert hyperopt.custom_hyperopt.generate_roi_table(params) == {0: 6, 15: 3, 25: 1, 30: 0}
|
|
|
|
|
|
def test_start_calls_optimizer(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result',
|
|
'params': {'buy': {}, 'sell': {}, 'roi': {}, 'stoploss': 0.0},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
},
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
# Co-test loading timeframe from strategy
|
|
del default_conf['ticker_interval']
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'default',
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert 'Best result:\n\n* 1/1: foo result Objective: 1.00000\n' in out
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_sell")
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_buy")
|
|
assert hasattr(hyperopt, "max_open_trades")
|
|
assert hyperopt.max_open_trades == default_conf['max_open_trades']
|
|
assert hasattr(hyperopt, "position_stacking")
|
|
|
|
|
|
def test_format_results(hyperopt):
|
|
# Test with BTC as stake_currency
|
|
trades = [
|
|
('ETH/BTC', 2, 2, 123),
|
|
('LTC/BTC', 1, 1, 123),
|
|
('XPR/BTC', -1, -2, -246)
|
|
]
|
|
labels = ['currency', 'profit_percent', 'profit_abs', 'trade_duration']
|
|
df = pd.DataFrame.from_records(trades, columns=labels)
|
|
results_metrics = hyperopt._calculate_results_metrics(df)
|
|
results_explanation = hyperopt._format_results_explanation_string(results_metrics)
|
|
total_profit = results_metrics['total_profit']
|
|
|
|
results = {
|
|
'loss': 0.0,
|
|
'params_dict': None,
|
|
'params_details': None,
|
|
'results_metrics': results_metrics,
|
|
'results_explanation': results_explanation,
|
|
'total_profit': total_profit,
|
|
'current_epoch': 1,
|
|
'is_initial_point': True,
|
|
}
|
|
|
|
result = hyperopt._format_explanation_string(results, 1)
|
|
assert result.find(' 66.67%')
|
|
assert result.find('Total profit 1.00000000 BTC')
|
|
assert result.find('2.0000Σ %')
|
|
|
|
# Test with EUR as stake_currency
|
|
trades = [
|
|
('ETH/EUR', 2, 2, 123),
|
|
('LTC/EUR', 1, 1, 123),
|
|
('XPR/EUR', -1, -2, -246)
|
|
]
|
|
df = pd.DataFrame.from_records(trades, columns=labels)
|
|
results_metrics = hyperopt._calculate_results_metrics(df)
|
|
results['total_profit'] = results_metrics['total_profit']
|
|
result = hyperopt._format_explanation_string(results, 1)
|
|
assert result.find('Total profit 1.00000000 EUR')
|
|
|
|
|
|
@pytest.mark.parametrize("spaces, expected_results", [
|
|
(['buy'],
|
|
{'buy': True, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': False}),
|
|
(['sell'],
|
|
{'buy': False, 'sell': True, 'roi': False, 'stoploss': False, 'trailing': False}),
|
|
(['roi'],
|
|
{'buy': False, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False}),
|
|
(['stoploss'],
|
|
{'buy': False, 'sell': False, 'roi': False, 'stoploss': True, 'trailing': False}),
|
|
(['trailing'],
|
|
{'buy': False, 'sell': False, 'roi': False, 'stoploss': False, 'trailing': True}),
|
|
(['buy', 'sell', 'roi', 'stoploss'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}),
|
|
(['buy', 'sell', 'roi', 'stoploss', 'trailing'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}),
|
|
(['buy', 'roi'],
|
|
{'buy': True, 'sell': False, 'roi': True, 'stoploss': False, 'trailing': False}),
|
|
(['all'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}),
|
|
(['default'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}),
|
|
(['default', 'trailing'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}),
|
|
(['all', 'buy'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': True}),
|
|
(['default', 'buy'],
|
|
{'buy': True, 'sell': True, 'roi': True, 'stoploss': True, 'trailing': False}),
|
|
])
|
|
def test_has_space(hyperopt, spaces, expected_results):
|
|
for s in ['buy', 'sell', 'roi', 'stoploss', 'trailing']:
|
|
hyperopt.config.update({'spaces': spaces})
|
|
assert hyperopt.has_space(s) == expected_results[s]
|
|
|
|
|
|
def test_populate_indicators(hyperopt, testdatadir) -> None:
|
|
data = load_data(testdatadir, '1m', ['UNITTEST/BTC'], fill_up_missing=True)
|
|
dataframes = hyperopt.backtesting.strategy.ohlcvdata_to_dataframe(data)
|
|
dataframe = hyperopt.custom_hyperopt.populate_indicators(dataframes['UNITTEST/BTC'],
|
|
{'pair': 'UNITTEST/BTC'})
|
|
|
|
# Check if some indicators are generated. We will not test all of them
|
|
assert 'adx' in dataframe
|
|
assert 'mfi' in dataframe
|
|
assert 'rsi' in dataframe
|
|
|
|
|
|
def test_buy_strategy_generator(hyperopt, testdatadir) -> None:
|
|
data = load_data(testdatadir, '1m', ['UNITTEST/BTC'], fill_up_missing=True)
|
|
dataframes = hyperopt.backtesting.strategy.ohlcvdata_to_dataframe(data)
|
|
dataframe = hyperopt.custom_hyperopt.populate_indicators(dataframes['UNITTEST/BTC'],
|
|
{'pair': 'UNITTEST/BTC'})
|
|
|
|
populate_buy_trend = hyperopt.custom_hyperopt.buy_strategy_generator(
|
|
{
|
|
'adx-value': 20,
|
|
'fastd-value': 20,
|
|
'mfi-value': 20,
|
|
'rsi-value': 20,
|
|
'adx-enabled': True,
|
|
'fastd-enabled': True,
|
|
'mfi-enabled': True,
|
|
'rsi-enabled': True,
|
|
'trigger': 'bb_lower'
|
|
}
|
|
)
|
|
result = populate_buy_trend(dataframe, {'pair': 'UNITTEST/BTC'})
|
|
# Check if some indicators are generated. We will not test all of them
|
|
assert 'buy' in result
|
|
assert 1 in result['buy']
|
|
|
|
|
|
def test_generate_optimizer(mocker, default_conf) -> None:
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'timerange': None,
|
|
'spaces': 'all',
|
|
'hyperopt_min_trades': 1,
|
|
})
|
|
|
|
trades = [
|
|
('TRX/BTC', 0.023117, 0.000233, 100)
|
|
]
|
|
labels = ['currency', 'profit_percent', 'profit_abs', 'trade_duration']
|
|
backtest_result = pd.DataFrame.from_records(trades, columns=labels)
|
|
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Backtesting.backtest',
|
|
MagicMock(return_value=backtest_result)
|
|
)
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(Arrow(2017, 12, 10), Arrow(2017, 12, 13)))
|
|
)
|
|
patch_exchange(mocker)
|
|
mocker.patch('freqtrade.optimize.hyperopt.load', MagicMock())
|
|
|
|
optimizer_param = {
|
|
'adx-value': 0,
|
|
'fastd-value': 35,
|
|
'mfi-value': 0,
|
|
'rsi-value': 0,
|
|
'adx-enabled': False,
|
|
'fastd-enabled': True,
|
|
'mfi-enabled': False,
|
|
'rsi-enabled': False,
|
|
'trigger': 'macd_cross_signal',
|
|
'sell-adx-value': 0,
|
|
'sell-fastd-value': 75,
|
|
'sell-mfi-value': 0,
|
|
'sell-rsi-value': 0,
|
|
'sell-adx-enabled': False,
|
|
'sell-fastd-enabled': True,
|
|
'sell-mfi-enabled': False,
|
|
'sell-rsi-enabled': False,
|
|
'sell-trigger': 'macd_cross_signal',
|
|
'roi_t1': 60.0,
|
|
'roi_t2': 30.0,
|
|
'roi_t3': 20.0,
|
|
'roi_p1': 0.01,
|
|
'roi_p2': 0.01,
|
|
'roi_p3': 0.1,
|
|
'stoploss': -0.4,
|
|
'trailing_stop': True,
|
|
'trailing_stop_positive': 0.02,
|
|
'trailing_stop_positive_offset_p1': 0.05,
|
|
'trailing_only_offset_is_reached': False,
|
|
}
|
|
response_expected = {
|
|
'loss': 1.9840569076926293,
|
|
'results_explanation': (' 1 trades. 1 wins. 0 draws. 0 losses. '
|
|
'Avg profit 2.31%. Median profit 2.31%. Total profit '
|
|
'0.00023300 BTC ( 2.31\N{GREEK CAPITAL LETTER SIGMA}%). '
|
|
'Avg duration 100.0 min.'
|
|
).encode(locale.getpreferredencoding(), 'replace').decode('utf-8'),
|
|
'params_details': {'buy': {'adx-enabled': False,
|
|
'adx-value': 0,
|
|
'fastd-enabled': True,
|
|
'fastd-value': 35,
|
|
'mfi-enabled': False,
|
|
'mfi-value': 0,
|
|
'rsi-enabled': False,
|
|
'rsi-value': 0,
|
|
'trigger': 'macd_cross_signal'},
|
|
'roi': {0: 0.12000000000000001,
|
|
20.0: 0.02,
|
|
50.0: 0.01,
|
|
110.0: 0},
|
|
'sell': {'sell-adx-enabled': False,
|
|
'sell-adx-value': 0,
|
|
'sell-fastd-enabled': True,
|
|
'sell-fastd-value': 75,
|
|
'sell-mfi-enabled': False,
|
|
'sell-mfi-value': 0,
|
|
'sell-rsi-enabled': False,
|
|
'sell-rsi-value': 0,
|
|
'sell-trigger': 'macd_cross_signal'},
|
|
'stoploss': {'stoploss': -0.4},
|
|
'trailing': {'trailing_only_offset_is_reached': False,
|
|
'trailing_stop': True,
|
|
'trailing_stop_positive': 0.02,
|
|
'trailing_stop_positive_offset': 0.07}},
|
|
'params_dict': optimizer_param,
|
|
'results_metrics': {'avg_profit': 2.3117,
|
|
'draws': 0,
|
|
'duration': 100.0,
|
|
'losses': 0,
|
|
'median_profit': 2.3117,
|
|
'profit': 2.3117,
|
|
'total_profit': 0.000233,
|
|
'trade_count': 1,
|
|
'wins': 1},
|
|
'total_profit': 0.00023300
|
|
}
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.dimensions = hyperopt.hyperopt_space()
|
|
generate_optimizer_value = hyperopt.generate_optimizer(list(optimizer_param.values()))
|
|
assert generate_optimizer_value == response_expected
|
|
|
|
|
|
def test_clean_hyperopt(mocker, default_conf, caplog):
|
|
patch_exchange(mocker)
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'default',
|
|
'hyperopt_jobs': 1,
|
|
})
|
|
mocker.patch("freqtrade.optimize.hyperopt.Path.is_file", MagicMock(return_value=True))
|
|
unlinkmock = mocker.patch("freqtrade.optimize.hyperopt.Path.unlink", MagicMock())
|
|
h = Hyperopt(default_conf)
|
|
|
|
assert unlinkmock.call_count == 2
|
|
assert log_has(f"Removing `{h.data_pickle_file}`.", caplog)
|
|
|
|
|
|
def test_continue_hyperopt(mocker, default_conf, caplog):
|
|
patch_exchange(mocker)
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'default',
|
|
'hyperopt_jobs': 1,
|
|
'hyperopt_continue': True
|
|
})
|
|
mocker.patch("freqtrade.optimize.hyperopt.Path.is_file", MagicMock(return_value=True))
|
|
unlinkmock = mocker.patch("freqtrade.optimize.hyperopt.Path.unlink", MagicMock())
|
|
Hyperopt(default_conf)
|
|
|
|
assert unlinkmock.call_count == 0
|
|
assert log_has(f"Continuing on previous hyperopt results.", caplog)
|
|
|
|
|
|
def test_print_json_spaces_all(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {},
|
|
'params_details': {
|
|
'buy': {'mfi-value': None},
|
|
'sell': {'sell-mfi-value': None},
|
|
'roi': {}, 'stoploss': {'stoploss': None},
|
|
'trailing': {'trailing_stop': None}
|
|
},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'all',
|
|
'hyperopt_jobs': 1,
|
|
'print_json': True,
|
|
})
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
result_str = (
|
|
'{"params":{"mfi-value":null,"sell-mfi-value":null},"minimal_roi"'
|
|
':{},"stoploss":null,"trailing_stop":null}'
|
|
)
|
|
assert result_str in out # noqa: E501
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
|
|
|
|
def test_print_json_spaces_default(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {},
|
|
'params_details': {
|
|
'buy': {'mfi-value': None},
|
|
'sell': {'sell-mfi-value': None},
|
|
'roi': {}, 'stoploss': {'stoploss': None}
|
|
},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'default',
|
|
'hyperopt_jobs': 1,
|
|
'print_json': True,
|
|
})
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert '{"params":{"mfi-value":null,"sell-mfi-value":null},"minimal_roi":{},"stoploss":null}' in out # noqa: E501
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
|
|
|
|
def test_print_json_spaces_roi_stoploss(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {},
|
|
'params_details': {'roi': {}, 'stoploss': {'stoploss': None}},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'roi stoploss',
|
|
'hyperopt_jobs': 1,
|
|
'print_json': True,
|
|
})
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert '{"minimal_roi":{},"stoploss":null}' in out
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
|
|
|
|
def test_simplified_interface_roi_stoploss(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {'stoploss': 0.0},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'roi stoploss',
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
del hyperopt.custom_hyperopt.__class__.buy_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.sell_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.indicator_space
|
|
del hyperopt.custom_hyperopt.__class__.sell_indicator_space
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert 'Best result:\n\n* 1/1: foo result Objective: 1.00000\n' in out
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_sell")
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_buy")
|
|
assert hasattr(hyperopt, "max_open_trades")
|
|
assert hyperopt.max_open_trades == default_conf['max_open_trades']
|
|
assert hasattr(hyperopt, "position_stacking")
|
|
|
|
|
|
def test_simplified_interface_all_failed(mocker, default_conf, caplog, capsys) -> None:
|
|
mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'all',
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
del hyperopt.custom_hyperopt.__class__.buy_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.sell_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.indicator_space
|
|
del hyperopt.custom_hyperopt.__class__.sell_indicator_space
|
|
|
|
with pytest.raises(OperationalException, match=r"The 'buy' space is included into *"):
|
|
hyperopt.start()
|
|
|
|
|
|
def test_simplified_interface_buy(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'buy',
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
# TODO: sell_strategy_generator() is actually not called because
|
|
# run_optimizer_parallel() is mocked
|
|
del hyperopt.custom_hyperopt.__class__.sell_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.sell_indicator_space
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert 'Best result:\n\n* 1/1: foo result Objective: 1.00000\n' in out
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_sell")
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_buy")
|
|
assert hasattr(hyperopt, "max_open_trades")
|
|
assert hyperopt.max_open_trades == default_conf['max_open_trades']
|
|
assert hasattr(hyperopt, "position_stacking")
|
|
|
|
|
|
def test_simplified_interface_sell(mocker, default_conf, caplog, capsys) -> None:
|
|
dumper = mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
parallel = mocker.patch(
|
|
'freqtrade.optimize.hyperopt.Hyperopt.run_optimizer_parallel',
|
|
MagicMock(return_value=[{
|
|
'loss': 1, 'results_explanation': 'foo result', 'params': {},
|
|
'results_metrics':
|
|
{
|
|
'trade_count': 1,
|
|
'avg_profit': 0.1,
|
|
'total_profit': 0.001,
|
|
'profit': 1.0,
|
|
'duration': 20.0
|
|
}
|
|
}])
|
|
)
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': 'sell',
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
# TODO: buy_strategy_generator() is actually not called because
|
|
# run_optimizer_parallel() is mocked
|
|
del hyperopt.custom_hyperopt.__class__.buy_strategy_generator
|
|
del hyperopt.custom_hyperopt.__class__.indicator_space
|
|
|
|
hyperopt.start()
|
|
|
|
parallel.assert_called_once()
|
|
|
|
out, err = capsys.readouterr()
|
|
assert 'Best result:\n\n* 1/1: foo result Objective: 1.00000\n' in out
|
|
assert dumper.called
|
|
# Should be called twice, once for historical candle data, once to save evaluations
|
|
assert dumper.call_count == 2
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_sell")
|
|
assert hasattr(hyperopt.backtesting.strategy, "advise_buy")
|
|
assert hasattr(hyperopt, "max_open_trades")
|
|
assert hyperopt.max_open_trades == default_conf['max_open_trades']
|
|
assert hasattr(hyperopt, "position_stacking")
|
|
|
|
|
|
@pytest.mark.parametrize("method,space", [
|
|
('buy_strategy_generator', 'buy'),
|
|
('indicator_space', 'buy'),
|
|
('sell_strategy_generator', 'sell'),
|
|
('sell_indicator_space', 'sell'),
|
|
])
|
|
def test_simplified_interface_failed(mocker, default_conf, caplog, capsys, method, space) -> None:
|
|
mocker.patch('freqtrade.optimize.hyperopt.dump', MagicMock())
|
|
mocker.patch('freqtrade.optimize.backtesting.Backtesting.load_bt_data',
|
|
MagicMock(return_value=(MagicMock(), None)))
|
|
mocker.patch(
|
|
'freqtrade.optimize.hyperopt.get_timerange',
|
|
MagicMock(return_value=(datetime(2017, 12, 10), datetime(2017, 12, 13)))
|
|
)
|
|
|
|
patch_exchange(mocker)
|
|
|
|
default_conf.update({'config': 'config.json.example',
|
|
'hyperopt': 'DefaultHyperOpt',
|
|
'epochs': 1,
|
|
'timerange': None,
|
|
'spaces': space,
|
|
'hyperopt_jobs': 1, })
|
|
|
|
hyperopt = Hyperopt(default_conf)
|
|
hyperopt.backtesting.strategy.ohlcvdata_to_dataframe = MagicMock()
|
|
hyperopt.custom_hyperopt.generate_roi_table = MagicMock(return_value={})
|
|
|
|
delattr(hyperopt.custom_hyperopt.__class__, method)
|
|
|
|
with pytest.raises(OperationalException, match=f"The '{space}' space is included into *"):
|
|
hyperopt.start()
|