(dataset, dataset_reader)
| 674 | |
| 675 | @pytest.mark.parquet |
| 676 | def test_count_rows(dataset, dataset_reader): |
| 677 | fragment = next(dataset.get_fragments()) |
| 678 | assert dataset_reader.count_rows(fragment) == 5 |
| 679 | assert dataset_reader.count_rows( |
| 680 | fragment, filter=ds.field("i64") == 4) == 1 |
| 681 | |
| 682 | assert dataset_reader.count_rows(dataset) == 10 |
| 683 | # Filter on partition key |
| 684 | assert dataset_reader.count_rows( |
| 685 | dataset, filter=ds.field("group") == 1) == 5 |
| 686 | # Filter on data |
| 687 | assert dataset_reader.count_rows(dataset, filter=ds.field("i64") >= 3) == 4 |
| 688 | assert dataset_reader.count_rows(dataset, filter=ds.field("i64") < 0) == 0 |
| 689 | |
| 690 | |
| 691 | def test_abstract_classes(): |
nothing calls this directly
no test coverage detected