add test for 20712

pola-rs · Jan 15, 2025 · 9dc2f2c · 9dc2f2c
1 parent 7f2a2bf
commit 9dc2f2c
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/py-polars/tests/unit/operations/test_slice.py b/py-polars/tests/unit/operations/test_slice.py
@@ -310,3 +310,14 @@ def test_slice_pushdown_panic_20216() -> None:
 
     assert_frame_equal(q.slice(0, 1).collect(), pl.DataFrame({"A": ["1"]}))
     assert_frame_equal(q.collect(), pl.DataFrame({"A": ["1"]}))
+
+
+def test_sorted_slice_after_function_20712() -> None:
+    assert_frame_equal(
+        pl.LazyFrame({"a": 10 * ["A"]})
+        .with_columns(b=pl.col("a").str.extract("(.*)"))
+        .sort("b")
+        .head(2)
+        .collect(),
+        pl.DataFrame({"a": ["A", "A"], "b": ["A", "A"]}),
+    )