adding result objects, renaming components
This commit is contained in:
@@ -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):
|
||||
|
||||
@@ -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,
|
||||
|
||||
Reference in New Issue
Block a user