Expect ColumnNotFoundError in multiple encountered errors test

pola-rs · Dec 16, 2023 · b16860c · b16860c
1 parent ae95c79
commit b16860c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/py-polars/tests/unit/test_errors.py b/py-polars/tests/unit/test_errors.py
@@ -683,7 +683,7 @@ def cool_stuff(lf: pl.LazyFrame) -> pl.LazyFrame:
     for _ in range(n):
         df = cool_stuff(df)
 
-    with pytest.raises(pl.ComputeError, match="Multiple AlreadyEncountered"):
+    with pytest.raises(pl.ColumnNotFoundError):
         df.collect()