Function test_dataset_memory_map

python/pyarrow/tests/parquet/test_dataset.py:746–757

(tempdir)

@pytest.mark.numpy
def test_dataset_memory_map(tempdir):
    # ARROW-2627: Check that we can use ParquetDataset with memory-mapping
    dirpath = tempdir / guid()
    dirpath.mkdir()

    table = _test_table(10, seed=0)
    path = dirpath / '0.parquet'
    _write_table(table, path, version='2.6')

    dataset = pq.ParquetDataset(
        dirpath, memory_map=True)
    assert dataset.read().equals(table)


@pytest.mark.numpy