map_variables: satisfy flake8

theano/scan_module/tests/test_scan_utils.py

@@ -84,7 +84,7 @@ class TestMapVariables(object):
 
         f = theano.function([x, outer], [t, t2])
         rval = f(x=numpy.array([1, 2, 3]), outer=0.5)
-        assert numpy.array_equal(rval, [[ 1,  3,  6], [-1, -3, -6]])
+        assert numpy.array_equal(rval, [[1, 3, 6], [-1, -3, -6]])
 
     def test_leaf_inside_scan(self):
         x = tensor.vector('x')
@@ -97,7 +97,5 @@ class TestMapVariables(object):
         s2, = map_variables(self.replacer, [s])
 
         f = theano.function([x, y, z], [s, s2])
-        assert numpy.array_equal(
-            f(x=numpy.array([1, 2, 3]), y=1, z=2),
-            [[ 1, 2, 3],
-             [ 2, 4, 6]])
+        rval = f(x=numpy.array([1, 2, 3]), y=1, z=2)
+        assert numpy.array_equal(rval, [[1, 2, 3], [2, 4, 6]])