fixed imports on unit tests for parity onboarding

backend/onboarders/tests/test_floor_remapping.py

@@ -1,14 +1,13 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
-    EpcConstructionAgeBand,
+from datatypes.epc.floor import EpcFloorDescriptions
-    EpcFloorDescriptions,
+
-)
+from backend.onboarders.mappings.parity.as_built_floor_classifiers import (
-from backend.onboarders.mappings.as_built_floor_classifiers import (
     unknown_floor_as_built,
     unknown_floor_retrofitted,
-    solid_floor_as_built,
+    map_solid_floor_as_built,
-    suspended_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

backend/onboarders/tests/test_roof_remapping.py

@@ -1,16 +1,14 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
-    EpcConstructionAgeBand,
+from datatypes.epc.roof import EpcRoofDescriptions
-    EpcRoofDescriptions,
+from datatypes.epc.efficiency import EpcEfficiency
-    EpcEfficiency,
-    resolve_roof_efficiency,
-)
 
-from backend.onboarders.mappings.as_built_roof_classifiers import (
+from backend.onboarders.mappings.parity.as_built_roof_classifiers import (
-    classify_flat_roof,
+    map_flat_roof,
-    classify_sloping_ceiling_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
 
 
 # ---------------------------------------------------------------------

backend/onboarders/tests/test_wall_remapping.py

@@ -1,13 +1,11 @@
 import pytest
 
-from backend.onboarders.epc_descriptions import (
+from datatypes.epc.construction_age_band import EpcConstructionAgeBand
-    EpcConstructionAgeBand,
+from datatypes.epc.walls import EpcWallDescriptions
-    EpcWallDescriptions,
+from datatypes.epc.efficiency import EpcEfficiency
-    EpcEfficiency,
-    resolve_wall_efficiency,
-)
 
-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,