From c34ed0bbe07eb7cb39a9ddf85949291c19ef9bd7 Mon Sep 17 00:00:00 2001
From: Khalim Conn-Kowlessar <kconnkowlessar@gmail.com>
Date: Sat, 7 Feb 2026 21:29:07 +0000
Subject: [PATCH] fixed imports on unit tests for parity onboarding

---
 .../onboarders/tests/test_floor_remapping.py  | 17 ++++++++--------
 .../onboarders/tests/test_roof_remapping.py   | 20 +++++++++----------
 .../onboarders/tests/test_wall_remapping.py   | 12 +++++------
 3 files changed, 22 insertions(+), 27 deletions(-)

diff --git a/backend/onboarders/tests/test_floor_remapping.py b/backend/onboarders/tests/test_floor_remapping.py
index 26a05d22..c20372b7 100644
--- a/backend/onboarders/tests/test_floor_remapping.py
+++ b/backend/onboarders/tests/test_floor_remapping.py
@@ -1,14 +1,13 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
-    EpcConstructionAgeBand,
-    EpcFloorDescriptions,
-)
-from backend.onboarders.mappings.as_built_floor_classifiers import (
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
+from datatypes.epc.floor import EpcFloorDescriptions
+
+from backend.onboarders.mappings.parity.as_built_floor_classifiers import (
     unknown_floor_as_built,
     unknown_floor_retrofitted,
-    solid_floor_as_built,
-    suspended_floor_as_built,
+    map_solid_floor_as_built,
+    map_suspended_floor_as_built,
 )
 
 
@@ -75,7 +74,7 @@ def test_unknown_floor_retrofitted(age_band, expected):
     ],
 )
 def test_solid_floor_as_built(age_band, expected):
-    assert solid_floor_as_built(age_band) == expected
+    assert map_solid_floor_as_built(age_band) == expected
 
 
 @pytest.mark.parametrize(
@@ -95,4 +94,4 @@ def test_solid_floor_as_built(age_band, expected):
     ],
 )
 def test_suspended_floor_as_built(age_band, expected):
-    assert suspended_floor_as_built(age_band) == expected
+    assert map_suspended_floor_as_built(age_band) == expected
diff --git a/backend/onboarders/tests/test_roof_remapping.py b/backend/onboarders/tests/test_roof_remapping.py
index a08471f9..cc19e057 100644
--- a/backend/onboarders/tests/test_roof_remapping.py
+++ b/backend/onboarders/tests/test_roof_remapping.py
@@ -1,16 +1,14 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
-    EpcConstructionAgeBand,
-    EpcRoofDescriptions,
-    EpcEfficiency,
-    resolve_roof_efficiency,
-)
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
+from datatypes.epc.roof import EpcRoofDescriptions
+from datatypes.epc.efficiency import EpcEfficiency
 
-from backend.onboarders.mappings.as_built_roof_classifiers import (
-    classify_flat_roof,
-    classify_sloping_ceiling_roof,
+from backend.onboarders.mappings.parity.as_built_roof_classifiers import (
+    map_flat_roof,
+    map_sloping_ceiling_roof,
 )
+from backend.onboarders.mappings.parity.roof import resolve_roof_efficiency
 
 
 # ---------------------------------------------------------------------
@@ -30,7 +28,7 @@ from backend.onboarders.mappings.as_built_roof_classifiers import (
     ],
 )
 def test_classify_flat_roof(age_band, expected):
-    assert classify_flat_roof(age_band) == expected
+    assert map_flat_roof(age_band) == expected
 
 
 @pytest.mark.parametrize(
@@ -45,7 +43,7 @@ def test_classify_flat_roof(age_band, expected):
     ],
 )
 def test_classify_sloping_ceiling_roof(age_band, expected):
-    assert classify_sloping_ceiling_roof(age_band) == expected
+    assert map_sloping_ceiling_roof(age_band) == expected
 
 
 # ---------------------------------------------------------------------
diff --git a/backend/onboarders/tests/test_wall_remapping.py b/backend/onboarders/tests/test_wall_remapping.py
index eaac5afb..c9476211 100644
--- a/backend/onboarders/tests/test_wall_remapping.py
+++ b/backend/onboarders/tests/test_wall_remapping.py
@@ -1,13 +1,11 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
-    EpcConstructionAgeBand,
-    EpcWallDescriptions,
-    EpcEfficiency,
-    resolve_wall_efficiency,
-)
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
+from datatypes.epc.walls import EpcWallDescriptions
+from datatypes.epc.efficiency import EpcEfficiency
 
-from backend.onboarders.mappings.as_built_wall_classifiers import (
+from backend.onboarders.mappings.parity.walls import resolve_wall_efficiency
+from backend.onboarders.mappings.parity.as_built_wall_classifiers import (
     map_cavity_wall_insulation,
     map_solid_wall_insulation,
     map_timber_frame_wall_insulation,