Remove Numba DiffOp dispatch test

tests/link/test_numba.py

@@ -1514,50 +1514,6 @@ def test_CumOp(val, axis, mode):
     )
 
 
-@pytest.mark.parametrize(
-    "val, n, axis",
-    [
-        (
-            set_test_value(at.matrix(), rng.normal(size=(3, 2)).astype(config.floatX)),
-            0,
-            0,
-        ),
-        (
-            set_test_value(at.matrix(), rng.normal(size=(3, 2)).astype(config.floatX)),
-            0,
-            1,
-        ),
-        (
-            set_test_value(at.matrix(), rng.normal(size=(3, 2)).astype(config.floatX)),
-            1,
-            0,
-        ),
-        (
-            set_test_value(at.matrix(), rng.normal(size=(3, 2)).astype(config.floatX)),
-            1,
-            1,
-        ),
-        (
-            set_test_value(at.lmatrix(), rng.poisson(size=(3, 2))),
-            0,
-            0,
-        ),
-    ],
-)
-def test_DiffOp(val, axis, n):
-    g = extra_ops.DiffOp(n=n, axis=axis)(val)
-    g_fg = FunctionGraph(outputs=[g])
-
-    (res,) = compare_numba_and_py(
-        g_fg,
-        [
-            i.tag.test_value
-            for i in g_fg.inputs
-            if not isinstance(i, (SharedVariable, Constant))
-        ],
-    )
-
-
 @pytest.mark.parametrize(
     "a, val",
     [

tests/tensor/test_extra_ops.py

@@ -8,7 +8,6 @@ from aesara import function
 from aesara import tensor as at
 from aesara.compile.mode import Mode
 from aesara.configdefaults import config
-from aesara.gradient import grad
 from aesara.graph.basic import applys_between
 from aesara.graph.optdb import OptimizationQuery
 from aesara.raise_op import Assert
@@ -298,8 +297,8 @@ class TestBinCount(utt.InferShapeTester):
                     f5(a)
 
 
-class TestDiffOp(utt.InferShapeTester):
-    def test_diffOp(self):
+class TestDiff(utt.InferShapeTester):
+    def test_diff(self):
         x = matrix("x")
         a = np.random.random((30, 50)).astype(config.floatX)
 
@@ -333,11 +332,6 @@ class TestDiffOp(utt.InferShapeTester):
                     else:
                         assert out.type.shape[i] == out_test.shape[i]
 
-    def test_grad_not_implemented(self):
-        x = at.matrix("x")
-        with pytest.raises(NotImplementedError):
-            grad(diff(x).sum(), x)
-
 
 class TestSqueeze(utt.InferShapeTester):
     shape_list = [(1, 3), (1, 2, 3), (1, 5, 1, 1, 6)]