RussellSpitzer commented on code in PR #10659:
URL: https://github.com/apache/iceberg/pull/10659#discussion_r1693627142
##########
spark/v3.5/spark/src/test/java/org/apache/iceberg/spark/source/TestSparkScan.java:
##########
@@ -130,6 +142,98 @@ public void testEstimatedRowCount() throws
NoSuchTableException {
assertThat(stats.numRows().getAsLong()).isEqualTo(10000L);
}
+ @TestTemplate
+ public void testColStats() throws NoSuchTableException {
+ sql("CREATE TABLE %s (id int, data string) USING iceberg", tableName);
+
+ List<SimpleRecord> records =
+ Lists.newArrayList(
+ new SimpleRecord(1, "a"),
+ new SimpleRecord(2, "b"),
+ new SimpleRecord(3, "a"),
+ new SimpleRecord(4, "b"));
+ spark
+ .createDataset(records, Encoders.bean(SimpleRecord.class))
+ .coalesce(1)
+ .writeTo(tableName)
+ .append();
+
+ Table table = validationCatalog.loadTable(tableIdent);
+ long snapshotId = table.currentSnapshot().snapshotId();
+
+ SparkScanBuilder scanBuilder =
+ new SparkScanBuilder(spark, table, CaseInsensitiveStringMap.empty());
+ SparkScan scan = (SparkScan) scanBuilder.build();
+
+ Map<String, String> reportColStatsDisabled =
+ ImmutableMap.of(
+ SQLConf.CBO_ENABLED().key(), "true",
SparkSQLProperties.REPORT_COLUMN_STATS, "false");
+
+ Map<String, String> reportColStatsEnabled =
+ ImmutableMap.of(SQLConf.CBO_ENABLED().key(), "true");
+
+ // Test table does not have col stats
+ checkColStatisticsNotReported(scan, 4L);
+ withSQLConf(reportColStatsDisabled, () ->
checkColStatisticsNotReported(scan, 4L));
+ // The expected col NDVs are nulls
+ withSQLConf(reportColStatsEnabled, () -> checkColStatisticsReported(scan,
4L, new HashMap<>()));
+
+ GenericStatisticsFile statisticsFile =
+ new GenericStatisticsFile(
+ snapshotId,
+ "/test/statistics/file.puffin",
+ 100,
+ 42,
+ ImmutableList.of(
+ new GenericBlobMetadata(
+ APACHE_DATASKETCHES_THETA_V1,
+ snapshotId,
+ 1,
+ ImmutableList.of(1),
+ ImmutableMap.of("ndv", "4"))));
+
+ table.updateStatistics().setStatistics(snapshotId,
statisticsFile).commit();
+
+ // Test where 1 column has NDV and the other does not have NDV
+ checkColStatisticsNotReported(scan, 4L);
+ withSQLConf(reportColStatsDisabled, () ->
checkColStatisticsNotReported(scan, 4L));
+
+ HashMap<String, Long> expectedOneNDV = new HashMap<>();
+ expectedOneNDV.put("id", 4L);
+ withSQLConf(reportColStatsEnabled, () -> checkColStatisticsReported(scan,
4L, expectedOneNDV));
+
+ statisticsFile =
Review Comment:
I'd split this one out into another test, minor nit
--
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
