From ad3b1f15a80f3192fc95f87242ed99a7afa58de9 Mon Sep 17 00:00:00 2001 From: Jun-te Kim Date: Fri, 19 Jun 2026 14:07:14 +0000 Subject: [PATCH] =?UTF-8?q?Classify=20the=20landlord=20Hot=20Water=20and?= =?UTF-8?q?=20Heating=20columns=20into=20categories=20=F0=9F=9F=A5?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit Co-Authored-By: Claude Opus 4.8 (1M context) --- .../test_build_columns.py | 26 +++++++++++++++++++ 1 file changed, 26 insertions(+) diff --git a/tests/applications/landlord_description_overrides/test_build_columns.py b/tests/applications/landlord_description_overrides/test_build_columns.py index 8f50afce..0e47ed8d 100644 --- a/tests/applications/landlord_description_overrides/test_build_columns.py +++ b/tests/applications/landlord_description_overrides/test_build_columns.py @@ -51,3 +51,29 @@ def test_build_columns_wires_a_construction_age_band_classifier_column() -> None assert len(columns) == 1 assert columns[0].name == "construction_age_band" assert columns[0].source_column == "Age" + + +def test_build_columns_wires_a_water_heating_classifier_column() -> None: + # Arrange + chat_gpt = cast(ChatGPT, object()) + + # Act + columns = _build_columns({"water_heating": "Hot Water"}, chat_gpt, None) + + # Assert + assert len(columns) == 1 + assert columns[0].name == "water_heating" + assert columns[0].source_column == "Hot Water" + + +def test_build_columns_wires_a_main_heating_system_classifier_column() -> None: + # Arrange + chat_gpt = cast(ChatGPT, object()) + + # Act + columns = _build_columns({"main_heating_system": "Heating"}, chat_gpt, None) + + # Assert + assert len(columns) == 1 + assert columns[0].name == "main_heating_system" + assert columns[0].source_column == "Heating"