From 8f883f2310a05d6401616a3f1ee9a6b12981be6f Mon Sep 17 00:00:00 2001 From: robcaulk Date: Tue, 19 Sep 2023 12:20:44 +0200 Subject: [PATCH] chore: fix set_initial_hist_preds test --- tests/freqai/test_freqai_datadrawer.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/tests/freqai/test_freqai_datadrawer.py b/tests/freqai/test_freqai_datadrawer.py index b511f3c7b..ca4749747 100644 --- a/tests/freqai/test_freqai_datadrawer.py +++ b/tests/freqai/test_freqai_datadrawer.py @@ -181,7 +181,6 @@ def test_set_initial_return_values(mocker, freqai_conf): 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 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] 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 hist_pred_df.shape[0] == 9 # Total rows: 5 from historic and 4 new zeros