From fc591c65502c73a4464a0d35bc002b3da09245a2 Mon Sep 17 00:00:00 2001 From: Jun-te Kim Date: Fri, 19 Jun 2026 13:43:47 +0000 Subject: [PATCH] =?UTF-8?q?Classify=20the=20landlord=20Age=20column=20into?= =?UTF-8?q?=20a=20construction-age-band=20category=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 | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/applications/landlord_description_overrides/test_build_columns.py b/tests/applications/landlord_description_overrides/test_build_columns.py index c6748728..8f50afce 100644 --- a/tests/applications/landlord_description_overrides/test_build_columns.py +++ b/tests/applications/landlord_description_overrides/test_build_columns.py @@ -38,3 +38,16 @@ def test_build_columns_wires_a_glazing_classifier_column() -> None: assert len(columns) == 1 assert columns[0].name == "glazing" assert columns[0].source_column == "Glazing" + + +def test_build_columns_wires_a_construction_age_band_classifier_column() -> None: + # Arrange + chat_gpt = cast(ChatGPT, object()) + + # Act + columns = _build_columns({"construction_age_band": "Age"}, chat_gpt, None) + + # Assert — one column, named construction_age_band, reading the "Age" header. + assert len(columns) == 1 + assert columns[0].name == "construction_age_band" + assert columns[0].source_column == "Age"