vibes

finder/test_transform.py

@@ -0,0 +1,48 @@
+from transform import (
+    clean_listing_address,
+    extract_full_postcode,
+    transform_property,
+)
+
+
+class StubPostcodeIndex:
+    def nearest(self, lat: float, lng: float) -> str:
+        return "SW1A 9ZZ"
+
+
+def test_extract_full_postcode_normalizes_spacing() -> None:
+    assert extract_full_postcode("10 Downing Street SW1A2AA") == "SW1A 2AA"
+    assert extract_full_postcode("10 Downing Street, SW1A 2AA") == "SW1A 2AA"
+    assert extract_full_postcode("Downing Street, Westminster") is None
+
+
+def test_clean_listing_address_removes_postcode_and_outcode_suffixes() -> None:
+    assert clean_listing_address("10 Downing Street, SW1A 2AA") == "10 Downing Street"
+    assert clean_listing_address("Hawthorne Road, Bromley, Kent, BR1") == (
+        "Hawthorne Road, Bromley, Kent"
+    )
+    assert clean_listing_address("Kings Avenue, Bromley") == "Kings Avenue, Bromley"
+
+
+def test_rightmove_transform_prefers_postcode_from_display_address() -> None:
+    prop = {
+        "id": "123",
+        "location": {"latitude": 51.5, "longitude": -0.1},
+        "price": {"amount": 750000, "displayPrices": []},
+        "propertySubType": "Terraced",
+        "bedrooms": 3,
+        "bathrooms": 1,
+        "keyFeatures": [],
+        "propertyUrl": "/properties/123",
+        "displayAddress": "Flat 2, 10 Downing Street, SW1A 2AA",
+    }
+
+    result = transform_property(prop, "SW1A", StubPostcodeIndex())
+
+    assert result is not None
+    assert result["Postcode"] == "SW1A 2AA"
+    assert result["Postcode source"] == "address"
+    assert result["Extracted postcode"] == "SW1A 2AA"
+    assert result["Inferred postcode"] == "SW1A 9ZZ"
+    assert result["Listing raw address"] == "Flat 2, 10 Downing Street, SW1A 2AA"
+    assert result["Address per Property Register"] == "Flat 2, 10 Downing Street"