Add startup test case

This commit is contained in:
Matthias 2023-05-20 11:28:52 +02:00
parent 2e675efa13
commit 209eb63ede
2 changed files with 50 additions and 13 deletions

View File

@ -6,7 +6,6 @@ from freqtrade.configuration import setup_utils_configuration
from freqtrade.enums import RunMode
from freqtrade.exceptions import OperationalException
from freqtrade.misc import round_coin_value
from freqtrade.optimize.lookahead_analysis import LookaheadAnalysisSubFunctions
from freqtrade.resolvers import StrategyResolver
@ -142,12 +141,16 @@ def start_lookahead_analysis(args: Dict[str, Any]) -> None:
:param args: Cli args from Arguments()
:return: None
"""
from freqtrade.optimize.lookahead_analysis import LookaheadAnalysisSubFunctions
config = setup_utils_configuration(args, RunMode.UTIL_NO_EXCHANGE)
if config['targeted_trade_amount'] < config['minimum_trade_amount']:
# add logic that tells the user to check the configuration
# since this combo doesn't make any sense.
pass
raise OperationalException(
"targeted trade amount can't be smaller than minimum trade amount."
)
strategy_objs = StrategyResolver.search_all_objects(
config, enum_failed=False, recursive=config.get('recursive_strategy_search', False))

View File

@ -1,17 +1,16 @@
# pragma pylint: disable=missing-docstring, W0212, line-too-long, C0103, unused-argument
from unittest.mock import PropertyMock
from pathlib import Path
from unittest.mock import MagicMock, PropertyMock
import numpy as np
import pytest
import freqtrade.commands.arguments
import freqtrade.optimize.lookahead_analysis
from freqtrade.configuration import TimeRange
from freqtrade.data import history
from freqtrade.data.converter import clean_ohlcv_dataframe
from freqtrade.commands.optimize_commands import start_lookahead_analysis
from freqtrade.data.history import get_timerange
from tests.conftest import generate_test_data, patch_exchange
from freqtrade.exceptions import OperationalException
from tests.conftest import CURRENT_TEST_STRATEGY, get_args, patch_exchange
@pytest.fixture
@ -22,11 +21,46 @@ def lookahead_conf(default_conf_usdt):
return default_conf_usdt
def trim_dictlist(dict_list, num):
new = {}
for pair, pair_data in dict_list.items():
new[pair] = pair_data[num:].reset_index()
return new
def test_start_start_lookahead_analysis(mocker):
single_mock = MagicMock()
mocker.patch.multiple(
'freqtrade.optimize.lookahead_analysis.LookaheadAnalysisSubFunctions',
initialize_single_lookahead_analysis=single_mock,
text_table_lookahead_analysis_instances=MagicMock(),
)
args = [
"lookahead-analysis",
"--strategy",
CURRENT_TEST_STRATEGY,
"--strategy-path",
str(Path(__file__).parent.parent / "strategy" / "strats"),
]
pargs = get_args(args)
pargs['config'] = None
start_lookahead_analysis(pargs)
assert single_mock.call_count == 1
single_mock.reset_mock()
# Test invalid config
args = [
"lookahead-analysis",
"--strategy",
CURRENT_TEST_STRATEGY,
"--strategy-path",
str(Path(__file__).parent.parent / "strategy" / "strats"),
"--targeted-trade-amount",
"10",
"--minimum-trade-amount",
"20",
]
pargs = get_args(args)
pargs['config'] = None
with pytest.raises(OperationalException,
match=r"targeted trade amount can't be smaller than .*"):
start_lookahead_analysis(pargs)
def test_biased_strategy(lookahead_conf, mocker, caplog) -> None: