Fix calling OpenMP runtime functions in interactive sessions

- Remove unnecessary inspect to check the caller context; __call__
method is only need by python mode
- Add tests for calling OpenMP runtime functions within jit and python
contexts

Author: ggeorgakoudis

src/numba/openmp/omp_runtime.py

@@ -6,15 +6,6 @@
 
 class _OpenmpExternalFunction(types.ExternalFunction):
     def __call__(self, *args):
-        import inspect
-
-        frm = inspect.stack()[1]
-        mod = inspect.getmodule(frm[0])
-        if mod.__name__.startswith("numba") and not mod.__name__.startswith(
-            "numba.openmp.tests"
-        ):
-            return super(ExternalFunction, self).__call__(*args)
-
         # Resolve the function address via llvmlite's symbol table so we
         # call the same LLVM-registered symbol the JIT uses. Then wrap
         # it with ctypes CFUNCTYPE to call from Python. This avoids

src/numba/openmp/tests/test_openmp.py

@@ -5051,6 +5051,27 @@ def test_impl(lb, num_steps, pi_comp_func):
         njit_output = njit(test_impl)(0, 1024, test_pi_comp_njit)
         self.assert_outputs_equal(py_output, njit_output)
 
+class TestOpenmpRuntimeFunctions(TestOpenmpBase):
+    def __init__(self, *args):
+        TestOpenmpBase.__init__(self, *args)
+
+    def test_omp_get_max_threads(self):
+        @njit
+        def test_impl():
+            return omp_get_max_threads()
+
+        jit_max_threads = test_impl()
+        python_max_threads = omp_get_max_threads()
+        self.assertEqual(jit_max_threads, python_max_threads)
+
+    def test_omp_get_num_procs(self):
+        @njit
+        def test_impl():
+            return omp_get_num_procs()
+
+        jit_num_procs = test_impl()
+        python_num_procs = omp_get_num_procs()
+        self.assertEqual(jit_num_procs, python_num_procs)
 
 if __name__ == "__main__":
     unittest.main()