Model/backend/tests/test_data/pre_heating_scenarios.py

# Each scenario: super explicit about inputs and expected mapping
pre_main_heating_scenarios = [
    # --- Mains gas boilers (radiators) ---
    {
        "description": "Boiler and radiators, mains gas (condensing expected)",
        "MAINHEAT_DESCRIPTION": "Boiler and radiators, mains gas",
        "MAIN_FUEL": "mains gas (not community)",
        "MAINHEAT_ENERGY_EFF": "Good",
        "expected": "Condensing Gas Boiler",
    },
    {
        "description": "Boiler and radiators, mains gas (non-condensing expected)",
        "MAINHEAT_DESCRIPTION": "Boiler and radiators, mains gas",
        "MAIN_FUEL": "mains gas - this is for backwards compatibility only and should not be used",
        "MAINHEAT_ENERGY_EFF": "Average",
        "expected": "Non Condensing Gas Boiler",
    },
    {
        "description": "Boiler and radiators, mains gas (very poor => back boiler to rads)",
        "MAINHEAT_DESCRIPTION": "Boiler and radiators, mains gas",
        "MAIN_FUEL": "Gas: mains gas",
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Gas Back Boiler to Radiators",
    },

    # --- Warm air (treated like gas boiler family in your mapper) ---
    {
        "description": "Warm air, mains gas (good => condensing)",
        "MAINHEAT_DESCRIPTION": "Warm air, mains gas",
        "MAIN_FUEL": "mains gas (not community)",
        "MAINHEAT_ENERGY_EFF": "Good",
        "expected": "Condensing Gas Boiler",
    },

    # --- Community scheme (CHP vs non-CHP depends on energy eff) ---
    {
        "description": "Community scheme (gas, good => CHP)",
        "MAINHEAT_DESCRIPTION": "Community scheme",
        "MAIN_FUEL": "mains gas (community)",
        "MAINHEAT_ENERGY_EFF": "Good",
        "expected": "DHS CHP",
    },
    {
        "description": "Community scheme (gas, average => non-CHP)",
        "MAINHEAT_DESCRIPTION": "Community scheme",
        "MAIN_FUEL": "mains gas (community)",
        "MAINHEAT_ENERGY_EFF": "Average",
        "expected": "DHS non-CHP",
    },
    {
        "description": "Community scheme (no fuel data, good => CHP)",
        "MAINHEAT_DESCRIPTION": "Community scheme",
        "MAIN_FUEL": "NO DATA!",
        "MAINHEAT_ENERGY_EFF": "Good",
        "expected": "DHS CHP",
    },

    # --- Electric storage heaters (ESH responsiveness split) ---
    {
        "description": "Electric storage heaters (average => responsiveness > 0.2)",
        "MAINHEAT_DESCRIPTION": "Electric storage heaters",
        "MAIN_FUEL": "electricity (not community)",
        "MAINHEAT_ENERGY_EFF": "Average",
        "expected": "Electric Storage Heaters Responsiveness >0.2",
    },
    {
        "description": "Electric storage heaters (poor => responsiveness > 0.2)",
        "MAINHEAT_DESCRIPTION": "Electric storage heaters",
        "MAIN_FUEL": "electricity - this is for backwards compatibility only and should not be used",
        "MAINHEAT_ENERGY_EFF": "Poor",
        "expected": "Electric Storage Heaters Responsiveness >0.2",
    },
    {
        "description": "Electric storage heaters (very poor => responsiveness <= 0.2)",
        "MAINHEAT_DESCRIPTION": "Electric storage heaters",
        "MAIN_FUEL": "electricity (not community)",
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Electric Storage Heaters Responsiveness <=0.2",
    },

    # --- Electric direct-acting / room heaters ---
    {
        "description": "Room heaters, electric (very poor)",
        "MAINHEAT_DESCRIPTION": "Room heaters, electric",
        "MAIN_FUEL": "electricity (not community)",
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Electric Room Heaters",
    },
    {
        "description": "Room heaters, electric (poor, unspecified tariff)",
        "MAINHEAT_DESCRIPTION": "Room heaters, electric",
        "MAIN_FUEL": "Electricity: electricity, unspecified tariff",
        "MAINHEAT_ENERGY_EFF": "Poor",
        "expected": "Electric Room Heaters",
    },
    {
        "description": "Portable electric heaters assumed for most rooms (maps to electric room heaters)",
        "MAINHEAT_DESCRIPTION": "Portable electric heaters assumed for most rooms",
        "MAIN_FUEL": "mains gas (not community)",  # weird in EPCs, but your mapper forces electric room heaters here
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Electric Room Heaters",
    },
    {
        "description": "No system present: electric heaters assumed",
        "MAINHEAT_DESCRIPTION": "No system present: electric heaters assumed",
        "MAIN_FUEL": "To be used only when there is no heating/hot-water system",
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Electric Room Heaters",
    },
    {
        "description": "Electric underfloor heating => direct-acting electric",
        "MAINHEAT_DESCRIPTION": "Electric underfloor heating",
        "MAIN_FUEL": "electricity (not community)",
        "MAINHEAT_ENERGY_EFF": "Average",
        "expected": "Electric Room Heaters",
    },

    # --- Gas room heaters ---
    {
        "description": "Room heaters, mains gas (average)",
        "MAINHEAT_DESCRIPTION": "Room heaters, mains gas",
        "MAIN_FUEL": "mains gas (not community)",
        "MAINHEAT_ENERGY_EFF": "Average",
        "expected": "Gas Room Heaters",
    },

    # --- Electric boiler ---
    {
        "description": "Boiler and radiators, electric (very poor => electric boiler)",
        "MAINHEAT_DESCRIPTION": "Boiler and radiators, electric",
        "MAIN_FUEL": "electricity (not community)",
        "MAINHEAT_ENERGY_EFF": "Very Poor",
        "expected": "Electric Boiler",
    },

    # --- Gas boiler + UFH (still ‘boiler’ logic) ---
    {
        "description": "Boiler and underfloor heating, mains gas (good => condensing)",
        "MAINHEAT_DESCRIPTION": "Boiler and underfloor heating, mains gas",
        "MAIN_FUEL": "mains gas (not community)",
        "MAINHEAT_ENERGY_EFF": "Good",
        "expected": "Condensing Gas Boiler",
    },
]