Skip to content

Commit 6378ce5

Browse files
committed
Update aggregate_test.cc
1 parent 24d5c7c commit 6378ce5

File tree

1 file changed

+52
-48
lines changed

1 file changed

+52
-48
lines changed

src/iceberg/test/aggregate_test.cc

Lines changed: 52 additions & 48 deletions
Original file line numberDiff line numberDiff line change
@@ -251,16 +251,15 @@ TEST(AggregateTest, AggregatesFromDataFileMetrics) {
251251
count_bound, count_null_bound, count_star_bound, max_bound, min_bound};
252252
ICEBERG_UNWRAP_OR_FAIL(auto evaluator, AggregateEvaluator::Make(aggregates));
253253

254-
DataFile file;
255-
file.record_count = 10;
256-
file.value_counts.emplace(1, 10);
257-
file.null_value_counts.emplace(1, 2);
258-
file.value_counts.emplace(2, 10);
259-
file.null_value_counts.emplace(2, 0);
260254
ICEBERG_UNWRAP_OR_FAIL(auto lower, Literal::Int(5).Serialize());
261255
ICEBERG_UNWRAP_OR_FAIL(auto upper, Literal::Int(50).Serialize());
262-
file.lower_bounds.emplace(2, lower);
263-
file.upper_bounds.emplace(2, upper);
256+
DataFile file{
257+
.record_count = 10,
258+
.value_counts = {{1, 10}, {2, 10}},
259+
.null_value_counts = {{1, 2}, {2, 0}},
260+
.lower_bounds = {{2, lower}},
261+
.upper_bounds = {{2, upper}},
262+
};
264263

265264
ASSERT_TRUE(evaluator->Update(file).has_value());
266265

@@ -287,8 +286,7 @@ TEST(AggregateTest, AggregatesFromDataFileMissingMetricsReturnNull) {
287286
count_bound, count_null_bound, count_star_bound, max_bound, min_bound};
288287
ICEBERG_UNWRAP_OR_FAIL(auto evaluator, AggregateEvaluator::Make(aggregates));
289288

290-
DataFile file;
291-
file.record_count = -1; // missing/invalid
289+
DataFile file{.record_count = -1}; // missing/invalid
292290

293291
ASSERT_TRUE(evaluator->Update(file).has_value());
294292

