add sum method to _sparse_py_operators

aesara/sparse/basic.py

@@ -299,6 +299,9 @@ class _sparse_py_operators:
     def __rdot__(right, left):
         return structured_dot(left, right)
 
+    def sum(self, axis=None, sparse_grad=False):
+        return sp_sum(self, axis=axis, sparse_grad=sparse_grad)
+
     dot = __dot__
 
     # N.B. THIS IS COMMENTED OUT ON PURPOSE!!!

tests/sparse/test_basic.py

@@ -2039,12 +2039,17 @@ class TestSpSum(utt.InferShapeTester):
         self.op_class = sparse.SpSum
         self.op = sparse.sp_sum
 
-    def test_op(self):
+    @pytest.mark.parametrize("op_type", ["func", "method"])
+    def test_op(self, op_type):
         for format in sparse.sparse_formats:
             for axis in self.possible_axis:
                 variable, data = sparse_random_inputs(format, shape=(10, 10))
 
-                z = sparse.sp_sum(variable[0], axis=axis)
+                if op_type == "func":
+                    z = sparse.sp_sum(variable[0], axis=axis)
+                if op_type == "method":
+                    z = variable[0].sum(axis=axis)
+
                 if axis is None:
                     assert z.type.broadcastable == ()
                 else: