132 lines
4.6 KiB
Python
132 lines
4.6 KiB
Python
"""运输计划多段：内部客户地磅默认值与数量平衡校验"""
|
|
from decimal import Decimal
|
|
|
|
from app.services.pl_internal_customer import INTERNAL_CUSTOMER_NAMES
|
|
from app.services.pl_plan_stop_rules import (
|
|
normalize_stop_dibang_fields,
|
|
resolve_stop_dibang_defaults,
|
|
warn_qty_imbalance,
|
|
)
|
|
|
|
|
|
def test_internal_shanxia_load_outbound():
|
|
name = "安庆市山峡化工有限公司"
|
|
assert name in INTERNAL_CUSTOMER_NAMES
|
|
assert resolve_stop_dibang_defaults(name, "load") == (True, "outbound")
|
|
|
|
|
|
def test_internal_shanxia_delivery_inbound():
|
|
name = "安庆市山峡化工有限公司"
|
|
assert resolve_stop_dibang_defaults(name, "delivery") == (True, "inbound")
|
|
|
|
|
|
def test_internal_yundong_load_outbound():
|
|
name = "安徽云栋化工有限公司"
|
|
assert resolve_stop_dibang_defaults(name, "load") == (True, "outbound")
|
|
|
|
|
|
def test_internal_yundong_delivery_inbound():
|
|
name = "安徽云栋化工有限公司"
|
|
assert resolve_stop_dibang_defaults(name, "delivery") == (True, "inbound")
|
|
|
|
|
|
def test_external_customer_no_dibang():
|
|
assert resolve_stop_dibang_defaults("外部客户有限公司", "load") == (False, "none")
|
|
assert resolve_stop_dibang_defaults("外部客户有限公司", "delivery") == (False, "none")
|
|
|
|
|
|
def test_delivery_defaults_use_delivery_customer():
|
|
"""送货段地磅默认以本段客户是否为内部客户为准。"""
|
|
internal = "安庆市山峡化工有限公司"
|
|
assert resolve_stop_dibang_defaults(internal, "delivery") == (True, "inbound")
|
|
assert resolve_stop_dibang_defaults("外部客户有限公司", "delivery") == (False, "none")
|
|
|
|
|
|
def test_normalize_stop_dibang_fields_repairs_none_direction():
|
|
gen, direction = normalize_stop_dibang_fields(
|
|
"安庆市山峡化工有限公司",
|
|
"delivery",
|
|
1,
|
|
"none",
|
|
)
|
|
assert gen == 1
|
|
assert direction == "inbound"
|
|
|
|
|
|
def test_external_empty_name_no_dibang():
|
|
assert resolve_stop_dibang_defaults(None, "load") == (False, "none")
|
|
assert resolve_stop_dibang_defaults(" ", "delivery") == (False, "none")
|
|
|
|
|
|
def test_warn_qty_balanced_returns_empty():
|
|
plan = [{"quantity": Decimal("30")}]
|
|
load = [{"plan_qty": Decimal("30")}]
|
|
delivery = [{"plan_qty": Decimal("30")}]
|
|
assert warn_qty_imbalance(plan, load, delivery) == []
|
|
|
|
|
|
def test_warn_qty_uses_plan_qty_on_plan_products():
|
|
plan = [{"plan_qty": Decimal("28")}]
|
|
load = [{"plan_qty": Decimal("28")}]
|
|
delivery = [{"plan_qty": Decimal("28")}]
|
|
assert warn_qty_imbalance(plan, load, delivery) == []
|
|
|
|
|
|
def test_warn_qty_load_vs_delivery_imbalance():
|
|
plan = [{"quantity": Decimal("28")}]
|
|
load = [{"plan_qty": Decimal("30")}]
|
|
delivery = [{"plan_qty": Decimal("28")}]
|
|
warnings = warn_qty_imbalance(plan, load, delivery)
|
|
assert warnings == [
|
|
"装货合计(30)与送货合计(28)不一致，请核对",
|
|
]
|
|
|
|
|
|
def test_warn_qty_delivery_vs_load_imbalance():
|
|
plan = [{"quantity": Decimal("30")}]
|
|
load = [{"plan_qty": Decimal("30")}]
|
|
delivery = [{"plan_qty": Decimal("25")}]
|
|
warnings = warn_qty_imbalance(plan, load, delivery)
|
|
assert warnings == [
|
|
"装货合计(30)与送货合计(25)不一致，请核对",
|
|
]
|
|
|
|
|
|
def test_warn_qty_load_delivery_decimal_imbalance():
|
|
plan = [{"quantity": Decimal("30")}]
|
|
load = [{"plan_qty": Decimal("30")}]
|
|
delivery = [{"plan_qty": Decimal("30.31")}]
|
|
warnings = warn_qty_imbalance(plan, load, delivery)
|
|
assert warnings == [
|
|
"装货合计(30)与送货合计(30.31)不一致，请核对",
|
|
]
|
|
|
|
|
|
def test_warn_qty_multi_stop_30_equals_10_15_5_balanced():
|
|
"""验收场景：计划 30 = 送货 10+15+5，装货 30。"""
|
|
plan = [{"quantity": Decimal("30")}]
|
|
load = [{"plan_qty": Decimal("30")}]
|
|
delivery = [
|
|
{"plan_qty": Decimal("10")},
|
|
{"plan_qty": Decimal("15")},
|
|
{"plan_qty": Decimal("5")},
|
|
]
|
|
assert warn_qty_imbalance(plan, load, delivery) == []
|
|
|
|
|
|
def test_warn_qty_sums_multiple_rows_balanced_across_stops():
|
|
plan = [{"quantity": Decimal("10")}, {"quantity": Decimal("20")}]
|
|
load = [{"plan_qty": Decimal("10")}, {"plan_qty": Decimal("15")}]
|
|
delivery = [{"plan_qty": Decimal("25")}]
|
|
assert warn_qty_imbalance(plan, load, delivery) == []
|
|
|
|
|
|
def test_warn_qty_sums_multiple_rows_imbalanced_across_stops():
|
|
plan = [{"quantity": Decimal("10")}, {"quantity": Decimal("20")}]
|
|
load = [{"plan_qty": Decimal("10")}, {"plan_qty": Decimal("15")}]
|
|
delivery = [{"plan_qty": Decimal("24")}]
|
|
warnings = warn_qty_imbalance(plan, load, delivery)
|
|
assert warnings == [
|
|
"装货合计(25)与送货合计(24)不一致，请核对",
|
|
]
|