Skip to content

Commit

Permalink
chore: update arrow (#5940)
Browse files Browse the repository at this point in the history
  • Loading branch information
ritchie46 committed Dec 30, 2022
1 parent 30613db commit 553069e
Show file tree
Hide file tree
Showing 3 changed files with 31 additions and 4 deletions.
4 changes: 2 additions & 2 deletions Cargo.toml
Original file line number Diff line number Diff line change
Expand Up @@ -34,10 +34,10 @@ bitflags = "1.3"
[workspace.dependencies.arrow]
package = "arrow2"
# git = "https://github.com/jorgecarleitao/arrow2"
# git = "https://github.com/ritchie46/arrow2"
git = "https://github.com/ritchie46/arrow2"
# rev = "368aacc173a27e2a763d2c6396682a688e5a2707"
# path = "../arrow2"
# branch = "polars_2022-12-13"
branch = "polars_2022-12-29"
version = "0.15"
default-features = false
features = [
Expand Down
3 changes: 1 addition & 2 deletions py-polars/Cargo.lock

Some generated files are not rendered by default. Learn more about how customized files appear on GitHub.

28 changes: 28 additions & 0 deletions py-polars/tests/unit/io/test_parquet.py
Original file line number Diff line number Diff line change
Expand Up @@ -2,6 +2,7 @@

import io
import os
import typing
from typing import TYPE_CHECKING

import numpy as np
Expand Down Expand Up @@ -304,3 +305,30 @@ def test_parquet_5795() -> None:
df_pd.to_parquet(f)
f.seek(0)
assert pl.read_parquet(f).frame_equal(pl.from_pandas(df_pd))


@typing.no_type_check
def test_parquet_nesting_structs_list() -> None:
f = io.BytesIO()
df = pl.from_records(
[
{
"id": 1,
"list_of_structs_col": [
{"a": 10, "b": [10, 11, 12]},
{"a": 11, "b": [13, 14, 15]},
],
},
{
"id": 2,
"list_of_structs_col": [
{"a": 44, "b": [12]},
],
},
]
)

df.write_parquet(f)
f.seek(0)

assert pl.read_parquet(f).frame_equal(df)

0 comments on commit 553069e

Please sign in to comment.