add tests

theano/tensor/tests/test_elemwise.py

@@ -98,6 +98,43 @@ class test_DimShuffle(unittest_tools.InferShapeTester):
         y = x.dimshuffle(('x',) * (numpy.MAXDIMS + 1))
         self.assertRaises(ValueError, y.eval, {x: 0})
 
+class test_reduce_axes(unittest.TestCase):
+
+    def test_sum_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.sum(a)
+
+    def test_mean_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.mean(a)
+
+    def test_max_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.max(a)
+
+    def test_min_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.min(a)
+
+    def test_argmax_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.argmax(a)
+
+    def test_var_axes(self):
+        axes = [None, 0, 1, [0, 1], numpy.array(1), [numpy.array(0), numpy.array(1)]]
+        for a in axes:
+            x = tensor.matrix()
+            m = x.var(a)
 
 class test_Broadcast(unittest.TestCase):
     def setUp(self):