OrgTJWater/pipeline-lifetime
2
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Clarify remaining life export column

Browse Source
This commit is contained in:
jiang
2026-07-20 15:27:21 +08:00
parent 72b6b63154
commit 1223227ea9
2 changed files with 8 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/test_prediction.py
+5 -4
View File
@@ -15,6 +15,7 @@ from app.prediction import (
ID_COLUMN,
INPUT_COLUMNS,
PIPE_AGE_COLUMN,
REMAINING_LIFE_COLUMN,
PredictionError,
estimate_remaining_life,
grade_info,
@@ -182,8 +183,8 @@ class PredictionHelpersTest(unittest.TestCase):
]
summary_rows = [{"pipe_id": "P001"}, {"pipe_id": "P002"}]
summary_sheet_rows = [
{ID_COLUMN: "P001", PIPE_AGE_COLUMN: "10 年", "当前健康状态": 0.7, "健康等级": "IV级", "预计剩余寿命（年）": ">63"},
{ID_COLUMN: "P002", PIPE_AGE_COLUMN: "12 年", "当前健康状态": 0.6, "健康等级": "III级", "预计剩余寿命（年）": ">61"},
{ID_COLUMN: "P001", PIPE_AGE_COLUMN: "10 年", "当前健康状态": 0.7, "健康等级": "IV级", REMAINING_LIFE_COLUMN: ">63"},
{ID_COLUMN: "P002", PIPE_AGE_COLUMN: "12 年", "当前健康状态": 0.6, "健康等级": "III级", REMAINING_LIFE_COLUMN: ">61"},
]
with TemporaryDirectory() as temp_dir:
@@ -196,7 +197,7 @@ class PredictionHelpersTest(unittest.TestCase):
summary_data = pd.read_excel(output_path, sheet_name="结果摘要")
self.assertEqual(
summary_data.columns.tolist(),
[ID_COLUMN, PIPE_AGE_COLUMN, "当前健康状态", "健康等级", "预计剩余寿命（年）"],
[ID_COLUMN, PIPE_AGE_COLUMN, "当前健康状态", "健康等级", REMAINING_LIFE_COLUMN],
)
sample_data = pd.read_excel(output_path, sheet_name="样本数据")
@@ -216,7 +217,7 @@ class PredictionHelpersTest(unittest.TestCase):
predictions = [dummy_prediction("00123", [1], [0.9], 10, 0.9)]
summary_rows = [{"pipe_id": "00123"}]
summary_sheet_rows = [
{ID_COLUMN: "00123", PIPE_AGE_COLUMN: "10 年", "当前健康状态": 0.9, "健康等级": "V级", "预计剩余寿命（年）": ">63"}
{ID_COLUMN: "00123", PIPE_AGE_COLUMN: "10 年", "当前健康状态": 0.9, "健康等级": "V级", REMAINING_LIFE_COLUMN: ">63"}
]
with TemporaryDirectory() as temp_dir: