Group builder modified for incremental building (#5098)

2019-12-10 14:33:56 +13:00
parent 1cb6bcc382
commit 979f74d51a
5 changed files with 126 additions and 30 deletions
--- a/tests/cpp/common/test_group_data.cc
+++ b/tests/cpp/common/test_group_data.cc
@@ -35,7 +35,8 @@ TEST(group_data, ParallelGroupBuilder) {
  EXPECT_EQ(offsets, expected_offsets);

  // Create new builder, add one more row given already populated offsets/data
-  ParallelGroupBuilder<Entry, size_t> builder2(&offsets, &data);
+  ParallelGroupBuilder<Entry, size_t> builder2(&offsets, &data,
+                                               offsets.size() - 1);
  builder2.InitBudget(0, 1);
  builder2.AddBudget(2, 0, 2);
  builder2.InitStorage();