Update error message

pola-rs · Apr 3, 2024 · 0402126 · 0402126
1 parent bab7462
commit 0402126
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/py-polars/polars/dataframe/frame.py b/py-polars/polars/dataframe/frame.py
@@ -6913,7 +6913,7 @@ def clear(self, n: int = 0) -> Self:
         └──────┴──────┴──────┘
         """
         if n < 0:
-            msg = "n should be greater than or equal to 0."
+            msg = f"`n` should be greater than or equal to 0, got {n}"
             raise ValueError(msg)
         # faster path
         if n == 0:

diff --git a/py-polars/polars/series/series.py b/py-polars/polars/series/series.py
@@ -4760,7 +4760,7 @@ def clear(self, n: int = 0) -> Series:
         ]
         """
         if n < 0:
-            msg = "n should be greater than or equal to 0."
+            msg = f"`n` should be greater than or equal to 0, got {n}"
             raise ValueError(msg)
         # faster path
         if n == 0:

diff --git a/py-polars/tests/unit/operations/test_clear.py b/py-polars/tests/unit/operations/test_clear.py
@@ -77,9 +77,9 @@ def test_clear_series_object_starting_with_null() -> None:
 
 def test_clear_raise_negative_n() -> None:
     s = pl.Series([1, 2, 3])
-    with pytest.raises(ValueError, match="n should be greater than or equal to 0"):
-        s.clear(-1)
 
-    df = pl.DataFrame({"a": [1, 2, 3]})
-    with pytest.raises(ValueError, match="n should be greater than or equal to 0"):
-        df.clear(-1)
+    msg = "`n` should be greater than or equal to 0, got -1"
+    with pytest.raises(ValueError, match=msg):
+        s.clear(-1)
+    with pytest.raises(ValueError, match=msg):
+        s.to_frame().clear(-1)