From 3f6d427084b8d2a7b6def504b3083b3d816623f5 Mon Sep 17 00:00:00 2001 From: robcaulk Date: Sun, 14 Aug 2022 21:46:37 +0200 Subject: [PATCH] add a check for number of training features in tests --- tests/freqai/test_freqai_interface.py | 1 + 1 file changed, 1 insertion(+) diff --git a/tests/freqai/test_freqai_interface.py b/tests/freqai/test_freqai_interface.py index a36f80280..3de8561ed 100644 --- a/tests/freqai/test_freqai_interface.py +++ b/tests/freqai/test_freqai_interface.py @@ -71,6 +71,7 @@ def test_train_model_in_series_LightGBMMultiModel(mocker, freqai_conf): assert Path(freqai.dk.data_path / f"{freqai.dk.model_filename}_metadata.json").is_file() assert Path(freqai.dk.data_path / f"{freqai.dk.model_filename}_trained_df.pkl").is_file() assert Path(freqai.dk.data_path / f"{freqai.dk.model_filename}_svm_model.joblib").is_file() + assert(len(freqai.dk.data['training_features_list']) == 26) shutil.rmtree(Path(freqai.dk.full_path))