fix codecov issue

PennyLaneAI · Dec 14, 2022 · 2ee7674 · 2ee7674
1 parent f281db8
commit 2ee7674
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/tests/math/test_multi_dispatch.py b/tests/math/test_multi_dispatch.py
@@ -173,11 +173,21 @@ def test_kron():
 @pytest.mark.parametrize(
     ("n", "t", "gamma_ref"),
     [
+        (
+            0.1,
+            jnp.array([0.2, 0.3, 0.4]),
+            jnp.array([0.87941963, 0.90835799, 0.92757383]),
+        ),
         (
             0.1,
             np.array([0.2, 0.3, 0.4]),
             np.array([0.87941963, 0.90835799, 0.92757383]),
         ),
+        (
+            0.1,
+            onp.array([0.2, 0.3, 0.4]),
+            onp.array([0.87941963, 0.90835799, 0.92757383]),
+        ),
     ],
 )
 def test_gammainc(n, t, gamma_ref):