import polars as pl

from pipeline.download.rental_prices import _latest_rents_long


def test_latest_rents_long_adds_iod_alias_codes_for_south_yorkshire():
    raw = pl.DataFrame(
        {
            "column_1": ["title", "header", "2026-02-01 00:00:00"],
            "column_2": ["", "", "E08000038"],
            "column_3": ["", "", "Barnsley"],
            "column_12": ["", "", "486"],
            "column_16": ["", "", "595"],
            "column_20": ["", "", "705"],
            "column_24": ["", "", "900"],
        }
    )

    result = _latest_rents_long(raw).filter(pl.col("bedrooms") == 1).sort("area_code")

    assert result.select("area_code", "mean_monthly_rent").to_dicts() == [
        {"area_code": "E08000016", "mean_monthly_rent": 486.0},
        {"area_code": "E08000038", "mean_monthly_rent": 486.0},
    ]