Classify the landlord Hot Water and Heating columns into categories 🟥

Co-Authored-By: Claude Opus 4.8 (1M context) <noreply@anthropic.com>
This commit is contained in:
Jun-te Kim 2026-06-19 14:07:14 +00:00
parent a6f2ae99df
commit ad3b1f15a8

View file

@ -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"