All good

pipeline/transform/test_transform_poi.py

@@ -79,6 +79,39 @@ def test_transform_grocery_retail_points_keeps_fascia_icon_category():
     ]
 
 
+def test_transform_grocery_retail_points_merges_cooperative_societies():
+    raw = pl.DataFrame(
+        {
+            "id": [101, 102, 103],
+            "retailer": [
+                "Central England Co-operative",
+                "Lincolnshire Co-operative",
+                "The Southern Co-operative",
+            ],
+            "fascia": [
+                "Central England Co-operative",
+                "The Co-operative Food",
+                None,
+            ],
+            "store_name": [
+                "Central Co-op Test",
+                "Lincolnshire Co-op Test",
+                "Southern Co-op Test",
+            ],
+            "long_wgs": [-0.141, -0.142, -0.143],
+            "lat_wgs": [51.515, 51.516, 51.517],
+        }
+    )
+
+    pois = transform_grocery_retail_points(raw, min_chain_locations=1)
+
+    assert pois.select("category", "icon_category").to_dicts() == [
+        {"category": "Co-op", "icon_category": "Co-op"},
+        {"category": "Co-op", "icon_category": "Co-op"},
+        {"category": "Co-op", "icon_category": "Co-op"},
+    ]
+
+
 def test_transform_grocery_retail_points_accepts_base_fascias():
     raw = pl.DataFrame(
         {