24 lines
834 B
Python
24 lines
834 B
Python
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},
|
|
]
|