adapt forecast pipeline to new output format

2025-03-05 15:06:24 +01:00
parent a1cb3ddbe9
commit 763d3c1aac
2 changed files with 5 additions and 7 deletions
--- a/tests/analysis/test_forecast.py
+++ b/tests/analysis/test_forecast.py
@@ -1,5 +1,3 @@
-import pytest
-
 from delta_barth.analysis import forecast as fc


@@ -7,7 +5,7 @@ def test_sales_per_customer_success(sales_data):
    customer_id = 1133
    res = fc.sales_per_customer(sales_data, customer_id)

-    assert res.status.status_code == 0
+    assert res.status.code == 0
    assert res.data is not None


@@ -15,7 +13,7 @@ def test_sales_per_customer_too_few_data_points(sales_data):
    customer_id = 1000
    res = fc.sales_per_customer(sales_data, customer_id)

-    assert res.status.status_code == 1
+    assert res.status.code == 1
    assert res.data is None