Update test_parfor.cpp: putting in times

kokkos · Apr 6, 2024 · f97a3b4 · f97a3b4
1 parent 4dea629
commit f97a3b4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/tests/sampler/test_parfor.cpp b/tests/sampler/test_parfor.cpp
@@ -8,6 +8,8 @@
 
 using ::testing::HasSubstr;
 using ::testing::Not;
+using ::testing::Contains;
+using ::testing::Times;
 
 struct Tester {
   template <typename execution_space>
@@ -67,6 +69,12 @@ TEST(SamplerTest, ktoEnvVarDefault) {
     EXPECT_THAT(output.str(), HasSubstr(matcher));
   }  // end TEST
 
+  EXPECT_THAT(output.str(), ::testing::Contains.Times(static_cast<int>(2), "calling child-begin function..."));
+  EXPECT_THAT(output.str(), ::testing::Contains.Times(static_cast<int>(2), "finished with child-begin function."));
+
+  EXPECT_THAT(output.str(), ::testing::Contains.Times(static_cast<int>(2), "calling child-end function..."));
+  EXPECT_THAT(output.str(), ::testing::Contains.Times(static_cast<int>(2), "finished with child-end function."));
+
   EXPECT_THAT(output.str(), Not(HasSubstr("KokkosP: FATAL: No child library of "
                                           "sampler utility library to call")));