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