Impove tests for yearly resample

This commit is contained in:
Matthias
2024-01-23 07:22:38 +01:00
parent 48ea43f954
commit 8fd2dcd257
2 changed files with 4 additions and 1 deletions

View File

@@ -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')

View File

@@ -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')