Fix python unit tests

Author: nedtaylor

CMakeLists.txt

@@ -365,6 +365,13 @@ if (BUILD_PYTHON)
         COMMENT "Copying raffle python files"
     )
 
+    # If parallel build, need to add the parallel flag
+    if (CMAKE_BUILD_TYPE MATCHES "Release*" OR CMAKE_BUILD_TYPE MATCHES "Parallel*" OR CMAKE_BUILD_TYPE MATCHES "Debug")
+        set(GOMPFLAGS "-lgomp")
+    else()
+        set(GOMPFLAGS "")
+    endif()
+
     # Create a Python module using f2py
     add_custom_command(
         OUTPUT ${F2PY_OUTPUT_FILE}
@@ -373,6 +380,7 @@ if (BUILD_PYTHON)
         -m _${PROJECT_NAME}
         -I${MODULE_DIR}
         --f90flags="${PPFLAGS}"
+        ${GOMPFLAGS}
         --backend meson
         ${F90WRAP_FILE}
         -L${CMAKE_BUILD_DIR}

test/test_raffle.py

@@ -30,21 +30,36 @@
         print(line)
         test_name = line.split()[0]
         test_names.append(test_name)
+print("Test names: ", test_names)
+
+def uses_openmp(lib_path):
+    try:
+        out = subprocess.check_output(["nm", "-g", lib_path], text=True)
+        return "_GOMP_parallel" in out
+    except Exception:
+        return False
 
 def build_fortran_test(test_name):
     os.makedirs(temp_test_dir, exist_ok=True)
     # make using a fortran compiler
     # point to raffle's pip install include and lib to include during compilation
 
-    build_result = subprocess.run(
-        [
+    if uses_openmp(os.path.join(lib_dir, "libraffle.dylib")):
+        compile_args = ["-fopenmp", "-lgomp"]
+    else:
+        compile_args = []
+
+    compile_list = [
             fc, "-o", test_name+".o",
             "../../test/test_io_utils.f90",
+            " ".join(compile_args),
             "-I", include_dir,
             "-I", etc_dir,
             "-L", lib_dir,
             "-lraffle"
-        ],
+    ]
+    build_result = subprocess.run(
+        compile_list,
         cwd=temp_test_dir
     )
     return build_result
@@ -271,4 +286,4 @@ def test_fortran(self, test_name):
         self.assertEqual(run_result.returncode, 0)
 
 if __name__ == '__main__':
-    unittest.main()
+    unittest.main()