freqtrade_origin/tests/data/test_history.py

703 lines
25 KiB
Python
Raw Normal View History

# pragma pylint: disable=missing-docstring, protected-access, C0103
import json
2023-06-17 12:55:23 +00:00
import logging
import uuid
2023-08-20 09:51:01 +00:00
from datetime import timedelta
from pathlib import Path
from shutil import copyfile
2019-08-25 13:02:40 +00:00
from unittest.mock import MagicMock, PropertyMock
2019-12-27 12:16:53 +00:00
import pytest
from pandas import DataFrame
2019-12-27 09:11:49 +00:00
from pandas.testing import assert_frame_equal
from freqtrade.configuration import TimeRange
2022-09-23 05:09:34 +00:00
from freqtrade.constants import DATETIME_PRINT_FORMAT
from freqtrade.data.converter import ohlcv_to_dataframe
from freqtrade.data.history import get_datahandler
2024-03-15 05:49:49 +00:00
from freqtrade.data.history.datahandlers.jsondatahandler import JsonDataHandler, JsonGzDataHandler
2024-05-12 13:08:40 +00:00
from freqtrade.data.history.history_utils import (
_download_pair_history,
_download_trades_history,
_load_cached_data_for_updating,
get_timerange,
load_data,
load_pair_history,
refresh_backtest_ohlcv_data,
refresh_backtest_trades_data,
refresh_data,
validate_backtest_data,
)
2024-03-02 12:17:45 +00:00
from freqtrade.enums import CandleType, TradingMode
from freqtrade.exchange import timeframe_to_minutes
from freqtrade.misc import file_dump_json
from freqtrade.resolvers import StrategyResolver
2023-08-20 09:51:01 +00:00
from freqtrade.util import dt_ts, dt_utc
2024-05-12 13:08:40 +00:00
from tests.conftest import (
CURRENT_TEST_STRATEGY,
EXMS,
get_patched_exchange,
log_has,
log_has_re,
patch_exchange,
)
2020-09-28 17:43:15 +00:00
2019-10-08 19:10:43 +00:00
def _clean_test_file(file: Path) -> None:
"""
Backup existing file to avoid deleting the user file
:param file: complete path to the file
:return: None
"""
2024-05-12 14:00:45 +00:00
file_swp = Path(str(file) + ".swp")
# 1. Delete file from the test
2019-10-08 19:10:43 +00:00
if file.is_file():
file.unlink()
# 2. Rollback to the initial file
2019-10-08 19:10:43 +00:00
if file_swp.is_file():
file_swp.rename(file)
2022-09-23 05:09:34 +00:00
def test_load_data_30min_timeframe(caplog, testdatadir) -> None:
2024-05-12 14:00:45 +00:00
ld = load_pair_history(pair="UNITTEST/BTC", timeframe="30m", datadir=testdatadir)
assert isinstance(ld, DataFrame)
2019-05-17 16:05:36 +00:00
assert not log_has(
2024-05-12 14:00:45 +00:00
'Download history data for pair: "UNITTEST/BTC", timeframe: 30m ' "and store in None.",
caplog,
2019-05-17 16:05:36 +00:00
)
2022-09-23 05:09:34 +00:00
def test_load_data_7min_timeframe(caplog, testdatadir) -> None:
2024-05-12 14:00:45 +00:00
ld = load_pair_history(pair="UNITTEST/BTC", timeframe="7m", datadir=testdatadir)
assert isinstance(ld, DataFrame)
assert ld.empty
assert log_has(
2024-05-12 14:00:45 +00:00
"No history for UNITTEST/BTC, spot, 7m found. "
"Use `freqtrade download-data` to download the data",
caplog,
)
def test_load_data_1min_timeframe(ohlcv_history, mocker, caplog, testdatadir) -> None:
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_ohlcv", return_value=ohlcv_history)
file = testdatadir / "UNITTEST_BTC-1m.feather"
load_data(datadir=testdatadir, timeframe="1m", pairs=["UNITTEST/BTC"])
2019-10-08 19:10:43 +00:00
assert file.is_file()
2019-05-17 16:05:36 +00:00
assert not log_has(
2024-05-12 14:00:45 +00:00
'Download history data for pair: "UNITTEST/BTC", interval: 1m ' "and store in None.", caplog
2019-05-17 16:05:36 +00:00
)
def test_load_data_mark(ohlcv_history, mocker, caplog, testdatadir) -> None:
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_ohlcv", return_value=ohlcv_history)
file = testdatadir / "futures/UNITTEST_USDT_USDT-1h-mark.feather"
load_data(datadir=testdatadir, timeframe="1h", pairs=["UNITTEST/BTC"], candle_type="mark")
assert file.is_file()
assert not log_has(
2024-05-12 14:00:45 +00:00
'Download history data for pair: "UNITTEST/USDT:USDT", interval: 1m ' "and store in None.",
caplog,
)
2022-09-23 05:09:34 +00:00
def test_load_data_startup_candles(mocker, testdatadir) -> None:
ltfmock = mocker.patch(
2024-05-12 14:00:45 +00:00
"freqtrade.data.history.datahandlers.featherdatahandler.FeatherDataHandler._ohlcv_load",
MagicMock(return_value=DataFrame()),
)
timerange = TimeRange("date", None, 1510639620, 0)
load_pair_history(
pair="UNITTEST/BTC",
timeframe="1m",
datadir=testdatadir,
timerange=timerange,
startup_candles=20,
)
2019-10-27 09:00:44 +00:00
assert ltfmock.call_count == 1
2024-05-12 14:00:45 +00:00
assert ltfmock.call_args_list[0][1]["timerange"] != timerange
2019-10-27 09:00:44 +00:00
# startts is 20 minutes earlier
2024-05-12 14:00:45 +00:00
assert ltfmock.call_args_list[0][1]["timerange"].startts == timerange.startts - 20 * 60
2019-10-27 09:00:44 +00:00
2024-05-12 14:00:45 +00:00
@pytest.mark.parametrize("candle_type", ["mark", ""])
def test_load_data_with_new_pair_1min(
ohlcv_history_list, mocker, caplog, default_conf, tmp_path, candle_type
) -> None:
"""
Test load_pair_history() with 1 min timeframe
"""
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_ohlcv", return_value=ohlcv_history_list)
exchange = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
file = tmp_path / "MEME_BTC-1m.feather"
# do not download a new pair if refresh_pairs isn't set
2024-05-12 14:00:45 +00:00
load_pair_history(datadir=tmp_path, timeframe="1m", pair="MEME/BTC", candle_type=candle_type)
2019-10-08 19:10:43 +00:00
assert not file.is_file()
assert log_has(
f"No history for MEME/BTC, {candle_type}, 1m found. "
2024-05-12 14:00:45 +00:00
"Use `freqtrade download-data` to download the data",
caplog,
)
# download a new pair if refresh_pairs is set
2024-05-12 14:00:45 +00:00
refresh_data(
datadir=tmp_path,
timeframe="1m",
pairs=["MEME/BTC"],
exchange=exchange,
candle_type=CandleType.SPOT,
)
load_pair_history(datadir=tmp_path, timeframe="1m", pair="MEME/BTC", candle_type=candle_type)
2019-10-08 19:10:43 +00:00
assert file.is_file()
assert log_has_re(r'Download history data for "MEME/BTC", 1m, ' r"spot and store in .*", caplog)
2019-09-07 18:56:03 +00:00
def test_testdata_path(testdatadir) -> None:
2024-05-12 14:00:45 +00:00
assert str(Path("tests") / "testdata") in str(testdatadir)
@pytest.mark.parametrize(
"pair,timeframe,expected_result,candle_type",
[
("ETH/BTC", "5m", "freqtrade/hello/world/ETH_BTC-5m.json", ""),
("ETH/USDT", "1M", "freqtrade/hello/world/ETH_USDT-1Mo.json", ""),
("Fabric Token/ETH", "5m", "freqtrade/hello/world/Fabric_Token_ETH-5m.json", ""),
("ETHH20", "5m", "freqtrade/hello/world/ETHH20-5m.json", ""),
(".XBTBON2H", "5m", "freqtrade/hello/world/_XBTBON2H-5m.json", ""),
("ETHUSD.d", "5m", "freqtrade/hello/world/ETHUSD_d-5m.json", ""),
("ACC_OLD/BTC", "5m", "freqtrade/hello/world/ACC_OLD_BTC-5m.json", ""),
("ETH/BTC", "5m", "freqtrade/hello/world/futures/ETH_BTC-5m-mark.json", "mark"),
("ACC_OLD/BTC", "5m", "freqtrade/hello/world/futures/ACC_OLD_BTC-5m-index.json", "index"),
],
)
def test_json_pair_data_filename(pair, timeframe, expected_result, candle_type):
fn = JsonDataHandler._pair_data_filename(
2024-05-12 14:00:45 +00:00
Path("freqtrade/hello/world"), pair, timeframe, CandleType.from_string(candle_type)
)
2019-08-14 16:56:46 +00:00
assert isinstance(fn, Path)
assert fn == Path(expected_result)
fn = JsonGzDataHandler._pair_data_filename(
2024-05-12 14:00:45 +00:00
Path("freqtrade/hello/world"),
pair,
2022-05-16 17:53:01 +00:00
timeframe,
2024-05-12 14:00:45 +00:00
candle_type=CandleType.from_string(candle_type),
)
2019-12-25 15:41:52 +00:00
assert isinstance(fn, Path)
2024-05-12 14:00:45 +00:00
assert fn == Path(expected_result + ".gz")
@pytest.mark.parametrize(
"pair,trading_mode,expected_result",
[
("ETH/BTC", "", "freqtrade/hello/world/ETH_BTC-trades.json"),
("ETH/USDT:USDT", "futures", "freqtrade/hello/world/futures/ETH_USDT_USDT-trades.json"),
("Fabric Token/ETH", "", "freqtrade/hello/world/Fabric_Token_ETH-trades.json"),
("ETHH20", "", "freqtrade/hello/world/ETHH20-trades.json"),
(".XBTBON2H", "", "freqtrade/hello/world/_XBTBON2H-trades.json"),
("ETHUSD.d", "", "freqtrade/hello/world/ETHUSD_d-trades.json"),
("ACC_OLD_BTC", "", "freqtrade/hello/world/ACC_OLD_BTC-trades.json"),
],
)
2024-03-02 12:17:45 +00:00
def test_json_pair_trades_filename(pair, trading_mode, expected_result):
2024-05-12 14:00:45 +00:00
fn = JsonDataHandler._pair_trades_filename(Path("freqtrade/hello/world"), pair, trading_mode)
2019-12-25 15:41:52 +00:00
assert isinstance(fn, Path)
assert fn == Path(expected_result)
2019-08-14 16:56:46 +00:00
2024-05-12 14:00:45 +00:00
fn = JsonGzDataHandler._pair_trades_filename(Path("freqtrade/hello/world"), pair, trading_mode)
2019-08-14 16:58:27 +00:00
assert isinstance(fn, Path)
2024-05-12 14:00:45 +00:00
assert fn == Path(expected_result + ".gz")
2019-08-14 16:58:27 +00:00
2019-12-27 09:11:49 +00:00
def test_load_cached_data_for_updating(mocker, testdatadir) -> None:
2024-05-12 14:00:45 +00:00
data_handler = get_datahandler(testdatadir, "json")
test_data = None
2024-05-12 14:00:45 +00:00
test_filename = testdatadir.joinpath("UNITTEST_BTC-1m.json")
2023-02-25 16:08:02 +00:00
with test_filename.open("rt") as file:
test_data = json.load(file)
2024-05-12 14:00:45 +00:00
test_data_df = ohlcv_to_dataframe(
test_data, "1m", "UNITTEST/BTC", fill_missing=False, drop_incomplete=False
)
# now = last cached item + 1 hour
now_ts = test_data[-1][0] / 1000 + 60 * 60
# timeframe starts earlier than the cached data
# should fully update data
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, test_data[0][0] / 1000 - 1, 0)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"UNITTEST/BTC", "1m", timerange, data_handler, CandleType.SPOT
)
2019-12-27 09:11:49 +00:00
assert data.empty
assert start_ts == test_data[0][0] - 1000
2022-04-30 13:28:01 +00:00
assert end_ts is None
# timeframe starts earlier than the cached data - prepending
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, test_data[0][0] / 1000 - 1, 0)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"UNITTEST/BTC", "1m", timerange, data_handler, CandleType.SPOT, True
)
2022-04-30 13:28:01 +00:00
assert_frame_equal(data, test_data_df.iloc[:-1])
assert start_ts == test_data[0][0] - 1000
assert end_ts == test_data[0][0]
# timeframe starts in the center of the cached data
2021-08-16 12:16:24 +00:00
# should return the cached data w/o the last item
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, test_data[0][0] / 1000 + 1, 0)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"UNITTEST/BTC", "1m", timerange, data_handler, CandleType.SPOT
)
2019-12-27 09:11:49 +00:00
assert_frame_equal(data, test_data_df.iloc[:-1])
assert test_data[-2][0] <= start_ts < test_data[-1][0]
2022-04-30 13:28:01 +00:00
assert end_ts is None
2021-08-16 12:16:24 +00:00
# timeframe starts after the cached data
# should return the cached data w/o the last item
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, test_data[-1][0] / 1000 + 100, 0)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"UNITTEST/BTC", "1m", timerange, data_handler, CandleType.SPOT
)
2019-12-27 09:11:49 +00:00
assert_frame_equal(data, test_data_df.iloc[:-1])
assert test_data[-2][0] <= start_ts < test_data[-1][0]
2022-04-30 13:28:01 +00:00
assert end_ts is None
# no datafile exist
# should return timestamp start time
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, now_ts - 10000, 0)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"NONEXIST/BTC", "1m", timerange, data_handler, CandleType.SPOT
)
2019-12-27 09:11:49 +00:00
assert data.empty
assert start_ts == (now_ts - 10000) * 1000
2022-04-30 13:28:01 +00:00
assert end_ts is None
# no datafile exist
# should return timestamp start and end time time
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", "date", now_ts - 1000000, now_ts - 100000)
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"NONEXIST/BTC", "1m", timerange, data_handler, CandleType.SPOT
)
2022-04-30 13:28:01 +00:00
assert data.empty
assert start_ts == (now_ts - 1000000) * 1000
assert end_ts == (now_ts - 100000) * 1000
# no datafile exist, no timeframe is set
# should return an empty array and None
2022-04-30 13:28:01 +00:00
data, start_ts, end_ts = _load_cached_data_for_updating(
2024-05-12 14:00:45 +00:00
"NONEXIST/BTC", "1m", None, data_handler, CandleType.SPOT
)
2019-12-27 09:11:49 +00:00
assert data.empty
assert start_ts is None
2022-04-30 13:28:01 +00:00
assert end_ts is None
2024-05-12 14:00:45 +00:00
@pytest.mark.parametrize(
"candle_type,subdir,file_tail",
[
("mark", "futures/", "-mark"),
("spot", "", ""),
],
)
def test_download_pair_history(
2024-05-12 14:00:45 +00:00
ohlcv_history_list, mocker, default_conf, tmp_path, candle_type, subdir, file_tail
) -> None:
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_ohlcv", return_value=ohlcv_history_list)
exchange = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
file1_1 = tmp_path / f"{subdir}MEME_BTC-1m{file_tail}.feather"
file1_5 = tmp_path / f"{subdir}MEME_BTC-5m{file_tail}.feather"
file2_1 = tmp_path / f"{subdir}CFI_BTC-1m{file_tail}.feather"
file2_5 = tmp_path / f"{subdir}CFI_BTC-5m{file_tail}.feather"
2018-12-16 09:30:13 +00:00
2019-10-08 19:10:43 +00:00
assert not file1_1.is_file()
assert not file2_1.is_file()
2018-12-16 09:30:13 +00:00
2024-05-12 14:00:45 +00:00
assert _download_pair_history(
datadir=tmp_path,
exchange=exchange,
pair="MEME/BTC",
timeframe="1m",
candle_type=candle_type,
)
assert _download_pair_history(
datadir=tmp_path, exchange=exchange, pair="CFI/BTC", timeframe="1m", candle_type=candle_type
)
2018-12-16 09:30:13 +00:00
assert not exchange._pairs_last_refresh_time
2019-10-08 19:10:43 +00:00
assert file1_1.is_file()
assert file2_1.is_file()
# clean files freshly downloaded
_clean_test_file(file1_1)
2018-12-16 09:30:13 +00:00
_clean_test_file(file2_1)
2019-10-08 19:10:43 +00:00
assert not file1_5.is_file()
assert not file2_5.is_file()
2024-05-12 14:00:45 +00:00
assert _download_pair_history(
datadir=tmp_path,
exchange=exchange,
pair="MEME/BTC",
timeframe="5m",
candle_type=candle_type,
)
assert _download_pair_history(
datadir=tmp_path, exchange=exchange, pair="CFI/BTC", timeframe="5m", candle_type=candle_type
)
assert not exchange._pairs_last_refresh_time
2019-10-08 19:10:43 +00:00
assert file1_5.is_file()
assert file2_5.is_file()
2019-09-07 18:56:03 +00:00
def test_download_pair_history2(mocker, default_conf, testdatadir) -> None:
tick = [
[1509836520000, 0.00162008, 0.00162008, 0.00162008, 0.00162008, 108.14853839],
2024-05-12 14:00:45 +00:00
[1509836580000, 0.00161, 0.00161, 0.00161, 0.00161, 82.390199],
]
2019-12-27 06:07:27 +00:00
json_dump_mock = mocker.patch(
2024-05-12 14:00:45 +00:00
"freqtrade.data.history.datahandlers.featherdatahandler.FeatherDataHandler.ohlcv_store",
return_value=None,
)
mocker.patch(f"{EXMS}.get_historic_ohlcv", return_value=tick)
exchange = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
_download_pair_history(
datadir=testdatadir,
exchange=exchange,
pair="UNITTEST/BTC",
timeframe="1m",
candle_type="spot",
)
_download_pair_history(
datadir=testdatadir,
exchange=exchange,
pair="UNITTEST/BTC",
timeframe="3m",
candle_type="spot",
)
_download_pair_history(
datadir=testdatadir,
exchange=exchange,
pair="UNITTEST/USDT",
timeframe="1h",
candle_type="mark",
)
assert json_dump_mock.call_count == 3
2023-11-05 15:18:28 +00:00
def test_download_backtesting_data_exception(mocker, caplog, default_conf, tmp_path) -> None:
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_ohlcv", side_effect=Exception("File Error"))
2018-12-16 09:30:13 +00:00
exchange = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
assert not _download_pair_history(
datadir=tmp_path, exchange=exchange, pair="MEME/BTC", timeframe="1m", candle_type="spot"
)
assert log_has('Failed to download history data for pair: "MEME/BTC", timeframe: 1m.', caplog)
2018-12-16 09:30:13 +00:00
2019-09-07 18:56:03 +00:00
def test_load_partial_missing(testdatadir, caplog) -> None:
# Make sure we start fresh - test missing data at start
2023-05-14 16:31:09 +00:00
start = dt_utc(2018, 1, 1)
end = dt_utc(2018, 1, 11)
2024-05-12 14:00:45 +00:00
data = load_data(
testdatadir,
"5m",
["UNITTEST/BTC"],
startup_candles=20,
timerange=TimeRange("date", "date", start.timestamp(), end.timestamp()),
)
2024-05-12 14:00:45 +00:00
assert log_has("Using indicator startup period: 20 ...", caplog)
# timedifference in 5 minutes
td = ((end - start).total_seconds() // 60 // 5) + 1
2024-05-12 14:00:45 +00:00
assert td != len(data["UNITTEST/BTC"])
start_real = data["UNITTEST/BTC"].iloc[0, 0]
assert log_has(
2024-05-12 15:51:21 +00:00
f"UNITTEST/BTC, spot, 5m, data starts at {start_real.strftime(DATETIME_PRINT_FORMAT)}",
2024-05-12 14:00:45 +00:00
caplog,
)
# Make sure we start fresh - test missing data at end
caplog.clear()
2023-05-14 16:31:09 +00:00
start = dt_utc(2018, 1, 10)
end = dt_utc(2018, 2, 20)
2024-05-12 14:00:45 +00:00
data = load_data(
datadir=testdatadir,
timeframe="5m",
pairs=["UNITTEST/BTC"],
timerange=TimeRange("date", "date", start.timestamp(), end.timestamp()),
)
# timedifference in 5 minutes
td = ((end - start).total_seconds() // 60 // 5) + 1
2024-05-12 14:00:45 +00:00
assert td != len(data["UNITTEST/BTC"])
2022-09-28 18:23:56 +00:00
# Shift endtime with +5
2024-05-12 14:00:45 +00:00
end_real = data["UNITTEST/BTC"].iloc[-1, 0].to_pydatetime()
assert log_has(
2024-05-12 15:51:21 +00:00
f"UNITTEST/BTC, spot, 5m, data ends at {end_real.strftime(DATETIME_PRINT_FORMAT)}",
2024-05-12 14:00:45 +00:00
caplog,
)
2022-09-23 05:09:34 +00:00
def test_init(default_conf) -> None:
2024-05-12 14:00:45 +00:00
assert {} == load_data(datadir=Path(), pairs=[], timeframe=default_conf["timeframe"])
def test_init_with_refresh(default_conf, mocker) -> None:
exchange = get_patched_exchange(mocker, default_conf)
refresh_data(
2023-08-02 15:57:49 +00:00
datadir=Path(),
pairs=[],
2024-05-12 14:00:45 +00:00
timeframe=default_conf["timeframe"],
exchange=exchange,
2024-05-12 14:00:45 +00:00
candle_type=CandleType.SPOT,
)
2024-05-12 14:00:45 +00:00
assert {} == load_data(datadir=Path(), pairs=[], timeframe=default_conf["timeframe"])
2019-10-08 19:10:43 +00:00
def test_file_dump_json_tofile(testdatadir) -> None:
2024-05-12 14:00:45 +00:00
file = testdatadir / f"test_{uuid.uuid4()}.json"
data = {"bar": "foo"}
# check the file we will create does not exist
2019-10-08 19:10:43 +00:00
assert not file.is_file()
# Create the Json file
file_dump_json(file, data)
# Check the file was create
2019-10-08 19:10:43 +00:00
assert file.is_file()
# Open the Json file created and test the data is in it
2019-10-08 19:10:43 +00:00
with file.open() as data_file:
json_from_file = json.load(data_file)
2024-05-12 14:00:45 +00:00
assert "bar" in json_from_file
assert json_from_file["bar"] == "foo"
# Remove the file
_clean_test_file(file)
2019-12-17 22:06:03 +00:00
def test_get_timerange(default_conf, mocker, testdatadir) -> None:
patch_exchange(mocker)
2024-05-12 14:00:45 +00:00
default_conf.update({"strategy": CURRENT_TEST_STRATEGY})
strategy = StrategyResolver.load_strategy(default_conf)
data = strategy.advise_all_indicators(
2024-05-12 14:00:45 +00:00
load_data(datadir=testdatadir, timeframe="1m", pairs=["UNITTEST/BTC"])
)
2019-12-17 22:06:03 +00:00
min_date, max_date = get_timerange(data)
2024-05-12 14:00:45 +00:00
assert min_date.isoformat() == "2017-11-04T23:02:00+00:00"
assert max_date.isoformat() == "2017-11-14T22:59:00+00:00"
2019-09-07 18:56:03 +00:00
def test_validate_backtest_data_warn(default_conf, mocker, caplog, testdatadir) -> None:
patch_exchange(mocker)
2024-05-12 14:00:45 +00:00
default_conf.update({"strategy": CURRENT_TEST_STRATEGY})
strategy = StrategyResolver.load_strategy(default_conf)
data = strategy.advise_all_indicators(
2019-12-17 08:36:26 +00:00
load_data(
2024-05-12 14:00:45 +00:00
datadir=testdatadir, timeframe="1m", pairs=["UNITTEST/BTC"], fill_up_missing=False
)
)
2019-12-17 22:06:03 +00:00
min_date, max_date = get_timerange(data)
caplog.clear()
2024-05-12 14:00:45 +00:00
assert validate_backtest_data(
data["UNITTEST/BTC"], "UNITTEST/BTC", min_date, max_date, timeframe_to_minutes("1m")
)
assert len(caplog.record_tuples) == 1
assert log_has(
2022-09-28 18:23:56 +00:00
"UNITTEST/BTC has missing frames: expected 14397, got 13681, that's 716 missing values",
2024-05-12 14:00:45 +00:00
caplog,
)
2019-09-07 18:56:03 +00:00
def test_validate_backtest_data(default_conf, mocker, caplog, testdatadir) -> None:
patch_exchange(mocker)
2024-05-12 14:00:45 +00:00
default_conf.update({"strategy": CURRENT_TEST_STRATEGY})
strategy = StrategyResolver.load_strategy(default_conf)
timerange = TimeRange()
data = strategy.advise_all_indicators(
2024-05-12 14:00:45 +00:00
load_data(datadir=testdatadir, timeframe="5m", pairs=["UNITTEST/BTC"], timerange=timerange)
)
2019-12-17 22:06:03 +00:00
min_date, max_date = get_timerange(data)
caplog.clear()
2024-05-12 14:00:45 +00:00
assert not validate_backtest_data(
data["UNITTEST/BTC"], "UNITTEST/BTC", min_date, max_date, timeframe_to_minutes("5m")
)
assert len(caplog.record_tuples) == 0
2019-08-25 13:02:40 +00:00
2024-05-12 14:00:45 +00:00
@pytest.mark.parametrize(
"trademode,callcount",
[
("spot", 4),
("margin", 4),
("futures", 8), # Called 8 times - 4 normal, 2 funding and 2 mark/index calls
],
)
2021-12-03 13:57:09 +00:00
def test_refresh_backtest_ohlcv_data(
2024-05-12 14:00:45 +00:00
mocker, default_conf, markets, caplog, testdatadir, trademode, callcount
):
2023-06-17 12:55:23 +00:00
caplog.set_level(logging.DEBUG)
2024-05-12 14:00:45 +00:00
dl_mock = mocker.patch("freqtrade.data.history.history_utils._download_pair_history")
mocker.patch(f"{EXMS}.markets", PropertyMock(return_value=markets))
2023-06-17 12:55:23 +00:00
2019-08-25 13:02:40 +00:00
mocker.patch.object(Path, "exists", MagicMock(return_value=True))
mocker.patch.object(Path, "unlink", MagicMock())
2024-05-12 14:00:45 +00:00
default_conf["trading_mode"] = trademode
2019-08-25 13:02:40 +00:00
ex = get_patched_exchange(mocker, default_conf, exchange="bybit")
2019-08-25 13:02:40 +00:00
timerange = TimeRange.parse_timerange("20190101-20190102")
2024-05-12 14:00:45 +00:00
refresh_backtest_ohlcv_data(
exchange=ex,
pairs=["ETH/BTC", "XRP/BTC"],
timeframes=["1m", "5m"],
datadir=testdatadir,
timerange=timerange,
erase=True,
trading_mode=trademode,
)
2019-08-25 13:02:40 +00:00
2021-12-03 13:57:09 +00:00
assert dl_mock.call_count == callcount
2024-05-12 14:00:45 +00:00
assert dl_mock.call_args[1]["timerange"].starttype == "date"
2019-08-25 13:02:40 +00:00
2023-06-17 12:55:23 +00:00
assert log_has_re(r"Downloading pair ETH/BTC, .* interval 1m\.", caplog)
2024-05-12 14:00:45 +00:00
if trademode == "futures":
assert log_has_re(r"Downloading pair ETH/BTC, funding_rate, interval 8h\.", caplog)
assert log_has_re(r"Downloading pair ETH/BTC, mark, interval 4h\.", caplog)
2019-08-25 13:02:40 +00:00
2019-09-07 18:56:03 +00:00
def test_download_data_no_markets(mocker, default_conf, caplog, testdatadir):
2024-05-12 14:00:45 +00:00
dl_mock = mocker.patch(
"freqtrade.data.history.history_utils._download_pair_history", MagicMock()
)
2019-10-26 11:24:26 +00:00
ex = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.markets", PropertyMock(return_value={}))
2019-08-25 13:02:40 +00:00
timerange = TimeRange.parse_timerange("20190101-20190102")
2024-05-12 14:00:45 +00:00
unav_pairs = refresh_backtest_ohlcv_data(
exchange=ex,
pairs=["BTT/BTC", "LTC/USDT"],
timeframes=["1m", "5m"],
datadir=testdatadir,
timerange=timerange,
erase=False,
trading_mode="spot",
)
2019-08-25 13:02:40 +00:00
assert dl_mock.call_count == 0
2019-10-26 11:24:26 +00:00
assert "BTT/BTC" in unav_pairs
assert "LTC/USDT" in unav_pairs
assert log_has("Skipping pair BTT/BTC...", caplog)
def test_refresh_backtest_trades_data(mocker, default_conf, markets, caplog, testdatadir):
2024-05-12 14:00:45 +00:00
dl_mock = mocker.patch(
"freqtrade.data.history.history_utils._download_trades_history", MagicMock()
)
mocker.patch(f"{EXMS}.markets", PropertyMock(return_value=markets))
mocker.patch.object(Path, "exists", MagicMock(return_value=True))
mocker.patch.object(Path, "unlink", MagicMock())
ex = get_patched_exchange(mocker, default_conf)
timerange = TimeRange.parse_timerange("20190101-20190102")
2024-05-12 14:00:45 +00:00
unavailable_pairs = refresh_backtest_trades_data(
exchange=ex,
pairs=["ETH/BTC", "XRP/BTC", "XRP/ETH"],
datadir=testdatadir,
timerange=timerange,
erase=True,
trading_mode=TradingMode.SPOT,
)
assert dl_mock.call_count == 2
2024-05-12 14:00:45 +00:00
assert dl_mock.call_args[1]["timerange"].starttype == "date"
assert log_has("Downloading trades for pair ETH/BTC.", caplog)
assert unavailable_pairs == ["XRP/ETH"]
assert log_has("Skipping pair XRP/ETH...", caplog)
2024-05-12 14:00:45 +00:00
def test_download_trades_history(
trades_history, mocker, default_conf, testdatadir, caplog, tmp_path, time_machine
) -> None:
2023-08-20 09:51:01 +00:00
start_dt = dt_utc(2023, 1, 1)
time_machine.move_to(start_dt, tick=False)
ght_mock = MagicMock(side_effect=lambda pair, *args, **kwargs: (pair, trades_history))
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_trades", ght_mock)
exchange = get_patched_exchange(mocker, default_conf)
2024-05-12 14:00:45 +00:00
file1 = tmp_path / "ETH_BTC-trades.json.gz"
data_handler = get_datahandler(tmp_path, data_format="jsongz")
assert not file1.is_file()
2024-05-12 14:00:45 +00:00
assert _download_trades_history(
data_handler=data_handler, exchange=exchange, pair="ETH/BTC", trading_mode=TradingMode.SPOT
)
2023-08-20 09:51:01 +00:00
assert log_has("Current Amount of trades: 0", caplog)
assert log_has("New Amount of trades: 6", caplog)
2023-08-20 09:51:01 +00:00
assert ght_mock.call_count == 1
# Default "since" - 30 days before current day.
2024-05-12 14:00:45 +00:00
assert ght_mock.call_args_list[0][1]["since"] == dt_ts(start_dt - timedelta(days=30))
assert file1.is_file()
2023-08-20 09:51:01 +00:00
caplog.clear()
2020-04-01 18:04:36 +00:00
ght_mock.reset_mock()
2020-04-01 18:31:21 +00:00
since_time = int(trades_history[-3][0] // 1000)
since_time2 = int(trades_history[-1][0] // 1000)
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, since_time, 0)
2024-03-02 12:17:45 +00:00
assert _download_trades_history(
2024-05-12 14:00:45 +00:00
data_handler=data_handler,
exchange=exchange,
pair="ETH/BTC",
timerange=timerange,
trading_mode=TradingMode.SPOT,
)
2020-04-01 18:04:36 +00:00
assert ght_mock.call_count == 1
# Check this in seconds - since we had to convert to seconds above too.
2024-05-12 14:00:45 +00:00
assert int(ght_mock.call_args_list[0][1]["since"] // 1000) == since_time2 - 5
assert ght_mock.call_args_list[0][1]["from_id"] is not None
2020-04-01 18:04:36 +00:00
file1.unlink()
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_trades", MagicMock(side_effect=ValueError))
2023-08-20 09:51:01 +00:00
caplog.clear()
2024-05-12 14:00:45 +00:00
assert not _download_trades_history(
data_handler=data_handler, exchange=exchange, pair="ETH/BTC", trading_mode=TradingMode.SPOT
)
assert log_has_re('Failed to download historic trades for pair: "ETH/BTC".*', caplog)
2024-05-12 14:00:45 +00:00
file2 = tmp_path / "XRP_ETH-trades.json.gz"
copyfile(testdatadir / file2.name, file2)
ght_mock.reset_mock()
2024-05-12 14:00:45 +00:00
mocker.patch(f"{EXMS}.get_historic_trades", ght_mock)
# Since before first start date
since_time = int(trades_history[0][0] // 1000) - 500
2024-05-12 14:00:45 +00:00
timerange = TimeRange("date", None, since_time, 0)
2024-03-02 12:17:45 +00:00
assert _download_trades_history(
2024-05-12 14:00:45 +00:00
data_handler=data_handler,
exchange=exchange,
pair="XRP/ETH",
timerange=timerange,
trading_mode=TradingMode.SPOT,
)
assert ght_mock.call_count == 1
2024-05-12 14:00:45 +00:00
assert int(ght_mock.call_args_list[0][1]["since"] // 1000) == since_time
assert ght_mock.call_args_list[0][1]["from_id"] is None
assert log_has_re(r"Start .* earlier than available data. Redownloading trades for.*", caplog)
_clean_test_file(file2)