Central utility crosswalk, tariff mapper fixes, and NY customer counts (#155)

* Add fetch-ny-utility-stats Justfile task, EIA-861 utils script, and ny_electric_utility_stats.csv

- Move NY data Justfile up to rate_design/ny/hp_rates/data/
- Add fetch-ny-utility-stats recipe (uses utils/get_utility_stats_from_eia861.py)
- Add utils/get_utility_stats_from_eia861.py from main
- Force-add ny_electric_utility_stats.csv (EIA-861 NY investor-owned utility stats)

Co-authored-by: Cursor <cursoragent@cursor.com>

* feat(utils): add central utility crosswalk and std_name types

- Add utils/utility_codes.py with UTILITIES list (one record per utility)
- Derive lookup functions: get_ny_open_data_to_std_name,
  get_eia_utility_id_to_std_name, get_std_name_to_gas_tariff_key, etc.
- Update utils/types.py: electric_utility and gas_utility Literals
  use std_name (coned, nyseg, nimo, keddy, kedli, etc.)

Co-authored-by: Cursor <cursoragent@cursor.com>

* test(utils): add tests for utility_codes crosswalk

Co-authored-by: Cursor <cursoragent@cursor.com>

* refactor(assign_utility_ny): use central crosswalk for NY Open Data names

- Replace inline utility_name_map with get_ny_open_data_to_std_name()
- Update tests to use _utility_name_map_lazy from utility_codes

Co-authored-by: Cursor <cursoragent@cursor.com>

* feat(get_utility_stats): add utility_code column (EIA ID → std_name)

Co-authored-by: Cursor <cursoragent@cursor.com>

* fix(tariff_mappers): use real assign_utility data and central crosswalk

- electric_tariff_mapper: use sb.electric_utility when present,
  support metadata_utility path, accept std_name
- gas_tariff_mapper: use get_std_name_to_gas_tariff_key for
  National Grid split (nimo/keddy/kedli), same metadata handling
- Add tests for both mappers

Co-authored-by: Cursor <cursoragent@cursor.com>

* chore(ny,ri): update Justfiles to std_name, regenerate NY utility stats

- Use coned, nimo, nyseg in map-electric-tariff and map-gas-tariff
- Add map-electric-rie-default / map-gas-rie-default for RI
- Regenerate ny_electric_utility_stats.csv with utility_code column

Co-authored-by: Cursor <cursoragent@cursor.com>

* docs(utility_codes): add guide for adding a new state

Co-authored-by: Cursor <cursoragent@cursor.com>

* Make ruff and ty happy

---------

Co-authored-by: Cursor <cursoragent@cursor.com>

Author: jpvelez

…ign/ny/hp_rates/data/tariff_map/Justfile rate_design/ny/hp_rates/data/Justfilerate_design/ny/hp_rates/data/tariff_map/Justfile renamed to rate_design/ny/hp_rates/data/Justfile rate_design/ny/hp_rates/data/tariff_map/Justfile renamed to rate_design/ny/hp_rates/data/Justfile

@@ -2,9 +2,16 @@
 project_root := `git rev-parse --show-toplevel`
 path_s3_resstock_metadata := "s3://data.sb/nrel/resstock/res_2024_amy2018_2/metadata"
 tariff_map_dir_ny := "{{project_root}}/rate_design/ny/hp_rates/data/tariff_map"
+data_dir_ny := project_root + "/rate_design/ny/hp_rates/data"
+
+# Fetch NY investor-owned utility customer counts from EIA-861, write to ny_electric_utility_stats.csv
+fetch-ny-utility-stats:
+  uv run python {{project_root}}/utils/get_utility_stats_from_eia861.py NY > {{data_dir_ny}}/ny_electric_utility_stats.csv
+  @echo "Wrote {{data_dir_ny}}/ny_electric_utility_stats.csv"
 
 # Usage: just map-electric-tariff [electric_utility] [SB_scenario_type] [SB_scenario_year] [upgrade_id] [output_dir]
-# Example: just map-electric-tariff Coned default 1 00 ./rate_design/ny/hp_rates/data/tariff_map
+# Example: just map-electric-tariff coned default 1 00 ./rate_design/ny/hp_rates/data/tariff_map
+# electric_utility: std_name (e.g. coned, nyseg, nimo)
 map-electric-tariff electric_utility SB_scenario_type SB_scenario_year upgrade_id output_dir:
   uv run python {{project_root}}/utils/electric_tariff_mapper.py \
     --metadata_path "{{path_s3_resstock_metadata}}" \
@@ -18,34 +25,34 @@ map-electric-tariff electric_utility SB_scenario_type SB_scenario_year upgrade_i
 # Some useful enumerations:
 
 map-electric-coned-default:
-  just map-electric-tariff Coned default 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff coned default 1 00 {{tariff_map_dir_ny}}
 
 map-electric-coned-seasonal:
-  just map-electric-tariff Coned seasonal 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff coned seasonal 1 00 {{tariff_map_dir_ny}}
 
-map-celectric-oned-class-specific-seasonal:
-  just map-electric-tariff Coned class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
+map-electric-coned-class-specific-seasonal:
+  just map-electric-tariff coned class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
 
 map-electric-national-grid-default:
-  just map-electric-tariff "National Grid" default 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nimo default 1 00 {{tariff_map_dir_ny}}
 
 map-electric-national-grid-seasonal:
-  just map-electric-tariff "National Grid" seasonal 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nimo seasonal 1 00 {{tariff_map_dir_ny}}
 
 map-electric-national-grid-class-specific-seasonal:
-  just map-electric-tariff "National Grid" class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nimo class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
 
 map-electric-nyseg-default:
-  just map-electric-tariff NYSEG default 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nyseg default 1 00 {{tariff_map_dir_ny}}
 
 map-electric-nyseg-seasonal:
-  just map-electric-tariff NYSEG seasonal 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nyseg seasonal 1 00 {{tariff_map_dir_ny}}
 
 map-electric-nyseg-class-specific-seasonal:
-  just map-electric-tariff NYSEG class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
+  just map-electric-tariff nyseg class_specific_seasonal 1 00 {{tariff_map_dir_ny}}
 
 # Usage: just map-gas-tariff [electric_utility] [upgrade_id] [output_dir]
-# Example: just map-gas-tariff NYSEG 00 ./rate_design/ny/hp_rates/data/tariff_map
+# Example: just map-gas-tariff nyseg 00 ./rate_design/ny/hp_rates/data/tariff_map
 map-gas-tariff electric_utility upgrade_id output_dir:
   uv run python {{project_root}}/utils/gas_tariff_mapper.py \
     --metadata_path "{{path_s3_resstock_metadata}}" \
@@ -57,28 +64,28 @@ map-gas-tariff electric_utility upgrade_id output_dir:
 # Some useful enumerations:
 
 map-gas-coned-default:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ny}}
 
 map-gas-coned-seasonal:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ny}}
 
 map-gas-coned-class-specific-seasonal:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ny}}
 
 map-gas-national-grid-default:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ny}}
 
 map-gas-national-grid-seasonal:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ny}}
 
 map-gas-national-grid-class-specific-seasonal:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ny}}
 
 map-gas-nyseg-default:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ny}}
 
 map-gas-nyseg-seasonal:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ny}}
 
 map-gas-nyseg-class-specific-seasonal:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ny}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ny}}

