flake8 for theano/gof/tests/test_sched.py

theano/gof/tests/test_sched.py

 from theano.gof.sched import (make_dependence_cmp, sort_apply_nodes,
                               reverse_dict, _toposort, posort)
 
-import theano
 from theano import tensor
 from theano.gof.graph import io_toposort
 from theano.compat import cmp
-from six.moves import xrange
 
 
 def test_dependence():
@@ -22,7 +20,10 @@ def test_dependence():
 def test_sort_apply_nodes():
     x = tensor.matrix('x')
     y = tensor.dot(x * 2, x + 1)
-    str_cmp = lambda a, b: cmp(str(a), str(b))  # lexicographical sort
+
+    def str_cmp(a, b):
+        return cmp(str(a), str(b))  # lexicographical sort
+
     nodes = sort_apply_nodes([x], [y], cmps=[str_cmp])
 
     for a, b in zip(nodes[:-1], nodes[1:]):
@@ -64,5 +65,5 @@ def test_posort():
     cmps = [lambda a, b: a % 10 - b % 10,
             lambda a, b: (a / 10) % 2 - (b / 10) % 2,
             lambda a, b: a - b]
-    assert posort(l, *cmps) == \
-            [10, 1, 11, 2, 12, 3, 13, 4, 14, 5, 15, 6, 16, 7, 17, 8, 18, 9, 19]
+    assert (posort(l, *cmps) ==
+            [10, 1, 11, 2, 12, 3, 13, 4, 14, 5, 15, 6, 16, 7, 17, 8, 18, 9, 19])