Skip to content

Commit b2c2c2b

Browse files
another try for nullable boolean series
1 parent a4de214 commit b2c2c2b

File tree

1 file changed

+4
-2
lines changed

1 file changed

+4
-2
lines changed

rdtools/test/analysis_chains_test.py

Lines changed: 4 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -403,7 +403,7 @@ def test_filter_ad_hoc_warnings(workflow, sensor_parameters):
403403
assert components["ad_hoc_filter"].all()
404404

405405
# warning about NaNs
406-
ad_hoc_filter = pd.Series(True, index=sensor_parameters["pv"].index)
406+
ad_hoc_filter = pd.Series(True, index=sensor_parameters["pv"].index, dtype="boolean")
407407
ad_hoc_filter.iloc[10] = pd.NA
408408
rd_analysis.filter_params["ad_hoc_filter"] = ad_hoc_filter
409409
with pytest.warns(
@@ -454,7 +454,9 @@ def test_aggregated_filter_ad_hoc_warnings(workflow, sensor_parameters):
454454
# disable all filters outside of CSI
455455
rd_analysis_2.filter_params = {"clearsky_filter": {"model": "csi"}}
456456
daily_ad_hoc_filter = pd.Series(True, index=sensor_parameters["pv"].index)
457-
daily_ad_hoc_filter = daily_ad_hoc_filter.resample("1D").first().dropna(how="all")
457+
daily_ad_hoc_filter = (
458+
daily_ad_hoc_filter.resample("1D").first().dropna(how="all").astype("boolean")
459+
)
458460
daily_ad_hoc_filter.iloc[10] = pd.NA
459461
rd_analysis_2.filter_params_aggregated["ad_hoc_filter"] = daily_ad_hoc_filter
460462
with pytest.warns(

0 commit comments

Comments
 (0)