rate_design/ny/hp_rates/data/ny_electric_utility_stats.csv

@@ -0,0 +1,10 @@
+utility_id_eia,utility_code,utility_name,business_model,entity_type,report_date,total_sales_mwh,total_sales_revenue,commercial_sales_mwh,commercial_sales_revenue,commercial_customers,industrial_sales_mwh,industrial_sales_revenue,industrial_customers,other_sales_mwh,other_sales_revenue,other_customers,residential_sales_mwh,residential_sales_revenue,residential_customers,transportation_sales_mwh,transportation_sales_revenue,transportation_customers
+4226,coned,Consolidated Edison Co-NY Inc,retail,Investor Owned,2024-01-01,52425972.0,10723693000.0,35041850.0,5565803000.0,572249.0,1303892.0,338525000.0,10498.0,0.0,0.0,0.0,13643968.0,4671811000.0,3064038.0,2436261.0,147553000.0,2.0
+13573,nimo,Niagara Mohawk Power Corp.,retail,Investor Owned,2024-01-01,32718092.0,2986044400.0,12576316.0,792737300.0,185072.0,8302774.0,220623000.0,1554.0,0.0,0.0,0.0,11833895.0,1972635400.0,1532294.0,5108.0,48600.0,1.0
+13511,nyseg,New York State Elec & Gas Corp,retail,Investor Owned,2024-01-01,15231769.0,1764984600.0,5806928.0,472634400.0,127567.0,2472844.0,62332000.0,1500.0,0.0,0.0,0.0,6919839.0,1228409300.0,792300.0,32158.0,1609000.0,2.0
+16183,rge,Rochester Gas & Electric Corp,retail,Investor Owned,2024-01-01,6871716.0,812878850.0,3044406.0,259093600.0,40227.0,984961.0,49085700.0,634.0,0.0,0.0,0.0,2842349.0,504699520.0,351481.0,0.0,0.0,0.0
+3249,cenhud,Central Hudson Gas & Elec Corp,retail,Investor Owned,2024-01-01,5018039.0,787803970.0,2009494.0,239328000.0,49472.0,840758.0,31443000.0,861.0,0.0,0.0,0.0,2167787.0,517033020.0,270859.0,0.0,0.0,0.0
+14154,or,Orange & Rockland Utils Inc,retail,Investor Owned,2024-01-01,4087784.0,594447900.0,2023377.0,203018600.0,33695.0,313637.0,16500600.0,85.0,0.0,0.0,0.0,1750770.0,374928700.0,211011.0,0.0,0.0,0.0
+66101,,FirstEnergy Pennsylvania Electric Co,retail,Investor Owned,2024-01-01,64611.0,8235400.0,30033.0,3107500.0,479.0,4022.0,530100.0,1.0,0.0,0.0,0.0,30556.0,4597800.0,3400.0,0.0,0.0,0.0
+14711,,Pennsylvania Electric Co,retail,Investor Owned,2023-01-01,60641.0,7912700.0,28931.0,3099000.0,486.0,1214.0,177500.0,1.0,0.0,0.0,0.0,30496.0,4636200.0,3400.0,0.0,0.0,0.0
+6369,,Fishers Island Utility Co Inc,retail,Investor Owned,2024-01-01,7079.0,2372900.0,2061.0,457600.0,113.0,0.0,0.0,0.0,0.0,0.0,0.0,5018.0,1915300.0,657.0,0.0,0.0,0.0

