diff --git a/cpp/tests/reductions/segmented_reduction_tests.cpp b/cpp/tests/reductions/segmented_reduction_tests.cpp index fd243d53dcc..3fa369d6a53 100644 --- a/cpp/tests/reductions/segmented_reduction_tests.cpp +++ b/cpp/tests/reductions/segmented_reduction_tests.cpp @@ -389,14 +389,10 @@ TEST_F(SegmentedReductionTestUntyped, ReduceEmptyColumn) TEST_F(SegmentedReductionTestUntyped, EmptyInputWithOffsets) { - // values: {} - // offsets: {0, 0} - // outputs: {XXX} - // output nullmask: {0} auto input = fixed_width_column_wrapper{}; - auto offsets = std::vector{0, 0}; + auto offsets = std::vector{0, 0, 0, 0, 0, 0}; auto d_offsets = thrust::device_vector(offsets); - auto expect = fixed_width_column_wrapper{{XXX}, {0}}; + auto expect = fixed_width_column_wrapper{{XXX, XXX, XXX, XXX, XXX}, {0, 0, 0, 0, 0}}; auto aggregates = std::vector{{XXX}, {0}}; + auto expect_bool = fixed_width_column_wrapper{{XXX, XXX, XXX, XXX, XXX}, {0, 0, 0, 0, 0}}; auto result = segmented_reduce(input, d_offsets,