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

cleanup

Browse Source
This commit is contained in:
Florian Förster 2025-03-13 07:38:25 +01:00
parent 9a06ec9ac8
commit 1849df94c9
1 changed files with 3 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
tests/analysis/test_forecast.py
View File

@ -6,6 +6,7 @@ import pandas as pd
import pytest import pytest
from pydantic import ValidationError from pydantic import ValidationError
import delta_barth.analysis.forecast
from delta_barth.analysis import forecast as fc from delta_barth.analysis import forecast as fc
from delta_barth.errors import STATUS_HANDLER from delta_barth.errors import STATUS_HANDLER
from delta_barth.types import DualDict, PipeResult from delta_barth.types import DualDict, PipeResult
@ -224,19 +225,15 @@ def test_postprocess_sales_per_customer_FailValidation(
assert "ValidationError" in pipe.status.description assert "ValidationError" in pipe.status.description
def test_sales_prognosis_pipeline(monkeypatch, exmpl_api_sales_prognosis_resp): def test_sales_prognosis_pipeline(exmpl_api_sales_prognosis_resp):
def mock_request(*args, **kwargs): def mock_request(*args, **kwargs):
return exmpl_api_sales_prognosis_resp, STATUS_HANDLER.SUCCESS return exmpl_api_sales_prognosis_resp, STATUS_HANDLER.SUCCESS
import delta_barth.api.requests
monkeypatch.setattr(delta_barth.api.requests, "get_sales_prognosis_data", mock_request)
importlib.reload(delta_barth.api.requests)
with patch( with patch(
"delta_barth.api.requests.get_sales_prognosis_data", "delta_barth.api.requests.get_sales_prognosis_data",
new=mock_request, new=mock_request,
): ):
importlib.reload(delta_barth.analysis.forecast) # type: ignore importlib.reload(delta_barth.analysis.forecast)
result = fc.pipeline(None) # type: ignore result = fc.pipeline(None) # type: ignore
assert result.status == STATUS_HANDLER.SUCCESS assert result.status == STATUS_HANDLER.SUCCESS