Test Feather/parquet datahandler init
This commit is contained in:
parent
983a16d937
commit
5fb56b09f2
@ -9,9 +9,11 @@ from pandas import DataFrame
|
|||||||
|
|
||||||
from freqtrade.configuration import TimeRange
|
from freqtrade.configuration import TimeRange
|
||||||
from freqtrade.constants import AVAILABLE_DATAHANDLERS
|
from freqtrade.constants import AVAILABLE_DATAHANDLERS
|
||||||
|
from freqtrade.data.history.featherdatahandler import FeatherDataHandler
|
||||||
from freqtrade.data.history.hdf5datahandler import HDF5DataHandler
|
from freqtrade.data.history.hdf5datahandler import HDF5DataHandler
|
||||||
from freqtrade.data.history.idatahandler import IDataHandler, get_datahandler, get_datahandlerclass
|
from freqtrade.data.history.idatahandler import IDataHandler, get_datahandler, get_datahandlerclass
|
||||||
from freqtrade.data.history.jsondatahandler import JsonDataHandler, JsonGzDataHandler
|
from freqtrade.data.history.jsondatahandler import JsonDataHandler, JsonGzDataHandler
|
||||||
|
from freqtrade.data.history.parquetdatahandler import ParquetDataHandler
|
||||||
from freqtrade.enums import CandleType, TradingMode
|
from freqtrade.enums import CandleType, TradingMode
|
||||||
from tests.conftest import log_has
|
from tests.conftest import log_has
|
||||||
|
|
||||||
@ -339,13 +341,24 @@ def test_gethandlerclass():
|
|||||||
cl = get_datahandlerclass('json')
|
cl = get_datahandlerclass('json')
|
||||||
assert cl == JsonDataHandler
|
assert cl == JsonDataHandler
|
||||||
assert issubclass(cl, IDataHandler)
|
assert issubclass(cl, IDataHandler)
|
||||||
|
|
||||||
cl = get_datahandlerclass('jsongz')
|
cl = get_datahandlerclass('jsongz')
|
||||||
assert cl == JsonGzDataHandler
|
assert cl == JsonGzDataHandler
|
||||||
assert issubclass(cl, IDataHandler)
|
assert issubclass(cl, IDataHandler)
|
||||||
assert issubclass(cl, JsonDataHandler)
|
assert issubclass(cl, JsonDataHandler)
|
||||||
|
|
||||||
cl = get_datahandlerclass('hdf5')
|
cl = get_datahandlerclass('hdf5')
|
||||||
assert cl == HDF5DataHandler
|
assert cl == HDF5DataHandler
|
||||||
assert issubclass(cl, IDataHandler)
|
assert issubclass(cl, IDataHandler)
|
||||||
|
|
||||||
|
cl = get_datahandlerclass('feather')
|
||||||
|
assert cl == FeatherDataHandler
|
||||||
|
assert issubclass(cl, IDataHandler)
|
||||||
|
|
||||||
|
cl = get_datahandlerclass('parquet')
|
||||||
|
assert cl == ParquetDataHandler
|
||||||
|
assert issubclass(cl, IDataHandler)
|
||||||
|
|
||||||
with pytest.raises(ValueError, match=r"No datahandler for .*"):
|
with pytest.raises(ValueError, match=r"No datahandler for .*"):
|
||||||
get_datahandlerclass('DeadBeef')
|
get_datahandlerclass('DeadBeef')
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user