(tempdir, dataset_reader)
| 3366 | |
| 3367 | |
| 3368 | def test_ipc_format(tempdir, dataset_reader): |
| 3369 | table = pa.table({'a': pa.array([1, 2, 3], type="int8"), |
| 3370 | 'b': pa.array([.1, .2, .3], type="float64")}) |
| 3371 | |
| 3372 | path = str(tempdir / 'test.arrow') |
| 3373 | with pa.output_stream(path) as sink: |
| 3374 | writer = pa.RecordBatchFileWriter(sink, table.schema) |
| 3375 | writer.write_batch(table.to_batches()[0]) |
| 3376 | writer.close() |
| 3377 | |
| 3378 | dataset = ds.dataset(path, format=ds.IpcFileFormat()) |
| 3379 | result = dataset_reader.to_table(dataset) |
| 3380 | assert result.equals(table) |
| 3381 | |
| 3382 | assert_dataset_fragment_convenience_methods(dataset) |
| 3383 | |
| 3384 | for format_str in ["ipc", "arrow"]: |
| 3385 | dataset = ds.dataset(path, format=format_str) |
| 3386 | result = dataset_reader.to_table(dataset) |
| 3387 | assert result.equals(table) |
| 3388 | |
| 3389 | |
| 3390 | @pytest.mark.orc |
nothing calls this directly
no test coverage detected