c

nameexhaustion · Feb 20, 2024 · f408a80 · f408a80
1 parent 39470e5
commit f408a80
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/py-polars/tests/unit/datatypes/test_float.py b/py-polars/tests/unit/datatypes/test_float.py
@@ -80,6 +80,23 @@ def test_unique(s: pl.Series, expect: pl.Series) -> None:
     assert_series_equal(expect, out)
 
 
+def test_unique_counts() -> None:
+    s = pl.Series(
+        "x",
+        [
+            -0.0,
+            0.0,
+            float("-nan"),
+            float("nan"),
+            1.0,
+            None,
+        ],
+    )
+    expect = pl.Series("x", [2, 2, 1, 1], dtype=pl.UInt32)
+    out = s.unique_counts()
+    assert_series_equal(expect, out)
+
+
 def test_hash() -> None:
     s = pl.Series(
         "x",