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

chore: fix set_initial_hist_preds test

Browse Source
This commit is contained in:
robcaulk
2023-09-19 12:20:44 +02:00
parent 1f1abfe798
commit 8f883f2310
1 changed files with 0 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/freqai/test_freqai_datadrawer.py
-2
View File
@@ -181,7 +181,6 @@ def test_set_initial_return_values(mocker, freqai_conf):
assert (hist_pred_df['date_pred'].iloc[-1] == assert (hist_pred_df['date_pred'].iloc[-1] ==
pd.Timestamp(end_x_plus_5) - pd.Timedelta(days=1)) pd.Timestamp(end_x_plus_5) - pd.Timedelta(days=1))
assert 'date' not in hist_pred_df.columns
assert 'date_pred' in hist_pred_df.columns assert 'date_pred' in hist_pred_df.columns
assert hist_pred_df.shape[0] == 7 # Total rows: 5 from historic and 2 new zeros assert hist_pred_df.shape[0] == 7 # Total rows: 5 from historic and 2 new zeros
@@ -236,7 +235,6 @@ def test_set_initial_return_values_warning(mocker, freqai_conf):
model_return_df = freqai.dd.model_return_values[pair] model_return_df = freqai.dd.model_return_values[pair]
assert hist_pred_df['date_pred'].iloc[-1] == pd.Timestamp(end_x_plus_5) - pd.Timedelta(days=1) assert hist_pred_df['date_pred'].iloc[-1] == pd.Timestamp(end_x_plus_5) - pd.Timedelta(days=1)
assert 'date' not in hist_pred_df.columns
assert 'date_pred' in hist_pred_df.columns assert 'date_pred' in hist_pred_df.columns
assert hist_pred_df.shape[0] == 9 # Total rows: 5 from historic and 4 new zeros assert hist_pred_df.shape[0] == 9 # Total rows: 5 from historic and 4 new zeros