#1934: Update unit tests to use new constructor of LBDataHolder

DARMA-tasking · Sep 23, 2024 · f519dca · f519dca
1 parent f252c53
commit f519dca
Show file tree

Hide file tree

Showing 5 changed files with 10 additions and 10 deletions.
diff --git a/tests/unit/collection/test_lb.extended.cc b/tests/unit/collection/test_lb.extended.cc
@@ -552,7 +552,7 @@ TEST_F(TestRestoreLBData, test_restore_lb_data_data_1) {
     );
   });
 
-  vt::vrt::collection::balance::LBDataHolder lbdh;
+  vt::vrt::collection::balance::LBDataHolder lbdh(num_phases);
   PhaseType write_phase = 0;
 
   using CommKey = vt::elm::CommKey;
@@ -749,7 +749,7 @@ TEST_P(TestDumpUserdefinedData, test_dump_userdefined_json) {
     );
   });
 
-  vt::vrt::collection::balance::LBDataHolder lbdh;
+  vt::vrt::collection::balance::LBDataHolder lbdh(num_phases);
   PhaseType write_phase = 0;
 
   {
@@ -812,7 +812,7 @@ TEST_P(TestDumpAttributesFieldData, test_dump_attributes_json) {
     );
   });
 
-  vt::vrt::collection::balance::LBDataHolder lbdh;
+  vt::vrt::collection::balance::LBDataHolder lbdh(num_phases);
   PhaseType phase = 0;
 
   {

diff --git a/tests/unit/collection/test_lb_data_holder.cc b/tests/unit/collection/test_lb_data_holder.cc
@@ -74,7 +74,7 @@ void addPhasesDataToJson(nlohmann::json& json, PhaseType amountOfPhasesToAdd, st
     }
   }
 
-  LBDataHolder dh;
+  LBDataHolder dh(amountOfPhasesToAdd);
   for (unsigned i = 0; i < amountOfPhasesToAdd; i++) {
     for (auto&& elm : ids[i]) {
       dh.node_data_[i][elm] = LoadSummary{3.};

diff --git a/tests/unit/collection/test_workload_data_migrator.cc b/tests/unit/collection/test_workload_data_migrator.cc
@@ -93,7 +93,7 @@ setupWorkloads(PhaseType phase, size_t numElements) {
     );
   }
 
-  auto lbdh = std::make_shared<LBDataHolder>();
+  auto lbdh = std::make_shared<LBDataHolder>(1);
 
   for (auto&& elmID : myElemList) {
     double tval = elmID.id * 2;
@@ -790,7 +790,7 @@ setupManyWorkloads(
     );
   }
 
-  auto lbdh = std::make_shared<LBDataHolder>();
+  auto lbdh = std::make_shared<LBDataHolder>(num_phases);
 
   PhaseType stop_phase = initial_phase + num_phases;
   for (PhaseType phase = initial_phase; phase < stop_phase; ++phase) {
@@ -804,7 +804,7 @@ setupManyWorkloads(
     }
   }
 
-  auto scrambled_lbdh = std::make_shared<LBDataHolder>();
+  auto scrambled_lbdh = std::make_shared<LBDataHolder>(num_phases);
 
   for (PhaseType phase = initial_phase; phase < stop_phase; ++phase) {
     auto base_load_model = setupBaseModel(phase, lbdh);

diff --git a/tests/unit/lb/test_offlinelb.cc b/tests/unit/lb/test_offlinelb.cc
@@ -125,7 +125,7 @@ TEST_F(TestOfflineLB, test_offlinelb_1) {
     ids[5].push_back(nid);
   }
 
-  LBDataHolder dh;
+  LBDataHolder dh(num_phases);
   for (PhaseType i = 0; i < num_phases; i++) {
     for (auto&& elm : ids[i]) {
       dh.node_data_[i][elm] = LoadSummary{3};
@@ -196,7 +196,7 @@ TEST_F(TestOfflineLB, test_offlinelb_2) {
     ids[7].push_back(nid);
   }
 
-  LBDataHolder dh;
+  LBDataHolder dh(num_phases);
   for (PhaseType i = 0; i < num_phases; i++) {
     if (i != 1 and i != 2 and i != 5 and i != 8 and i != 9) {
       auto& elms = ids[i];

diff --git a/tests/unit/runtime/test_initialization.cc b/tests/unit/runtime/test_initialization.cc
@@ -543,7 +543,7 @@ void prepareLBDataFiles(const std::string file_name_without_ext) {
     ids[5].push_back(nid);
   }
 
-  LBDataHolder dh;
+  LBDataHolder dh(num_phases);
   for (PhaseType i = 0; i < num_phases; i++) {
     for (auto&& elm : ids[i]) {
       dh.node_data_[i][elm] = LoadSummary{3};