From 130469888ab27a6f14d4e7a81541564aed35c9c2 Mon Sep 17 00:00:00 2001 From: Kyle Barron Date: Thu, 25 Sep 2025 14:24:00 -0600 Subject: [PATCH] test: Add test for reading and writing string_view arrays to Parquet files --- tests/io/test_parquet.py | 13 +++++++++++++ 1 file changed, 13 insertions(+) diff --git a/tests/io/test_parquet.py b/tests/io/test_parquet.py index 1d41d34a..59f7cbde 100644 --- a/tests/io/test_parquet.py +++ b/tests/io/test_parquet.py @@ -2,6 +2,7 @@ import pyarrow as pa import pyarrow.parquet as pq +from arro3.core import Array, DataType, Table from arro3.io import read_parquet, write_parquet @@ -42,3 +43,15 @@ def test_copy_parquet_kv_metadata(): reader = read_parquet("test.parquet") assert reader.schema.metadata[b"hello"] == b"world" + + +def test_string_view(): + arr = Array(["foo", "bar", "baz"], type=DataType.string_view()) + table = Table.from_arrays([arr], names=["a"]) + + bio = BytesIO() + write_parquet(table, bio) + bio.seek(0) + + table_retour = read_parquet(bio).read_all() + assert table == table_retour