delta-barth/delta-barth-py
1
0
Fork 0
Code Issues 3 Pull Requests Actions Packages Projects 3 Releases 3 Wiki Activity

tests for parsing

Browse Source
This commit is contained in:
Florian Förster 2025-03-12 16:01:22 +01:00
parent cdcd613db0
commit 64eac7f6cc
1 changed files with 2 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
tests/analysis/test_parse.py
View File

@ -36,7 +36,7 @@ def test_map_features_to_targets():
def test_preprocess_features(exmpl_api_sales_prognosis_resp): def test_preprocess_features(exmpl_api_sales_prognosis_resp):
resp = exmpl_api_sales_prognosis_resp resp = exmpl_api_sales_prognosis_resp
df = forecast.parse_api_resp_to_df(resp) df = forecast._parse_api_resp_to_df(resp)
feat_mapping: dict[str, str] = { feat_mapping: dict[str, str] = {
"artikelId": "artikel_refid", "artikelId": "artikel_refid",
"firmaId": "firma_refid", "firmaId": "firma_refid",
@ -53,7 +53,7 @@ def test_preprocess_features(exmpl_api_sales_prognosis_resp):
) )
assert all(feat in df.columns for feat in feat_mapping.keys()) assert all(feat in df.columns for feat in feat_mapping.keys())
data = parse.preprocess_features(df, feat_mapping, target_features) data = parse.process_features(df, feat_mapping, target_features)
assert len(data.columns) == len(df.columns) assert len(data.columns) == len(df.columns)
assert (data.columns != df.columns).any() assert (data.columns != df.columns).any()
assert any(feat not in data.columns for feat in feat_mapping.keys()) assert any(feat not in data.columns for feat in feat_mapping.keys())