Skip to content

Commit 00f6a6f

Browse files
committed
test: cover generic MVLR non-finite filtering
1 parent 9ae48d3 commit 00f6a6f

1 file changed

Lines changed: 20 additions & 0 deletions

File tree

tests/mvlr/test_source_data_filtering.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -102,3 +102,23 @@ def test_clean_regression_frame_keeps_original_data_when_filtering_too_much() ->
102102
assert not diagnostics.applied
103103
assert diagnostics.reason == "too much source data would be removed"
104104
assert len(cleaned) == 90
105+
106+
107+
def test_clean_regression_frame_removes_non_finite_rows_for_non_solar_models() -> None:
108+
"""Generic MVLR cleaning should drop non-finite observations."""
109+
index = pd.date_range("2025-04-01", periods=40, freq="D", tz="Europe/Brussels")
110+
frame = pd.DataFrame(
111+
{
112+
"energyConsumption": [10.0] * 39 + [float("nan")],
113+
"temperature": [12.0 + (i % 10) for i in range(40)],
114+
},
115+
index=index,
116+
)
117+
118+
cleaned, diagnostics = clean_regression_frame(frame, "energyConsumption")
119+
120+
assert diagnostics.applied
121+
assert diagnostics.reason == "generic non-finite filtering only"
122+
assert diagnostics.removed_non_finite_count == 1
123+
assert diagnostics.removed_observation_count == 1
124+
assert len(cleaned) == 39

0 commit comments

Comments
 (0)