From 64eac7f6cce3b5e17692e4b5fbe6d7d12cc136c0 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Florian=20F=C3=B6rster?= Date: Wed, 12 Mar 2025 16:01:22 +0100 Subject: [PATCH] tests for parsing --- tests/analysis/test_parse.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/analysis/test_parse.py b/tests/analysis/test_parse.py index 0047f8e..ef890f2 100644 --- a/tests/analysis/test_parse.py +++ b/tests/analysis/test_parse.py @@ -36,7 +36,7 @@ def test_map_features_to_targets(): def test_preprocess_features(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] = { "artikelId": "artikel_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()) - 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 (data.columns != df.columns).any() assert any(feat not in data.columns for feat in feat_mapping.keys())