adding result objects, renaming components

This commit is contained in:
2025-03-05 14:35:29 +01:00
parent ec63840ccc
commit c375d8e9d4
7 changed files with 73 additions and 57 deletions

View File

@@ -5,18 +5,18 @@ 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)
res = fc.sales_per_customer(sales_data, customer_id)
assert err == 0
assert res is not None
assert res.status.status_code == 0
assert res.data 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)
res = fc.sales_per_customer(sales_data, customer_id)
assert err == 1
assert res is None
assert res.status.status_code == 1
assert res.data is None
def test_parse_api_resp_to_df(exmpl_api_sales_prognosis_resp):

View File

@@ -5,25 +5,25 @@ from typing import cast
import delta_barth._management
from delta_barth import errors
from delta_barth.types import doptResponse
from delta_barth.types import Status
def test_error_handler_parsing():
predef_errs = errors.DATA_PIPELINE_ERRORS_DESCR
predef_errs = errors.DATA_PIPELINE_STATUS_DESCR
err_hdlr = delta_barth._management.ErrorHandler()
assert err_hdlr.data_pipelines is not None
parsed_pipe_errs = err_hdlr.data_pipelines
assert err_hdlr.pipe_states is not None
parsed_pipe_errs = err_hdlr.pipe_states
parsed_pipe_errs = asdict(parsed_pipe_errs)
for err in predef_errs:
dopt_err = cast(doptResponse, parsed_pipe_errs[err[0]])
assert isinstance(dopt_err, doptResponse)
dopt_err = cast(Status, parsed_pipe_errs[err[0]])
assert isinstance(dopt_err, Status)
assert dopt_err.status_code == err[1]
assert dopt_err.description == err[2]
assert dopt_err.message == ""
err_hdlr._parse_data_pipeline_errors()
err_hdlr._parse_data_pipe_states()
def test_error_handler_internal():
@@ -32,7 +32,7 @@ def test_error_handler_internal():
ERR_CODE = 101
err_hdlr = delta_barth._management.ErrorHandler()
new_err = err_hdlr.internal_error(
new_err = err_hdlr.error(
description=DESCRIPTION,
message=MESSAGE,
err_code=ERR_CODE,