diff --git a/tests/conftest.py b/tests/conftest.py index e175f2034..8ef40408d 100644 --- a/tests/conftest.py +++ b/tests/conftest.py @@ -110,7 +110,9 @@ def generate_test_data(timeframe: str, size: int, start: str = '2020-07-05'): np.random.seed(42) base = np.random.normal(20, 2, size=size) - if timeframe == '1M': + if timeframe == '1y': + date = pd.date_range(start, periods=size, freq='1YS', tz='UTC') + elif timeframe == '1M': date = pd.date_range(start, periods=size, freq='1MS', tz='UTC') elif timeframe == '3M': date = pd.date_range(start, periods=size, freq='3MS', tz='UTC') diff --git a/tests/data/test_converter.py b/tests/data/test_converter.py index 245a882b6..c8a961951 100644 --- a/tests/data/test_converter.py +++ b/tests/data/test_converter.py @@ -157,6 +157,7 @@ def test_ohlcv_to_dataframe_multi(timeframe): assert len(df) == len(data) - 1 df1 = ohlcv_to_dataframe(data, timeframe, 'UNITTEST/USDT', drop_incomplete=False) assert len(df1) == len(data) + assert data.equals(df1) data1 = data.copy() data1.loc[:, 'date'] = data1.loc[:, 'date'] + pd.to_timedelta('30s')