mirror of
https://github.com/Hestia-Homes/Model.git
synced 2026-06-30 13:10:47 +00:00
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:
parent
a6f2ae99df
commit
ad3b1f15a8
1 changed files with 26 additions and 0 deletions
|
|
@ -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"
|
||||
|
|
|
|||
Loading…
Add table
Reference in a new issue