tests: fix remaining hdf5 occurance
This commit is contained in:
@@ -92,11 +92,11 @@ def test_historic_trades(mocker, default_conf, trades_history_df):
|
|||||||
|
|
||||||
|
|
||||||
def test_historic_ohlcv_dataformat(mocker, default_conf, ohlcv_history):
|
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)
|
featherloadmock = MagicMock(return_value=ohlcv_history)
|
||||||
mocker.patch(
|
mocker.patch(
|
||||||
"freqtrade.data.history.datahandlers.hdf5datahandler.HDF5DataHandler._ohlcv_load",
|
"freqtrade.data.history.datahandlers.parquetdatahandler.ParquetDataHandler._ohlcv_load",
|
||||||
hdf5loadmock,
|
parquetloadmock,
|
||||||
)
|
)
|
||||||
mocker.patch(
|
mocker.patch(
|
||||||
"freqtrade.data.history.datahandlers.featherdatahandler.FeatherDataHandler._ohlcv_load",
|
"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)
|
dp = DataProvider(default_conf, exchange)
|
||||||
data = dp.historic_ohlcv("UNITTEST/BTC", "5m")
|
data = dp.historic_ohlcv("UNITTEST/BTC", "5m")
|
||||||
assert isinstance(data, DataFrame)
|
assert isinstance(data, DataFrame)
|
||||||
hdf5loadmock.assert_not_called()
|
parquetloadmock.assert_not_called()
|
||||||
featherloadmock.assert_called_once()
|
featherloadmock.assert_called_once()
|
||||||
|
|
||||||
# Switching to dataformat hdf5
|
# Switching to dataformat parquet
|
||||||
hdf5loadmock.reset_mock()
|
parquetloadmock.reset_mock()
|
||||||
featherloadmock.reset_mock()
|
featherloadmock.reset_mock()
|
||||||
default_conf["dataformat_ohlcv"] = "hdf5"
|
default_conf["dataformat_ohlcv"] = "parquet"
|
||||||
dp = DataProvider(default_conf, exchange)
|
dp = DataProvider(default_conf, exchange)
|
||||||
data = dp.historic_ohlcv("UNITTEST/BTC", "5m")
|
data = dp.historic_ohlcv("UNITTEST/BTC", "5m")
|
||||||
assert isinstance(data, DataFrame)
|
assert isinstance(data, DataFrame)
|
||||||
hdf5loadmock.assert_called_once()
|
parquetloadmock.assert_called_once()
|
||||||
featherloadmock.assert_not_called()
|
featherloadmock.assert_not_called()
|
||||||
|
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user