Fix generate-test-data for 1w data

This commit is contained in:
Matthias
2023-12-08 20:34:23 +01:00
parent 2745a5d334
commit 7321a14c35

View File

@@ -91,6 +91,8 @@ def generate_test_data(timeframe: str, size: int, start: str = '2020-07-05'):
base = np.random.normal(20, 2, size=size) base = np.random.normal(20, 2, size=size)
if timeframe == '1M': if timeframe == '1M':
date = pd.date_range(start, periods=size, freq='1MS', tz='UTC') date = pd.date_range(start, periods=size, freq='1MS', tz='UTC')
elif timeframe == '1w':
date = pd.date_range(start, periods=size, freq='1W-MON', tz='UTC')
else: else:
tf_mins = timeframe_to_minutes(timeframe) tf_mins = timeframe_to_minutes(timeframe)
date = pd.date_range(start, periods=size, freq=f'{tf_mins}min', tz='UTC') date = pd.date_range(start, periods=size, freq=f'{tf_mins}min', tz='UTC')