rate_design/ri/hp_rates/data/tariff_map/Justfile

@@ -4,7 +4,7 @@ path_s3_resstock_metadata := "s3://data.sb/nrel/resstock/res_2024_amy2018_2/meta
 tariff_map_dir_ri := "{{project_root}}/rate_design/ri/hp_rates/data/tariff_map"
 
 # Usage: just map-electric-tariff [electric_utility] [SB_scenario_type] [SB_scenario_year] [upgrade_id] [output_dir]
-# Example: just map-electric-tariff Coned default 1 00 ./rate_design/ri/hp_rates/data/tariff_map
+# Example: just map-electric-tariff rie default 1 00 ./rate_design/ri/hp_rates/data/tariff_map
 map-electric-tariff electric_utility SB_scenario_type SB_scenario_year upgrade_id output_dir:
   uv run python {{project_root}}/utils/electric_tariff_mapper.py \
     --metadata_path "{{path_s3_resstock_metadata}}" \
@@ -18,35 +18,38 @@ map-electric-tariff electric_utility SB_scenario_type SB_scenario_year upgrade_i
 # Some useful enumerations:
 
 map-electric-coned-default:
-  just map-electric-tariff Coned default 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff coned default 1 00 {{tariff_map_dir_ri}}
 
 map-electric-coned-seasonal:
-  just map-electric-tariff Coned seasonal 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff coned seasonal 1 00 {{tariff_map_dir_ri}}
 
-map-celectric-oned-class-specific-seasonal:
-  just map-electric-tariff Coned class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
+map-electric-coned-class-specific-seasonal:
+  just map-electric-tariff coned class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
 
 map-electric-national-grid-default:
-  just map-electric-tariff "National Grid" default 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nimo default 1 00 {{tariff_map_dir_ri}}
 
 map-electric-national-grid-seasonal:
-  just map-electric-tariff "National Grid" seasonal 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nimo seasonal 1 00 {{tariff_map_dir_ri}}
 
 map-electric-national-grid-class-specific-seasonal:
-  just map-electric-tariff "National Grid" class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nimo class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
 
 map-electric-nyseg-default:
-  just map-electric-tariff NYSEG default 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nyseg default 1 00 {{tariff_map_dir_ri}}
 
 map-electric-nyseg-seasonal:
-  just map-electric-tariff NYSEG seasonal 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nyseg seasonal 1 00 {{tariff_map_dir_ri}}
 
 map-electric-nyseg-class-specific-seasonal:
