Try different test (may revert)

Author: mcbarton

test/test_xcpp_kernel.py

@@ -232,33 +232,26 @@ class BaseXCppOpenMPTests(jupyter_kernel_test.KernelTests):
         # language_info.name in a kernel_info_reply should match this
         language_name = 'C++'
 
-        # Code that should write the exact string `hello, world` to STDOUT
-        code_omp="""
+        # Code that prints Hello from thread followed by thread number
+        omp_test_code = """
         #include <omp.h>
         #include <iostream>
         #include <clang/Interpreter/CppInterOp.h>
         Cpp::LoadLibrary("libomp");
-        int main() {
-            int max_threads = omp_get_max_threads();
-
-            printf("max threads: %d\n", max_threads);
-            omp_set_num_threads(max_threads);
-
-            #pragma omp parallel
-            {
-                int id = omp_get_thread_num();
-                printf("Hello World from thread = %d with %d threads\n", id, omp_get_num_threads());
-            }
-
-            printf("all done, with hopefully %d threads\n", max_threads);
+        omp_set_num_threads(4);
+        #pragma omp parallel
+        {
+            int id = omp_get_thread_num();
+            std::cout << "Hello from thread " << id << std::endl;
         }
-        main();
-        """
-    
+        """    
         def test_xcpp_omp(self):
-                self.flush_channels()
-                reply, output_msgs = self.execute_helper(code=self.code_omp,timeout=20)
-                self.assertEqual(output_msgs[0]['msg_type'], 'stream')
+            self.flush_channels()
+            reply, output_msgs = self.execute_helper(code=self.omp_test_code, timeout=20)
+            output_text = ''.join(msg['content']['text'] for msg in stream_msgs)
+            thread_ids = set(map(int, re.findall(r'Hello from thread (\d+)', output_text)))
+            expected_ids = set(range(4))
+            self.assertEqual(thread_ids, expected_ids,f"Expected thread IDs {expected_ids}, but got {thread_ids}")
 
     kernel_names = ['xcpp17-omp', 'xcpp20-omp', 'xcpp23-omp']