From a1efd6b7830c43502edb16847e28f1af892851ef Mon Sep 17 00:00:00 2001 From: Matthias Date: Sun, 9 Jul 2023 17:21:45 +0200 Subject: [PATCH] Update further tests to use feather --- tests/data/test_history.py | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/tests/data/test_history.py b/tests/data/test_history.py index ab2238b08..c6cb73d70 100644 --- a/tests/data/test_history.py +++ b/tests/data/test_history.py @@ -68,7 +68,7 @@ def test_load_data_7min_timeframe(caplog, testdatadir) -> None: def test_load_data_1min_timeframe(ohlcv_history, mocker, caplog, testdatadir) -> None: mocker.patch(f'{EXMS}.get_historic_ohlcv', return_value=ohlcv_history) - file = testdatadir / 'UNITTEST_BTC-1m.json' + file = testdatadir / 'UNITTEST_BTC-1m.feather' load_data(datadir=testdatadir, timeframe='1m', pairs=['UNITTEST/BTC']) assert file.is_file() assert not log_has( @@ -79,7 +79,7 @@ def test_load_data_1min_timeframe(ohlcv_history, mocker, caplog, testdatadir) -> def test_load_data_mark(ohlcv_history, mocker, caplog, testdatadir) -> None: mocker.patch(f'{EXMS}.get_historic_ohlcv', return_value=ohlcv_history) - file = testdatadir / 'futures/UNITTEST_USDT_USDT-1h-mark.json' + 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( @@ -90,7 +90,7 @@ def test_load_data_mark(ohlcv_history, mocker, caplog, testdatadir) -> None: def test_load_data_startup_candles(mocker, testdatadir) -> None: ltfmock = mocker.patch( - 'freqtrade.data.history.jsondatahandler.JsonDataHandler._ohlcv_load', + 'freqtrade.data.history.featherdatahandler.FeatherDataHandler._ohlcv_load', MagicMock(return_value=DataFrame())) timerange = TimeRange('date', None, 1510639620, 0) load_pair_history(pair='UNITTEST/BTC', timeframe='1m', @@ -112,7 +112,7 @@ def test_load_data_with_new_pair_1min(ohlcv_history_list, mocker, caplog, tmpdir1 = Path(tmpdir) mocker.patch(f'{EXMS}.get_historic_ohlcv', return_value=ohlcv_history_list) exchange = get_patched_exchange(mocker, default_conf) - file = tmpdir1 / 'MEME_BTC-1m.json' + file = tmpdir1 / 'MEME_BTC-1m.feather' # do not download a new pair if refresh_pairs isn't set load_pair_history(datadir=tmpdir1, timeframe='1m', pair='MEME/BTC', candle_type=candle_type) @@ -326,7 +326,7 @@ def test_download_pair_history2(mocker, default_conf, testdatadir) -> None: [1509836580000, 0.00161, 0.00161, 0.00161, 0.00161, 82.390199] ] json_dump_mock = mocker.patch( - 'freqtrade.data.history.jsondatahandler.JsonDataHandler.ohlcv_store', + 'freqtrade.data.history.featherdatahandler.FeatherDataHandler.ohlcv_store', return_value=None) mocker.patch(f'{EXMS}.get_historic_ohlcv', return_value=tick) exchange = get_patched_exchange(mocker, default_conf) @@ -627,8 +627,8 @@ def test_download_trades_history(trades_history, mocker, default_conf, testdatad def test_convert_trades_to_ohlcv(testdatadir, tmpdir, caplog): tmpdir1 = Path(tmpdir) pair = 'XRP/ETH' - file1 = tmpdir1 / 'XRP_ETH-1m.json' - file5 = tmpdir1 / 'XRP_ETH-5m.json' + file1 = tmpdir1 / 'XRP_ETH-1m.feather' + file5 = tmpdir1 / 'XRP_ETH-5m.feather' filetrades = tmpdir1 / 'XRP_ETH-trades.json.gz' copyfile(testdatadir / file1.name, file1) copyfile(testdatadir / file5.name, file5)