@@ -309,14 +307,15 @@ TEST(AggregateTest, AggregatesFromDataFileWithTransform) {
309307
std::vector<std::shared_ptr<BoundAggregate>> aggregates{max_bound, min_bound};
310308
ICEBERG_UNWRAP_OR_FAIL(auto evaluator, AggregateEvaluator::Make(aggregates));
311309

312-
DataFile file;
313-
file.record_count = 5;
314-
file.value_counts.emplace(1, 5);
315-
file.null_value_counts.emplace(1, 0);
316310
ICEBERG_UNWRAP_OR_FAIL(auto lower, Literal::Int(5).Serialize());
317311
ICEBERG_UNWRAP_OR_FAIL(auto upper, Literal::Int(23).Serialize());
318-
file.lower_bounds.emplace(1, lower);
319-
file.upper_bounds.emplace(1, upper);
312+
DataFile file{
313+
.record_count = 5,
314+
.value_counts = {{1, 5}},
315+
.null_value_counts = {{1, 0}},
316+
.lower_bounds = {{1, lower}},
317+
.upper_bounds = {{1, upper}},
318+
};
320319

321320
ASSERT_TRUE(evaluator->Update(file).has_value());
322321

@@ -328,7 +327,7 @@ TEST(AggregateTest, AggregatesFromDataFileWithTransform) {
328327
EXPECT_TRUE(evaluator->AllAggregatorsValid());
329328
}
330329

331-
TEST(AggregateTest, DataFileAggregatorParityWithJava) {
330+
TEST(AggregateTest, DataFileAggregatorParity) {
332331
Schema schema({SchemaField::MakeRequired(1, "id", int32()),
333332
SchemaField::MakeOptional(2, "no_stats", int32()),
334333
SchemaField::MakeOptional(3, "all_nulls", string()),
@@ -345,43 +344,48 @@ TEST(AggregateTest, DataFileAggregatorParityWithJava) {
345344
};
346345

347346
auto [b1_lower, b1_upper] = make_bounds(1, 33, 2345);
348-
DataFile file;
349-
file.file_path = "file.avro";
350-
file.record_count = 50;
351-
file.value_counts = {{1, 50}, {3, 50}, {4, 50}};
352-
file.null_value_counts = {{1, 10}, {3, 50}, {4, 10}};
353-
file.lower_bounds = b1_lower;
354-
file.upper_bounds = b1_upper;
347+
DataFile file{
348+
.file_path = "file.avro",
349+
.record_count = 50,
350+
.value_counts = {{1, 50}, {3, 50}, {4, 50}},
351+
.null_value_counts = {{1, 10}, {3, 50}, {4, 10}},
352+
.lower_bounds = std::move(b1_lower),
353+
.upper_bounds = std::move(b1_upper),
354+
};
355355

356356
auto [b2_lower, b2_upper] = make_bounds(1, 33, 100);
357-
DataFile missing_some_nulls_1;
358-
missing_some_nulls_1.file_path = "file_2.avro";
359-
missing_some_nulls_1.record_count = 20;
360-
missing_some_nulls_1.value_counts = {{1, 20}, {3, 20}};
361-
missing_some_nulls_1.null_value_counts = {{1, 0}, {3, 20}};
362-
missing_some_nulls_1.lower_bounds = b2_lower;
363-
missing_some_nulls_1.upper_bounds = b2_upper;
357+
DataFile missing_some_nulls_1{
358+
.file_path = "file_2.avro",
359+
.record_count = 20,
360+
.value_counts = {{1, 20}, {3, 20}},
361+
.null_value_counts = {{1, 0}, {3, 20}},
362+
.lower_bounds = std::move(b2_lower),
363+
.upper_bounds = std::move(b2_upper),
364+
};
364365

365366
auto [b3_lower, b3_upper] = make_bounds(1, -33, 3333);
366-
DataFile missing_some_nulls_2;
367-
missing_some_nulls_2.file_path = "file_3.avro";
368-
missing_some_nulls_2.record_count = 20;
369-
missing_some_nulls_2.value_counts = {{1, 20}, {3, 20}};
370-
missing_some_nulls_2.null_value_counts = {{1, 20}, {3, 20}};
371-
missing_some_nulls_2.lower_bounds = b3_lower;
372-
missing_some_nulls_2.upper_bounds = b3_upper;
373-
374-
DataFile missing_some_stats;
375-
missing_some_stats.file_path = "file_missing_stats.avro";
376-
missing_some_stats.record_count = 20;
377-
missing_some_stats.value_counts = {{1, 20}, {4, 10}};
367+
DataFile missing_some_nulls_2{
368+
.file_path = "file_3.avro",
369+
.record_count = 20,
370+
.value_counts = {{1, 20}, {3, 20}},
371+
.null_value_counts = {{1, 20}, {3, 20}},
372+
.lower_bounds = std::move(b3_lower),
373+
.upper_bounds = std::move(b3_upper),
374+
};
375+
376+
DataFile missing_some_stats{
377+
.file_path = "file_missing_stats.avro",
378+
.record_count = 20,
379+
.value_counts = {{1, 20}, {4, 10}},
380+
};
378381
auto [b4_lower, b4_upper] = make_bounds(1, -3, 1333);
379-
missing_some_stats.lower_bounds = b4_lower;
380-
missing_some_stats.upper_bounds = b4_upper;
382+
missing_some_stats.lower_bounds = std::move(b4_lower);
383+
missing_some_stats.upper_bounds = std::move(b4_upper);
381384

382-
DataFile missing_all_optional_stats;
383-
missing_all_optional_stats.file_path = "file_null_stats.avro";
384-
missing_all_optional_stats.record_count = 20;
385+
DataFile missing_all_optional_stats{
386+
.file_path = "file_null_stats.avro",
387+
.record_count = 20,
388+
};
385389

386390
auto run_case = [&](const std::vector<std::shared_ptr<Expression>>& exprs,
387391
const std::vector<DataFile>& files,

0 commit comments

Comments
 (0)