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

remove normalizatio test from kitchen and add pickle metadata assertions

Browse Source
This commit is contained in:
Zohar Kol 2023-03-30 10:20:53 +03:00
parent 1d8327bb95
commit 630b2b1996
2 changed files with 2 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tests/freqai/test_freqai_datakitchen.py
View File

@ -125,14 +125,6 @@ def test_principal_component_analysis(mocker, freqai_conf, caplog):
) )
def test_normalize_data(mocker, freqai_conf):
freqai = make_data_dictionary(mocker, freqai_conf)
data_dict = freqai.dk.data_dictionary
freqai.dk.normalize_data(data_dict)
assert any('_max' in entry for entry in freqai.dk.data.keys())
assert any('_min' in entry for entry in freqai.dk.data.keys())
def test_filter_features(mocker, freqai_conf): def test_filter_features(mocker, freqai_conf):
freqai, unfiltered_dataframe = make_unfiltered_dataframe(mocker, freqai_conf) freqai, unfiltered_dataframe = make_unfiltered_dataframe(mocker, freqai_conf)
freqai.dk.find_features(unfiltered_dataframe) freqai.dk.find_features(unfiltered_dataframe)

2
tests/freqai/test_normalization.py
View File

@ -33,6 +33,7 @@ def test_legacy_normalization_add_max_min_columns(mocker, freqai_conf):
assert any('_max' in entry for entry in freqai.dk.data.keys()) assert any('_max' in entry for entry in freqai.dk.data.keys())
assert any('_min' in entry for entry in freqai.dk.data.keys()) assert any('_min' in entry for entry in freqai.dk.data.keys())
assert all(not entry.endswith('_scaler') for entry in freqai.dk.pkl_data.keys())
def test_standard_normalization_dont_add_max_min_columns(mocker, freqai_conf): def test_standard_normalization_dont_add_max_min_columns(mocker, freqai_conf):
@ -42,6 +43,7 @@ def test_standard_normalization_dont_add_max_min_columns(mocker, freqai_conf):
freqai.dk.normalize_data(data_dict) freqai.dk.normalize_data(data_dict)
assert all(not entry.endswith('_max') for entry in freqai.dk.data.keys()) assert all(not entry.endswith('_max') for entry in freqai.dk.data.keys())
assert all(not entry.endswith('_min') for entry in freqai.dk.data.keys()) assert all(not entry.endswith('_min') for entry in freqai.dk.data.keys())
assert any(entry.endswith('_scaler') for entry in freqai.dk.pkl_data.keys())
def test_legacy_and_standard_normalization_difference(mocker, freqai_conf): def test_legacy_and_standard_normalization_difference(mocker, freqai_conf):