Add xfail test with psi inner function

Author: ricardoV94

tests/link/numba/test_elemwise.py

@@ -606,6 +606,9 @@ def test_scalar_loop():
     )
 
 
+@pytest.mark.xfail(
+    reason="Numba fails due to https://github.com/numba/numba/issues/10098"
+)
 def test_gammainc_wrt_k_grad():
     x = pt.vector("x", dtype="float64")
     k = pt.vector("k", dtype="float64")

tests/link/numba/test_scalar.py

@@ -245,3 +245,15 @@ def test_scalar_while_loop(self):
         np.testing.assert_allclose(fn(n_steps=20, x0=1), [10, True])
         np.testing.assert_allclose(fn(n_steps=5, x0=1), [6, False])
         np.testing.assert_allclose(fn(n_steps=0, x0=1), [1, False])
+
+    @pytest.mark.xfail("Fails due to https://github.com/numba/numba/issues/10098")
+    def test_loop_with_cython_wrapped_op(self):
+        x = ps.float64("x")
+        op = ScalarLoop(init=[x], update=[ps.psi(x)])
+        out = op(1, x)
+
+        fn = function([x], out, mode=numba_mode)
+        x_test = np.float64(0.5)
+        res = fn(x_test)
+        expected_res = ps.psi(x).eval({x: x_test})
+        np.testing.assert_allclose(res, expected_res)