#1122: test: expect different result without lb

DARMA-tasking · Nov 6, 2020 · beb19ae · beb19ae
1 parent da88437
commit beb19ae
Showing 1 changed file with 31 additions and 23 deletions.
diff --git a/tests/unit/collection/test_lbstats_retention.cc b/tests/unit/collection/test_lbstats_retention.cc
@@ -74,29 +74,37 @@ struct TestCol : vt::Collection<TestCol,vt::Index1D> {
     auto sp_load_phase_count = stats.getSubphaseLoadPhaseCount();
     auto sp_comm_phase_count = stats.getSubphaseCommPhaseCount();
 
-    if (phase > phases_needed) {
-      // syncNextPhase will have caused timings to be added for the
-      // next phase already
-      EXPECT_EQ(load_phase_count, phases_needed + 1);
-      EXPECT_EQ(sp_load_phase_count, phases_needed + 1);
-
-      EXPECT_EQ(comm_phase_count, phases_needed);
-      EXPECT_EQ(sp_comm_phase_count, phases_needed);
-    } else if (phase == 0) {
-      EXPECT_EQ(load_phase_count, phase);
-      EXPECT_EQ(sp_load_phase_count, phase);
-
-      EXPECT_EQ(comm_phase_count, phase);
-      EXPECT_EQ(sp_comm_phase_count, phase);
-    } else {
-      // syncNextPhase will have caused timings to be added for the
-      // next phase already
-      EXPECT_EQ(load_phase_count, phase + 1);
-      EXPECT_EQ(sp_load_phase_count, phase + 1);
-
-      EXPECT_EQ(comm_phase_count, phase);
-      EXPECT_EQ(sp_comm_phase_count, phase);
-    }
+    #if vt_check_enabled(lblite)
+      if (phase > phases_needed) {
+        // syncNextPhase will have caused timings to be added for the
+        // next phase already
+        EXPECT_EQ(load_phase_count, phases_needed + 1);
+        EXPECT_EQ(sp_load_phase_count, phases_needed + 1);
+
+        EXPECT_EQ(comm_phase_count, phases_needed);
+        EXPECT_EQ(sp_comm_phase_count, phases_needed);
+      } else if (phase == 0) {
+        EXPECT_EQ(load_phase_count, phase);
+        EXPECT_EQ(sp_load_phase_count, phase);
+
+        EXPECT_EQ(comm_phase_count, phase);
+        EXPECT_EQ(sp_comm_phase_count, phase);
+      } else {
+        // syncNextPhase will have caused timings to be added for the
+        // next phase already
+        EXPECT_EQ(load_phase_count, phase + 1);
+        EXPECT_EQ(sp_load_phase_count, phase + 1);
+
+        EXPECT_EQ(comm_phase_count, phase);
+        EXPECT_EQ(sp_comm_phase_count, phase);
+      }
+    #else
+      EXPECT_EQ(load_phase_count, 0);
+      EXPECT_EQ(sp_load_phase_count, 0);
+
+      EXPECT_EQ(comm_phase_count, 0);
+      EXPECT_EQ(sp_comm_phase_count, 0);
+    #endif
   }
 };