mohammadian7/freqtrade
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Enhance generate_test_data with parametrizable random seed

Browse Source
This commit is contained in:
Matthias
2024-02-24 13:30:38 +01:00
parent eaf70428c1
commit 31e2543134
1 changed files with 4 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/conftest.py
+4 -4
View File
@@ -142,8 +142,8 @@ def generate_trades_history(n_rows, start_date: Optional[datetime] = None, days=
return df return df
def generate_test_data(timeframe: str, size: int, start: str = '2020-07-05'): def generate_test_data(timeframe: str, size: int, start: str = '2020-07-05', random_seed=42):
np.random.seed(42) np.random.seed(random_seed)
base = np.random.normal(20, 2, size=size) base = np.random.normal(20, 2, size=size)
if timeframe == '1y': if timeframe == '1y':
@@ -174,9 +174,9 @@ def generate_test_data(timeframe: str, size: int, start: str = '2020-07-05'):
return df return df
def generate_test_data_raw(timeframe: str, size: int, start: str = '2020-07-05'): def generate_test_data_raw(timeframe: str, size: int, start: str = '2020-07-05', random_seed=42):
""" Generates data in the ohlcv format used by ccxt """ """ Generates data in the ohlcv format used by ccxt """
df = generate_test_data(timeframe, size, start) df = generate_test_data(timeframe, size, start, random_seed)
df['date'] = df.loc[:, 'date'].view(np.int64) // 1000 // 1000 df['date'] = df.loc[:, 'date'].view(np.int64) // 1000 // 1000
return list(list(x) for x in zip(*(df[x].values.tolist() for x in df.columns))) return list(list(x) for x in zip(*(df[x].values.tolist() for x in df.columns)))