Refine prediction workbook summary note

This commit is contained in:
2026-07-21 15:20:14 +08:00
parent f599647fbd
commit 8554d2db97
2 changed files with 28 additions and 7 deletions
+7 -4
View File
@@ -212,8 +212,9 @@ class PredictionHelpersTest(unittest.TestCase):
summary_data.columns.tolist(),
[ID_COLUMN, PIPE_AGE_COLUMN, "当前健康状态", "健康等级", REMAINING_LIFE_COLUMN],
)
self.assertEqual(summary_data[PIPE_AGE_COLUMN].tolist(), [10, 12])
self.assertEqual(summary_data[REMAINING_LIFE_COLUMN].tolist(), [63, 61])
summary_records = summary_data[summary_data[PIPE_AGE_COLUMN].notna()]
self.assertEqual(summary_records[PIPE_AGE_COLUMN].tolist(), [10, 12])
self.assertEqual(summary_records[REMAINING_LIFE_COLUMN].tolist(), [63, 61])
sample_data = pd.read_excel(output_path, sheet_name="样本数据")
self.assertEqual(sample_data.columns.tolist(), ["管龄(年)", "P001", "P002"])
@@ -229,8 +230,10 @@ class PredictionHelpersTest(unittest.TestCase):
self.assertEqual(sample_worksheet.auto_filter.ref, "A1:C3")
self.assertEqual(summary_worksheet["C2"].number_format, "0.00")
self.assertEqual(sample_worksheet["B2"].number_format, "0.00")
self.assertIn("73减去当前管龄", summary_worksheet["E1"].value)
self.assertNotIn("73 - 当前管龄", summary_worksheet["E1"].value)
self.assertEqual(summary_worksheet["E1"].value, "剩余寿命(年)")
self.assertIn("A5:E8", [str(range_) for range_ in summary_worksheet.merged_cells.ranges])
self.assertIn("73减去当前管龄", summary_worksheet["A5"].value)
self.assertIn("目前收集到模型数据管龄最大为73年", summary_worksheet["A5"].value)
def test_prediction_workbook_writes_pipe_ids_as_excel_text(self) -> None:
predictions = [dummy_prediction("00123", [1], [0.9], 10, 0.9)]