From 690cd5234051c8b9caf5756cce6891c7612006cb Mon Sep 17 00:00:00 2001 From: Luke Manley Date: Mon, 23 Dec 2024 21:52:38 -0500 Subject: [PATCH] update test --- py-polars/tests/unit/series/test_series.py | 8 +++----- 1 file changed, 3 insertions(+), 5 deletions(-) diff --git a/py-polars/tests/unit/series/test_series.py b/py-polars/tests/unit/series/test_series.py index e2676f0d5c6f..683966ca783c 100644 --- a/py-polars/tests/unit/series/test_series.py +++ b/py-polars/tests/unit/series/test_series.py @@ -1373,13 +1373,11 @@ def test_filter() -> None: assert_series_equal(s.filter(mask), pl.Series("a", [1, 3])) assert_series_equal(s.filter([True, False, True]), pl.Series("a", [1, 3])) - - -def test_filter_sequence() -> None: - s = pl.Series("a", [1, 2, 3]) - assert_series_equal(s.filter([True, False, True]), pl.Series("a", [1, 3])) assert_series_equal(s.filter(np.array([True, False, True])), pl.Series("a", [1, 3])) + with pytest.raises(RuntimeError, match="Expected a boolean mask"): + s.filter(np.array([1, 0, 1])) + def test_gather_every() -> None: s = pl.Series("a", [1, 2, 3, 4])