()
| 294 | |
| 295 | |
| 296 | def test_buffer_readwrite(): |
| 297 | from pyarrow import orc |
| 298 | |
| 299 | buffer_output_stream = pa.BufferOutputStream() |
| 300 | a = pa.array([1, None, 3, None]) |
| 301 | b = pa.array([None, "Arrow", None, "ORC"]) |
| 302 | table = pa.table({"int64": a, "utf8": b}) |
| 303 | orc.write_table(table, buffer_output_stream) |
| 304 | buffer_reader = pa.BufferReader(buffer_output_stream.getvalue()) |
| 305 | orc_file = orc.ORCFile(buffer_reader) |
| 306 | output_table = orc_file.read() |
| 307 | assert table.equals(output_table) |
| 308 | # Check for default WriteOptions |
| 309 | assert orc_file.compression == 'UNCOMPRESSED' |
| 310 | assert orc_file.file_version == '0.12' |
| 311 | assert orc_file.row_index_stride == 10000 |
| 312 | assert orc_file.compression_size == 65536 |
| 313 | |
| 314 | # deprecated keyword order |
| 315 | buffer_output_stream = pa.BufferOutputStream() |
| 316 | with pytest.warns(FutureWarning): |
| 317 | orc.write_table(buffer_output_stream, table) |
| 318 | buffer_reader = pa.BufferReader(buffer_output_stream.getvalue()) |
| 319 | orc_file = orc.ORCFile(buffer_reader) |
| 320 | output_table = orc_file.read() |
| 321 | assert table.equals(output_table) |
| 322 | # Check for default WriteOptions |
| 323 | assert orc_file.compression == 'UNCOMPRESSED' |
| 324 | assert orc_file.file_version == '0.12' |
| 325 | assert orc_file.row_index_stride == 10000 |
| 326 | assert orc_file.compression_size == 65536 |
| 327 | |
| 328 | |
| 329 | @pytest.mark.snappy |
nothing calls this directly
no test coverage detected