adelorenzo/stable
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

fix tests

Browse Source
This commit is contained in:
Timothy Pogue 2022-10-23 12:25:39 -06:00
parent 07e813dfa0
commit bb06745227
1 changed files with 8 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

18
tests/freqai/test_freqai_interface.py
View File

@ -160,12 +160,12 @@ def test_extract_data_and_train_model_Classifiers(mocker, freqai_conf, model):
@pytest.mark.parametrize( @pytest.mark.parametrize(
"model, num_files, strat", "model, num_files, strat",
[ [
("LightGBMRegressor", 7, "freqai_test_strat"), ("LightGBMRegressor", 6, "freqai_test_strat"),
("XGBoostRegressor", 7, "freqai_test_strat"), ("XGBoostRegressor", 6, "freqai_test_strat"),
("CatboostRegressor", 7, "freqai_test_strat"), ("CatboostRegressor", 6, "freqai_test_strat"),
("XGBoostClassifier", 7, "freqai_test_classifier"), ("XGBoostClassifier", 6, "freqai_test_classifier"),
("LightGBMClassifier", 7, "freqai_test_classifier"), ("LightGBMClassifier", 6, "freqai_test_classifier"),
("CatboostClassifier", 7, "freqai_test_classifier") ("CatboostClassifier", 6, "freqai_test_classifier")
], ],
) )
def test_start_backtesting(mocker, freqai_conf, model, num_files, strat, caplog): def test_start_backtesting(mocker, freqai_conf, model, num_files, strat, caplog):
@ -200,7 +200,6 @@ def test_start_backtesting(mocker, freqai_conf, model, num_files, strat, caplog)
freqai.start_backtesting(df, metadata, freqai.dk) freqai.start_backtesting(df, metadata, freqai.dk)
model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()] model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()]
# Changed from 6 to 7 because of the /configs directory
assert len(model_folders) == num_files assert len(model_folders) == num_files
assert log_has_re( assert log_has_re(
"Removed features ", "Removed features ",
@ -236,8 +235,7 @@ def test_start_backtesting_subdaily_backtest_period(mocker, freqai_conf):
freqai.start_backtesting(df, metadata, freqai.dk) freqai.start_backtesting(df, metadata, freqai.dk)
model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()] model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()]
# Changed from 9 to 10 because of the /configs dir assert len(model_folders) == 9
assert len(model_folders) == 10
shutil.rmtree(Path(freqai.dk.full_path)) shutil.rmtree(Path(freqai.dk.full_path))
@ -263,7 +261,7 @@ def test_start_backtesting_from_existing_folder(mocker, freqai_conf, caplog):
freqai.start_backtesting(df, metadata, freqai.dk) freqai.start_backtesting(df, metadata, freqai.dk)
model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()] model_folders = [x for x in freqai.dd.full_path.iterdir() if x.is_dir()]
assert len(model_folders) == 7 assert len(model_folders) == 6
# without deleting the existing folder structure, re-run # without deleting the existing folder structure, re-run