-  just map-electric-tariff NYSEG class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
+  just map-electric-tariff nyseg class_specific_seasonal 1 00 {{tariff_map_dir_ri}}
+
+map-electric-rie-default:
+  just map-electric-tariff rie default 1 00 {{tariff_map_dir_ri}}
 
 
 # Usage: just map-gas-tariff [electric_utility] [upgrade_id] [output_dir]
-# Example: just map-gas-tariff "National Grid" 00 ./rate_design/ri/hp_rates/data/tariff_map
+# Example: just map-gas-tariff rie 00 ./rate_design/ri/hp_rates/data/tariff_map
 map-gas-tariff electric_utility upgrade_id output_dir:
   uv run python {{project_root}}/utils/gas_tariff_mapper.py \
     --metadata_path "{{path_s3_resstock_metadata}}" \
@@ -58,28 +61,31 @@ map-gas-tariff electric_utility upgrade_id output_dir:
 # Some useful enumerations:
 
 map-gas-coned-default:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ri}}
 
 map-gas-coned-seasonal:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ri}}
 
 map-gas-coned-class-specific-seasonal:
-  just map-gas-tariff Coned 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff coned 00 {{tariff_map_dir_ri}}
 
 map-gas-national-grid-default:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ri}}
 
 map-gas-national-grid-seasonal:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ri}}
 
 map-gas-national-grid-class-specific-seasonal:
-  just map-gas-tariff "National Grid" 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nimo 00 {{tariff_map_dir_ri}}
 
 map-gas-nyseg-default:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ri}}
 
 map-gas-nyseg-seasonal:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ri}}
 
 map-gas-nyseg-class-specific-seasonal:
-  just map-gas-tariff NYSEG 00 {{tariff_map_dir_ri}}
+  just map-gas-tariff nyseg 00 {{tariff_map_dir_ri}}
+
+map-gas-rie-default:
+  just map-gas-tariff rie 00 {{tariff_map_dir_ri}}

tests/test_assign_utility_ny.py

@@ -7,11 +7,21 @@
 import polars as pl
 
 from utils.assign_utility_ny import (
-    CONFIGS,
     _calculate_prior_distributions,
     _calculate_utility_probabilities,
     _sample_utility_per_building,
 )
+from utils.utility_codes import get_ny_open_data_to_std_name
+
+
+def _utility_name_map_lazy() -> pl.LazyFrame:
+    """Build utility_name_map from central crosswalk (matches assign_utility_ny)."""
+    return pl.DataFrame(
+        [
+            {"state_name": k, "std_name": v}
+            for k, v in get_ny_open_data_to_std_name().items()
+        ]
+    ).lazy()
 
 
 def test_calculate_utility_probabilities_basic():
@@ -29,7 +39,7 @@ def test_calculate_utility_probabilities_basic():
             "pct_overlap": [60.0, 30.0, 10.0, 50.0, 50.0],
         }
     )
-    utility_name_map = pl.LazyFrame(CONFIGS["utility_name_map"])
+    utility_name_map = _utility_name_map_lazy()
 
     result = _calculate_utility_probabilities(
         puma_overlap,
@@ -60,7 +70,7 @@ def test_calculate_utility_probabilities_filter_none_false():
             "pct_overlap": [70.0, 30.0],
         }
     )
-    utility_name_map = pl.LazyFrame(CONFIGS["utility_name_map"])
+    utility_name_map = _utility_name_map_lazy()
 
     result = _calculate_utility_probabilities(
         puma_overlap,
@@ -84,7 +94,7 @@ def test_calculate_utility_probabilities_handle_municipal():
             "pct_overlap": [40.0, 60.0],
         }
     )
