adding test cases

2025-02-19 12:47:07 +01:00
parent 8fa429d477
commit 0c76dc0325
4 changed files with 125485 additions and 0 deletions
--- a/tests/analysis/init.py
+++ b/tests/analysis/init.py
--- a/tests/analysis/test_forecast.py
+++ b/tests/analysis/test_forecast.py
@@ -0,0 +1,17 @@
+from delta_barth.analysis import forecast as fc
+
+
+def test_sales_per_customer_success(sales_data):
+    customer_id = 1133
+    err, res = fc.sales_per_customer(sales_data, customer_id)
+
+    assert err == 0
+    assert res is not None
+
+
+def test_sales_per_customer_too_few_data_points(sales_data):
+    customer_id = 1000
+    err, res = fc.sales_per_customer(sales_data, customer_id)
+
+    assert err == 1
+    assert res is None