ADd test for data_interpolate
This commit is contained in:
parent
a021cd3ae2
commit
03389d961f
@ -64,6 +64,7 @@ def ohlcv_fill_up_missing_data(dataframe: DataFrame, ticker_interval: str) -> Da
|
|||||||
'low': df['close'],
|
'low': df['close'],
|
||||||
})
|
})
|
||||||
df.reset_index(inplace=True)
|
df.reset_index(inplace=True)
|
||||||
|
logger.debug(f"Missing data fillup: before: {len(dataframe)} - after: {len(df)}")
|
||||||
return df
|
return df
|
||||||
|
|
||||||
|
|
||||||
|
@ -1,7 +1,8 @@
|
|||||||
# pragma pylint: disable=missing-docstring, C0103
|
# pragma pylint: disable=missing-docstring, C0103
|
||||||
import logging
|
import logging
|
||||||
|
|
||||||
from freqtrade.data.converter import parse_ticker_dataframe
|
from freqtrade.data.converter import parse_ticker_dataframe, ohlcv_fill_up_missing_data
|
||||||
|
from freqtrade.data.history import load_pair_history
|
||||||
from freqtrade.tests.conftest import log_has
|
from freqtrade.tests.conftest import log_has
|
||||||
|
|
||||||
|
|
||||||
@ -11,8 +12,7 @@ def test_dataframe_correct_length(result):
|
|||||||
|
|
||||||
|
|
||||||
def test_dataframe_correct_columns(result):
|
def test_dataframe_correct_columns(result):
|
||||||
assert result.columns.tolist() == \
|
assert result.columns.tolist() == ['date', 'open', 'high', 'low', 'close', 'volume']
|
||||||
['date', 'open', 'high', 'low', 'close', 'volume']
|
|
||||||
|
|
||||||
|
|
||||||
def test_parse_ticker_dataframe(ticker_history, caplog):
|
def test_parse_ticker_dataframe(ticker_history, caplog):
|
||||||
@ -23,3 +23,18 @@ def test_parse_ticker_dataframe(ticker_history, caplog):
|
|||||||
dataframe = parse_ticker_dataframe(ticker_history)
|
dataframe = parse_ticker_dataframe(ticker_history)
|
||||||
assert dataframe.columns.tolist() == columns
|
assert dataframe.columns.tolist() == columns
|
||||||
assert log_has('Parsing tickerlist to dataframe', caplog.record_tuples)
|
assert log_has('Parsing tickerlist to dataframe', caplog.record_tuples)
|
||||||
|
|
||||||
|
|
||||||
|
def test_ohlcv_fill_up_missing_data(caplog):
|
||||||
|
data = load_pair_history(datadir=None,
|
||||||
|
ticker_interval='1m',
|
||||||
|
refresh_pairs=False,
|
||||||
|
pair='UNITTEST/BTC')
|
||||||
|
caplog.set_level(logging.DEBUG)
|
||||||
|
data2 = ohlcv_fill_up_missing_data(data, '1m')
|
||||||
|
assert len(data2) > len(data)
|
||||||
|
# Column names should not change
|
||||||
|
assert (data.columns == data2.columns).all()
|
||||||
|
|
||||||
|
assert log_has(f"Missing data fillup: before: {len(data)} - after: {len(data2)}",
|
||||||
|
caplog.record_tuples)
|
||||||
|
Loading…
Reference in New Issue
Block a user