small clean up, test more corner case and pep8.

theano/tensor/tests/test_opt.py

@@ -3473,7 +3473,7 @@ class T_local_sum(unittest.TestCase):
 
     def test_local_sum_all_to_none(self):
         a = T.tensor3()
-        input = numpy.arange(3 * 3 * 3, dtype=config.floatX).reshape(3, 3, 3)
+        input = numpy.arange(3 * 4 * 5, dtype=config.floatX).reshape(3, 4, 5)
         f = theano.function([a], a.sum(), mode=self.mode)
         assert len(f.maker.fgraph.apply_nodes) == 1
         assert numpy.allclose(f(input), input.sum())
@@ -3527,7 +3527,6 @@ class T_local_sum(unittest.TestCase):
                 f = theano.function([a], a.sum(d).sum(None), mode=self.mode)
                 assert numpy.allclose(f(input), input.sum(d).sum())
                 assert len(f.maker.fgraph.apply_nodes) == 1
-            for d in [0, 1, 2]:
             f = theano.function([a], a.sum(None).sum(), mode=self.mode)
             assert numpy.allclose(f(input), input.sum())
             assert len(f.maker.fgraph.apply_nodes) == 1