Update test_parreduce.cpp: put in times

kokkos · Apr 6, 2024 · 69c694a · 69c694a
1 parent 336e304
commit 69c694a
Showing 1 changed file with 17 additions and 0 deletions.
diff --git a/tests/sampler/test_parreduce.cpp b/tests/sampler/test_parreduce.cpp
@@ -8,6 +8,9 @@
 
 using ::testing::HasSubstr;
 using ::testing::Not;
+using ::testing::Contains;
+using ::testing::Times;
+
 struct Tester {
   template <typename execution_space>
   explicit Tester(const execution_space& space) {
@@ -69,6 +72,20 @@ TEST(SamplerTest, ktoEnvVarDefault) {
     EXPECT_THAT(output.str(), HasSubstr(matcher));
   }
 
+ 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")));