mirror of
https://github.com/freqtrade/freqtrade.git
synced 2026-01-20 14:00:38 +00:00
Merge pull request #10856 from freqtrade/fix/bt_parallel
Fix backtest parallelism with timeframe-detail
This commit is contained in:
@@ -732,7 +732,6 @@ class Backtesting:
|
||||
trade.close_date = current_time
|
||||
trade.close(order.ft_price, show_msg=False)
|
||||
|
||||
# logger.debug(f"{pair} - Backtesting exit {trade}")
|
||||
LocalTrade.close_bt_trade(trade)
|
||||
self.wallets.update()
|
||||
self.run_protections(pair, current_time, trade.trade_direction)
|
||||
@@ -1103,6 +1102,7 @@ class Backtesting:
|
||||
fee_close=self.fee,
|
||||
is_open=True,
|
||||
enter_tag=entry_tag,
|
||||
timeframe=self.timeframe_min,
|
||||
exchange=self._exchange_name,
|
||||
is_short=is_short,
|
||||
trading_mode=self.trading_mode,
|
||||
@@ -1358,7 +1358,7 @@ class Backtesting:
|
||||
and (self._position_stacking or len(LocalTrade.bt_trades_open_pp[pair]) == 0)
|
||||
and not PairLocks.is_pair_locked(pair, row[DATE_IDX], trade_dir)
|
||||
):
|
||||
if self.trade_slot_available(LocalTrade.bt_open_open_trade_count):
|
||||
if self.trade_slot_available(LocalTrade.bt_open_open_trade_count_candle):
|
||||
trade = self._enter_trade(pair, row, trade_dir)
|
||||
if trade:
|
||||
self.wallets.update()
|
||||
@@ -1432,6 +1432,10 @@ class Backtesting:
|
||||
):
|
||||
if is_first_call:
|
||||
self.check_abort()
|
||||
# Reset open trade count for this candle
|
||||
# Critical to avoid exceeding max_open_trades in backtesting
|
||||
# when timeframe-detail is used and trades close within the opening candle.
|
||||
LocalTrade.bt_open_open_trade_count_candle = LocalTrade.bt_open_open_trade_count
|
||||
strategy_safe_wrapper(self.strategy.bot_loop_start, supress_error=True)(
|
||||
current_time=current_time
|
||||
)
|
||||
|
||||
@@ -379,6 +379,7 @@ class LocalTrade:
|
||||
# Copy of trades_open - but indexed by pair
|
||||
bt_trades_open_pp: dict[str, list["LocalTrade"]] = defaultdict(list)
|
||||
bt_open_open_trade_count: int = 0
|
||||
bt_open_open_trade_count_candle: int = 0
|
||||
bt_total_profit: float = 0
|
||||
realized_profit: float = 0
|
||||
|
||||
@@ -747,6 +748,7 @@ class LocalTrade:
|
||||
LocalTrade.bt_trades_open = []
|
||||
LocalTrade.bt_trades_open_pp = defaultdict(list)
|
||||
LocalTrade.bt_open_open_trade_count = 0
|
||||
LocalTrade.bt_open_open_trade_count_candle = 0
|
||||
LocalTrade.bt_total_profit = 0
|
||||
|
||||
def adjust_min_max_rates(self, current_price: float, current_price_low: float) -> None:
|
||||
@@ -1442,6 +1444,11 @@ class LocalTrade:
|
||||
LocalTrade.bt_trades_open.remove(trade)
|
||||
LocalTrade.bt_trades_open_pp[trade.pair].remove(trade)
|
||||
LocalTrade.bt_open_open_trade_count -= 1
|
||||
if (trade.close_date_utc - trade.open_date_utc) > timedelta(minutes=trade.timeframe):
|
||||
# Only subtract trades that are open for more than 1 candle
|
||||
# To avoid exceeding max_open_trades.
|
||||
# Must be reset at the start of every candle during backesting.
|
||||
LocalTrade.bt_open_open_trade_count_candle -= 1
|
||||
LocalTrade.bt_trades.append(trade)
|
||||
LocalTrade.bt_total_profit += trade.close_profit_abs
|
||||
|
||||
@@ -1451,6 +1458,7 @@ class LocalTrade:
|
||||
LocalTrade.bt_trades_open.append(trade)
|
||||
LocalTrade.bt_trades_open_pp[trade.pair].append(trade)
|
||||
LocalTrade.bt_open_open_trade_count += 1
|
||||
LocalTrade.bt_open_open_trade_count_candle += 1
|
||||
else:
|
||||
LocalTrade.bt_trades.append(trade)
|
||||
|
||||
@@ -1459,6 +1467,9 @@ class LocalTrade:
|
||||
LocalTrade.bt_trades_open.remove(trade)
|
||||
LocalTrade.bt_trades_open_pp[trade.pair].remove(trade)
|
||||
LocalTrade.bt_open_open_trade_count -= 1
|
||||
# TODO: The below may have odd behavior in case of canceled entries
|
||||
# It might need to be removed so the trade "counts" as open for this candle.
|
||||
LocalTrade.bt_open_open_trade_count_candle -= 1
|
||||
|
||||
@staticmethod
|
||||
def get_open_trades() -> list[Any]:
|
||||
|
||||
@@ -1681,6 +1681,131 @@ def test_backtest_multi_pair_detail(
|
||||
assert len(evaluate_result_multi(results["results"], "5m", 1)) == 0
|
||||
|
||||
|
||||
@pytest.mark.parametrize("use_detail", [True, False])
|
||||
@pytest.mark.parametrize("pair", ["ADA/USDT", "LTC/USDT"])
|
||||
@pytest.mark.parametrize("tres", [0, 20, 30])
|
||||
def test_backtest_multi_pair_detail_simplified(
|
||||
default_conf_usdt,
|
||||
fee,
|
||||
mocker,
|
||||
tres,
|
||||
pair,
|
||||
use_detail,
|
||||
):
|
||||
"""
|
||||
literally the same as test_backtest_multi_pair_detail
|
||||
but with an "always enter" strategy, exiting after about half of the candle duration.
|
||||
"""
|
||||
|
||||
def _always_buy(dataframe, metadata):
|
||||
"""
|
||||
Buy every xth candle - sell every other xth -2 (hold on to pairs a bit)
|
||||
"""
|
||||
dataframe["enter_long"] = 1
|
||||
dataframe["enter_short"] = 0
|
||||
dataframe["exit_short"] = 0
|
||||
return dataframe
|
||||
|
||||
def custom_exit(
|
||||
trade: Trade,
|
||||
current_time: datetime,
|
||||
**kwargs,
|
||||
) -> str | bool | None:
|
||||
# Exit within the same candle.
|
||||
if (trade.open_date_utc + timedelta(minutes=20)) < current_time:
|
||||
return "exit after 20 minutes"
|
||||
|
||||
default_conf_usdt.update(
|
||||
{
|
||||
"runmode": "backtest",
|
||||
"stoploss": -1.0,
|
||||
"minimal_roi": {"0": 100},
|
||||
}
|
||||
)
|
||||
|
||||
if use_detail:
|
||||
default_conf_usdt["timeframe_detail"] = "5m"
|
||||
|
||||
mocker.patch(f"{EXMS}.get_min_pair_stake_amount", return_value=0.00001)
|
||||
mocker.patch(f"{EXMS}.get_max_pair_stake_amount", return_value=float("inf"))
|
||||
mocker.patch(f"{EXMS}.get_fee", fee)
|
||||
patch_exchange(mocker)
|
||||
|
||||
raw_candles_5m = generate_test_data("5m", 1000, "2022-01-03 12:00:00+00:00")
|
||||
raw_candles = ohlcv_fill_up_missing_data(raw_candles_5m, "1h", "dummy")
|
||||
|
||||
pairs = ["ADA/USDT", "DASH/USDT", "ETH/USDT", "LTC/USDT", "NXT/USDT"]
|
||||
data = {pair: raw_candles for pair in pairs}
|
||||
detail_data = {pair: raw_candles_5m for pair in pairs}
|
||||
|
||||
# Only use 500 lines to increase performance
|
||||
data = trim_dictlist(data, -200)
|
||||
|
||||
# Remove data for one pair from the beginning of the data
|
||||
if tres > 0:
|
||||
data[pair] = data[pair][tres:].reset_index()
|
||||
default_conf_usdt["timeframe"] = "1h"
|
||||
default_conf_usdt["max_open_trades"] = 3
|
||||
|
||||
backtesting = Backtesting(default_conf_usdt)
|
||||
vr_spy = mocker.spy(backtesting, "validate_row")
|
||||
bl_spy = mocker.spy(backtesting, "backtest_loop")
|
||||
backtesting.detail_data = detail_data
|
||||
backtesting._set_strategy(backtesting.strategylist[0])
|
||||
backtesting.strategy.bot_loop_start = MagicMock()
|
||||
backtesting.strategy.advise_entry = _always_buy # Override
|
||||
backtesting.strategy.advise_exit = _always_buy # Override
|
||||
backtesting.strategy.custom_exit = custom_exit # Override
|
||||
|
||||
processed = backtesting.strategy.advise_all_indicators(data)
|
||||
min_date, max_date = get_timerange(processed)
|
||||
|
||||
backtest_conf = {
|
||||
"processed": deepcopy(processed),
|
||||
"start_date": min_date,
|
||||
"end_date": max_date,
|
||||
}
|
||||
|
||||
results = backtesting.backtest(**backtest_conf)
|
||||
|
||||
# bot_loop_start is called once per candle.
|
||||
# assert backtesting.strategy.bot_loop_start.call_count == 83
|
||||
# Validated row once per candle and pair
|
||||
assert vr_spy.call_count == 415
|
||||
|
||||
if use_detail:
|
||||
# Backtest loop is called once per candle per pair
|
||||
# Exact numbers depend on trade state - but should be around 3_800
|
||||
assert bl_spy.call_count > 3_350
|
||||
assert bl_spy.call_count < 3_800
|
||||
else:
|
||||
assert bl_spy.call_count < 995
|
||||
|
||||
# Make sure we have parallel trades
|
||||
assert len(evaluate_result_multi(results["results"], "1h", 2)) > 0
|
||||
# make sure we don't have trades with more than configured max_open_trades
|
||||
assert len(evaluate_result_multi(results["results"], "1h", 3)) == 0
|
||||
|
||||
# # Cached data correctly removed amounts
|
||||
offset = 1 if tres == 0 else 0
|
||||
removed_candles = len(data[pair]) - offset
|
||||
assert len(backtesting.dataprovider.get_analyzed_dataframe(pair, "1h")[0]) == removed_candles
|
||||
assert (
|
||||
len(backtesting.dataprovider.get_analyzed_dataframe("NXT/USDT", "1h")[0])
|
||||
== len(data["NXT/USDT"]) - 1
|
||||
)
|
||||
|
||||
backtesting.strategy.max_open_trades = 1
|
||||
backtesting.config.update({"max_open_trades": 1})
|
||||
backtest_conf = {
|
||||
"processed": deepcopy(processed),
|
||||
"start_date": min_date,
|
||||
"end_date": max_date,
|
||||
}
|
||||
results = backtesting.backtest(**backtest_conf)
|
||||
assert len(evaluate_result_multi(results["results"], "1h", 1)) == 0
|
||||
|
||||
|
||||
@pytest.mark.parametrize("use_detail", [True, False])
|
||||
def test_backtest_multi_pair_long_short_switch(
|
||||
default_conf_usdt,
|
||||
|
||||
@@ -2144,6 +2144,7 @@ def test_Trade_object_idem():
|
||||
"bt_trades_open",
|
||||
"bt_trades_open_pp",
|
||||
"bt_open_open_trade_count",
|
||||
"bt_open_open_trade_count_candle",
|
||||
"bt_total_profit",
|
||||
"from_json",
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user