fixed imports on unit tests for parity onboarding

This commit is contained in:
Khalim Conn-Kowlessar 2026-02-07 21:29:07 +00:00
parent 7bb7972549
commit c34ed0bbe0
3 changed files with 22 additions and 27 deletions

View file

@ -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

View file

@ -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
# ---------------------------------------------------------------------

View file

@ -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,