oops

py-polars/tests/unit/io/test_scan.py

@@ -390,6 +390,38 @@ def test_scan_with_row_index_limit_and_filter(
     )
 
 
+@pytest.mark.write_disk()
+def test_scan_with_row_index_filter_and_limit(
+    capfd: Any, monkeypatch: pytest.MonkeyPatch, data_file: _DataFile, force_async: bool
+) -> None:
+    if data_file.path.suffix == ".csv" and force_async:
+        pytest.skip(reason="async reading of .csv not yet implemented")
+
+    if force_async:
+        _enable_force_async(monkeypatch)
+
+    df = (
+        _scan(data_file.path, data_file.df.schema, row_index=_RowIndex())
+        .filter(pl.col("sequence") % 2 == 0)
+        .limit(4483)
+        .collect()
+    )
+
+    if force_async:
+        _assert_force_async(capfd)
+
+    assert_frame_equal(
+        df,
+        pl.DataFrame(
+            {
+                "index": (2 * x for x in range(4483)),
+                "sequence": (2 * x for x in range(4483)),
+            },
+            schema_overrides={"index": pl.UInt32},
+        ),
+    )
+
+
 @pytest.mark.write_disk()
 @pytest.mark.parametrize(
     ("scan_func", "write_func"),