fix(rust): fix nested writer (#5777)

pola-rs · Dec 12, 2022 · fe2a03a · fe2a03a
1 parent d89aef6
commit fe2a03a
Show file tree

Hide file tree

Showing 3 changed files with 17 additions and 3 deletions.
diff --git a/Cargo.toml b/Cargo.toml
@@ -35,8 +35,8 @@ package = "arrow2"
 # git = "https://github.com/jorgecarleitao/arrow2"
 git = "https://github.com/ritchie46/arrow2"
 # rev = "368aacc173a27e2a763d2c6396682a688e5a2707"
-# path = "../../../arrow2"
-branch = "polars_2022-12-10"
+# path = "../arrow2"
+branch = "polars_2022-12-11"
 version = "0.14.1"
 default-features = false
 features = [

diff --git a/py-polars/Cargo.lock b/py-polars/Cargo.lock
diff --git a/py-polars/tests/unit/io/test_parquet.py b/py-polars/tests/unit/io/test_parquet.py
@@ -250,3 +250,17 @@ def test_row_group_size_saturation() -> None:
     df.write_parquet(f, row_group_size=1024)
     f.seek(0)
     assert pl.read_parquet(f).frame_equal(df)
+
+
+def test_nested_sliced() -> None:
+    for df in [
+        pl.Series([[1, 2], [3, 4], [5, 6]]).slice(2, 2).to_frame(),
+        pl.Series([[None, 2], [3, 4], [5, 6]]).to_frame(),
+        pl.Series([[None, 2], [3, 4], [5, 6]]).slice(2, 2).to_frame(),
+        pl.Series([["a", "a"], ["", "a"], ["c", "de"]]).slice(3, 2).to_frame(),
+        pl.Series([[None, True], [False, False], [True, True]]).slice(2, 2).to_frame(),
+    ]:
+        f = io.BytesIO()
+        df.write_parquet(f)
+        f.seek(0)
+        assert pl.read_parquet(f).frame_equal(df)