Wes McKinney created ARROW-10121:
------------------------------------
Summary: [C++][Python] Variable dictionaries do not survive
roundtrip to IPC stream
Key: ARROW-10121
URL: https://issues.apache.org/jira/browse/ARROW-10121
Project: Apache Arrow
Issue Type: Bug
Components: C++, Python
Reporter: Wes McKinney
Fix For: 2.0.0
Failing test case (from dev@
https://lists.apache.org/thread.html/r338942b4e9f9316b48e87aab41ac49c7ffedd45733d4a6349523b7eb%40%3Cdev.arrow.apache.org%3E)
{code}
import pyarrow as pa
from io import BytesIO
pa.__version__
schema = pa.schema([pa.field('foo', pa.int32()), pa.field('bar',
pa.dictionary(pa.int32(), pa.string()))] )
r1 = pa.record_batch(
[
[1, 2, 3, 4, 5],
pa.array(["a", "b", "c", "d", "e"]).dictionary_encode()
],
schema
)
r1.validate()
r2 = pa.record_batch(
[
[1, 2, 3, 4, 5],
pa.array(["c", "c", "e", "f", "g"]).dictionary_encode()
],
schema
)
r2.validate()
assert r1.column(1).dictionary != r2.column(1).dictionary
sink = pa.BufferOutputStream()
writer = pa.RecordBatchStreamWriter(sink, schema)
writer.write(r1)
writer.write(r2)
serialized = BytesIO(sink.getvalue().to_pybytes())
stream = pa.ipc.open_stream(serialized)
deserialized = []
while True:
try:
deserialized.append(stream.read_next_batch())
except StopIteration:
break
assert deserialized[1][1].to_pylist() == r2[1].to_pylist()
{code}
--
This message was sent by Atlassian Jira
(v8.3.4#803005)
