binayakd commented on code in PR #1354: URL: https://github.com/apache/iceberg-python/pull/1354#discussion_r1855019099
########## tests/io/test_pyarrow_stats.py: ########## @@ -681,6 +685,39 @@ def test_stats_types(table_schema_nested: Schema) -> None: ] +def test_read_missing_statistics() -> None: + # write statistics for only for "strings" column + metadata, table_metadata = construct_test_table(write_statistics=["strings"]) + + # expect only "strings" column to have statistics in metadata + assert metadata.row_group(0).column(0).is_stats_set is True + assert metadata.row_group(0).column(0).statistics is not None + + # expect all other columns to have no statistics + for r in range(metadata.num_row_groups): + for pos in range(1, metadata.num_columns): + assert metadata.row_group(r).column(pos).is_stats_set is False + assert metadata.row_group(r).column(pos).statistics is None + + schema = get_current_schema(table_metadata) + statistics = data_file_statistics_from_parquet_metadata( + parquet_metadata=metadata, + stats_columns=compute_statistics_plan(schema, table_metadata.properties), + parquet_column_mapping=parquet_path_to_id_mapping(schema), + ) + + datafile = DataFile(**statistics.to_serialized_dict()) + + # expect only "strings" column values to be reflected in the + # upper_bound, lower_bound and null_value_counts props of datafile + assert len(datafile.lower_bounds) == 1 + assert datafile.lower_bounds[1].decode() == "aaaaaaaaaaaaaaaa" Review Comment: There the bounds attributes are actually Dicts, and `1` is the key to access the values for the column in position 1, the "strings" column. So looks like Pyarrow sets up the column positions with 1 index. -- This is an automated message from the Apache Git Service. To respond to the message, please log on to GitHub and use the URL above to go to the specific comment. To unsubscribe, e-mail: issues-unsubscr...@iceberg.apache.org For queries about this service, please contact Infrastructure at: us...@infra.apache.org --------------------------------------------------------------------- To unsubscribe, e-mail: issues-unsubscr...@iceberg.apache.org For additional commands, e-mail: issues-h...@iceberg.apache.org