MCPcopy Create free account
hub / github.com/apache/arrow / assert_yields_projected

Function assert_yields_projected

python/pyarrow/tests/test_dataset.py:1452–1460  ·  view source on GitHub ↗
(fragment, row_slice,
                                columns=None, filter=None)

Source from the content-addressed store, hash-verified

1450 table, dataset = _create_dataset_for_fragments(tempdir)
1451
1452 def assert_yields_projected(fragment, row_slice,
1453 columns=None, filter=None):
1454 actual = fragment.to_table(
1455 schema=table.schema, columns=columns, filter=filter)
1456 column_names = columns if columns else table.column_names
1457 assert actual.column_names == column_names
1458
1459 expected = table.slice(*row_slice).select(column_names)
1460 assert actual.equals(expected)
1461
1462 fragment = list(dataset.get_fragments())[0]
1463 parquet_format = fragment.format

Callers 1

Calls 4

equalsMethod · 0.80
to_tableMethod · 0.45
selectMethod · 0.45
sliceMethod · 0.45

Tested by

no test coverage detected