chore: restructure python

This commit is contained in:
Stijnvandenbroek
2026-03-04 10:36:54 +00:00
parent f6158d561a
commit cf38ac521c
5 changed files with 200 additions and 182 deletions

View File

@@ -1,156 +1,143 @@
"""Tests for pure helper functions in data_platform.assets.funda."""
"""Tests for pure helper functions in data_platform.assets.helpers."""
from data_platform.assets.funda import (
_details_preview_table,
_safe,
_safe_int,
_search_preview_table,
from data_platform.assets.helpers import (
format_area,
format_euro,
md_preview_table,
safe,
safe_int,
)
# ── _safe_int ───────────────────────────────────────────────────────────────
# ── safe_int ───────────────────────────────────────────────────────────────
class TestSafeInt:
def test_none_returns_none(self):
assert _safe_int(None) is None
assert safe_int(None) is None
def test_integer_passthrough(self):
assert _safe_int(42) == 42
assert safe_int(42) == 42
def test_negative_integer(self):
assert _safe_int(-10) == -10
assert safe_int(-10) == -10
def test_zero(self):
assert _safe_int(0) == 0
assert safe_int(0) == 0
def test_string_int(self):
assert _safe_int("123") == 123
assert safe_int("123") == 123
def test_float_truncated(self):
assert _safe_int(3.9) == 3
assert safe_int(3.9) == 3
def test_float_string(self):
assert _safe_int("7.0") == 7
assert safe_int("7.0") == 7
def test_non_numeric_string_returns_none(self):
assert _safe_int("abc") is None
assert safe_int("abc") is None
def test_empty_string_returns_none(self):
assert _safe_int("") is None
assert safe_int("") is None
def test_list_returns_none(self):
assert _safe_int([1, 2, 3]) is None
assert safe_int([1, 2, 3]) is None
# ── _safe ────────────────────────────────────────────────────────────────────
# ── safe ────────────────────────────────────────────────────────────────────
class TestSafe:
def test_dict_becomes_json_string(self):
result = _safe({"key": "val"})
result = safe({"key": "val"})
assert result == '{"key": "val"}'
def test_list_becomes_json_string(self):
result = _safe([1, 2, 3])
result = safe([1, 2, 3])
assert result == "[1, 2, 3]"
def test_tuple_becomes_json_string(self):
result = _safe((1, 2))
result = safe((1, 2))
assert result == "[1, 2]"
def test_string_passthrough(self):
assert _safe("hello") == "hello"
assert safe("hello") == "hello"
def test_integer_passthrough(self):
assert _safe(99) == 99
assert safe(99) == 99
def test_none_passthrough(self):
assert _safe(None) is None
assert safe(None) is None
def test_nested_dict_serialised(self):
data = {"a": {"b": [1, 2]}}
result = _safe(data)
result = safe(data)
import json
assert json.loads(result) == data
# ── _search_preview_table ────────────────────────────────────────────────────
# ── format_euro ──────────────────────────────────────────────────────────────
class TestSearchPreviewTable:
class TestFormatEuro:
def test_formats_price(self):
assert format_euro(350000) == "€350,000"
def test_none_returns_dash(self):
assert format_euro(None) == ""
def test_zero_returns_dash(self):
assert format_euro(0) == ""
# ── format_area ──────────────────────────────────────────────────────────────
class TestFormatArea:
def test_formats_area(self):
assert format_area(80) == "80 m²"
def test_none_returns_dash(self):
assert format_area(None) == ""
def test_zero_returns_dash(self):
assert format_area(0) == ""
# ── md_preview_table ─────────────────────────────────────────────────────────
class TestMdPreviewTable:
def test_empty_rows_returns_header_only(self):
result = _search_preview_table([])
result = md_preview_table([], columns=[("title", "Title"), ("city", "City")])
lines = result.split("\n")
assert len(lines) == 2
assert "Title" in lines[0]
assert "---" in lines[1]
def test_single_row_appears(self):
rows = [
{
"title": "Teststraat 1",
"city": "Amsterdam",
"price": 350000,
"living_area": 80,
"bedrooms": 3,
}
]
result = _search_preview_table(rows)
rows = [{"title": "Teststraat 1", "city": "Amsterdam", "price": 350000}]
result = md_preview_table(
rows,
columns=[("title", "Title"), ("city", "City"), ("price", "Price")],
formatters={"price": format_euro},
)
assert "Teststraat 1" in result
assert "Amsterdam" in result
assert "€350,000" in result
assert "80 m²" in result
assert "3" in result
def test_missing_price_shows_dash(self):
rows = [{"title": "No Price", "city": "Rotterdam", "price": None}]
result = _search_preview_table(rows)
assert "" in result
def test_missing_area_shows_dash(self):
rows = [{"title": "No Area", "city": "Utrecht", "living_area": None}]
result = _search_preview_table(rows)
def test_missing_value_shows_dash(self):
rows = [{"title": "No Price", "city": "Rotterdam"}]
result = md_preview_table(
rows,
columns=[("title", "Title"), ("city", "City"), ("price", "Price")],
formatters={"price": format_euro},
)
assert "" in result
def test_multiple_rows_correct_count(self):
rows = [
{"title": f"St {i}", "city": "City", "price": i * 1000} for i in range(5)
]
result = _search_preview_table(rows)
rows = [{"title": f"St {i}", "city": "City"} for i in range(5)]
result = md_preview_table(rows, columns=[("title", "Title"), ("city", "City")])
lines = result.split("\n")
# header + separator + 5 data rows
assert len(lines) == 7
# ── _details_preview_table ───────────────────────────────────────────────────
class TestDetailsPreviewTable:
def test_empty_rows_returns_header_only(self):
result = _details_preview_table([])
lines = result.split("\n")
assert len(lines) == 2
assert "Title" in lines[0]
def test_row_with_all_fields(self):
rows = [
{
"title": "Kerkstraat 5",
"city": "Haarlem",
"price": 425000,
"status": "available",
"energy_label": "A",
}
]
result = _details_preview_table(rows)
assert "Kerkstraat 5" in result
assert "Haarlem" in result
assert "€425,000" in result
assert "available" in result
assert "A" in result
def test_missing_price_shows_dash(self):
rows = [{"title": "T", "city": "C", "price": None, "status": "sold"}]
result = _details_preview_table(rows)
assert "" in result