diff --git a/tests/data/test_dataprovider.py b/tests/data/test_dataprovider.py index 220aafef0..bef894eff 100644 --- a/tests/data/test_dataprovider.py +++ b/tests/data/test_dataprovider.py @@ -92,11 +92,11 @@ def test_historic_trades(mocker, default_conf, trades_history_df): def test_historic_ohlcv_dataformat(mocker, default_conf, ohlcv_history): - hdf5loadmock = MagicMock(return_value=ohlcv_history) + parquetloadmock = MagicMock(return_value=ohlcv_history) featherloadmock = MagicMock(return_value=ohlcv_history) mocker.patch( - "freqtrade.data.history.datahandlers.hdf5datahandler.HDF5DataHandler._ohlcv_load", - hdf5loadmock, + "freqtrade.data.history.datahandlers.parquetdatahandler.ParquetDataHandler._ohlcv_load", + parquetloadmock, ) mocker.patch( "freqtrade.data.history.datahandlers.featherdatahandler.FeatherDataHandler._ohlcv_load", @@ -108,17 +108,17 @@ def test_historic_ohlcv_dataformat(mocker, default_conf, ohlcv_history): dp = DataProvider(default_conf, exchange) data = dp.historic_ohlcv("UNITTEST/BTC", "5m") assert isinstance(data, DataFrame) - hdf5loadmock.assert_not_called() + parquetloadmock.assert_not_called() featherloadmock.assert_called_once() - # Switching to dataformat hdf5 - hdf5loadmock.reset_mock() + # Switching to dataformat parquet + parquetloadmock.reset_mock() featherloadmock.reset_mock() - default_conf["dataformat_ohlcv"] = "hdf5" + default_conf["dataformat_ohlcv"] = "parquet" dp = DataProvider(default_conf, exchange) data = dp.historic_ohlcv("UNITTEST/BTC", "5m") assert isinstance(data, DataFrame) - hdf5loadmock.assert_called_once() + parquetloadmock.assert_called_once() featherloadmock.assert_not_called()