-    utility_name_map = pl.LazyFrame(CONFIGS["utility_name_map"])
+    utility_name_map = _utility_name_map_lazy()
 
     result = _calculate_utility_probabilities(
         puma_overlap,

tests/test_electric_tariff_mapper.py

@@ -0,0 +1,50 @@
+"""Tests for electric tariff mapper."""
+
+from typing import cast
+
+import polars as pl
+
+from utils.electric_tariff_mapper import map_electric_tariff
+from utils.types import SBScenario
+
+
+def test_map_electric_tariff_filters_by_std_name():
+    """map_electric_tariff filters metadata by sb.electric_utility (std_name)."""
+    metadata = pl.LazyFrame(
+        {
+            "bldg_id": [1, 2, 3, 4],
+            "sb.electric_utility": ["coned", "coned", "nyseg", "nimo"],
+            "postprocess_group.has_hp": [True, False, True, False],
+        }
+    )
+    sb_scenario = SBScenario("default", 1)
+    result = map_electric_tariff(
+        SB_metadata_df=metadata,
+        electric_utility="coned",
+        SB_scenario=sb_scenario,
+        state="NY",
+    )
+    df = cast(pl.DataFrame, result.collect())
+    assert df.height == 2
+    assert set(df["bldg_id"].to_list()) == {1, 2}
+    assert all(df["tariff_key"].str.starts_with("coned_"))
+
+
+def test_map_electric_tariff_tariff_key_format():
+    """tariff_key uses std_name and scenario."""
+    metadata = pl.LazyFrame(
+        {
+            "bldg_id": [1],
+            "sb.electric_utility": ["nyseg"],
+            "postprocess_group.has_hp": [True],
+        }
+    )
+    sb_scenario = SBScenario("seasonal", 1)
+    result = map_electric_tariff(
+        SB_metadata_df=metadata,
+        electric_utility="nyseg",
+        SB_scenario=sb_scenario,
+        state="NY",
+    )
+    df = cast(pl.DataFrame, result.collect())
+    assert df["tariff_key"][0] == "nyseg_seasonal_1_HP.csv"

tests/test_gas_tariff_mapper.py

@@ -0,0 +1,29 @@
+"""Tests for gas tariff mapper."""
+
+from typing import cast
+
+import polars as pl
+
+from utils.gas_tariff_mapper import map_gas_tariff
+
+
+def test_map_gas_tariff_uses_crosswalk_for_tariff_key():
+    """map_gas_tariff maps sb.gas_utility (std_name) to tariff_key via crosswalk."""
+    metadata = pl.LazyFrame(
+        {
+            "bldg_id": [1, 2, 3],
+            "sb.electric_utility": ["coned", "coned", "coned"],
+            "sb.gas_utility": ["nimo", "nyseg", "coned"],
+        }
+    )
+    result = map_gas_tariff(
+        SB_metadata_df=metadata,
+        electric_utility_name="coned",
+    )
+    df = cast(pl.DataFrame, result.collect())
+    assert df.height == 3
+    # nimo -> national_grid, nyseg -> nyseg, coned -> coned
+    tariff_keys = df["tariff_key"].to_list()
+    assert "national_grid" in tariff_keys
+    assert "nyseg" in tariff_keys
+    assert "coned" in tariff_keys

tests/test_get_utility_stats_from_eia861.py

@@ -4,6 +4,8 @@
 dynamic aggregation is safe. Uses the same parquet URL as the script.
 """
 
+import subprocess
+
 import polars as pl
 
 # Must match utils/get_utility_stats_from_eia861.py
@@ -71,3 +73,38 @@ def test_eia861_yearly_sales_has_investor_owned_data_for_sample_state():
     assert frozenset(classes_in_data) == EXPECTED_CUSTOMER_CLASSES, (
         f"NY IOU data missing some customer classes: got {set(classes_in_data)}"
     )
+
+
+def test_utility_code_column_present_and_mapped():
+    """Script output includes utility_code; known EIA IDs map to expected std_names."""
+    from pathlib import Path
+
+    project_root = Path(__file__).resolve().parent.parent
+    result = subprocess.run(
+        ["uv", "run", "python", "utils/get_utility_stats_from_eia861.py", "NY"],
+        capture_output=True,
+        text=True,
+        cwd=project_root,
+    )
+    assert result.returncode == 0, f"Script failed: {result.stderr}"
+
+    from io import StringIO
+
+    df = pl.read_csv(StringIO(result.stdout))
+    assert "utility_code" in df.columns
+
+    # Known EIA ID -> std_name mappings
+    expected = {
+        4226: "coned",
+        13573: "nimo",
+        13511: "nyseg",
+        16183: "rge",
+        3249: "cenhud",
+        14154: "or",
+    }
+    for eia_id, std_name in expected.items():
+        row = df.filter(pl.col("utility_id_eia") == eia_id)
+        assert not row.is_empty(), f"EIA ID {eia_id} not in output"
+        assert row["utility_code"][0] == std_name, (
+            f"EIA ID {eia_id}: expected utility_code {std_name}, got {row['utility_code'][0]}"
+        )