fix: flake8 is a cruel mistress

cobra/test/test_compartment.py

@@ -3,31 +3,30 @@
 from __future__ import absolute_import
 
 import pytest
-from cobra.core import Metabolite, Model, Reaction, Compartment
-
+from cobra.core import Compartment
 
 def test__eq__():
     a = b = Compartment("x")
     c = "x"
     d = Compartment("x")
-    e = Compartment ("z")
-    assert a.__eq__(b) == True
-    assert a.__eq__(c) == True
-    assert a.__eq__(d) == True
-    assert a.__eq__(e) == False
+    e = Compartment("z")
+    assert a.__eq__(b) is True
+    assert a.__eq__(c) is True
+    assert a.__eq__(d) is True
+    assert a.__eq__(e) is False
 
 
 def test__ne__():
     a = b = Compartment("x")
     c = "x"
     d = Compartment("x")
-    e = Compartment ("z")
-    assert a.__ne__(b) == False
-    assert a.__ne__(c) == False
-    assert a.__ne__(d) == False
-    assert a.__ne__(e) == True
+    e = Compartment("z")
+    assert a.__ne__(b) is False
+    assert a.__ne__(c) is False
+    assert a.__ne__(d) is False
+    assert a.__ne__(e) is True
 
 
 def test_error_with_non_sane_id():
     with pytest.raises(TypeError):
-        Compartment("Trust Me This ID Is Sane")
+        Compartment("Trust Me This ID Is Sane")

cobra/test/test_metabolite.py

@@ -3,14 +3,15 @@
 from __future__ import absolute_import
 
 import pytest
-from cobra.core import Metabolite, Model, Reaction, Compartment
+from cobra.core import Metabolite, Model, Compartment
+
 
 def test_compartment_setter():
     met = Metabolite('x', compartment=None)
     cytosol = Compartment('c')
     model = Model()
     model.add_compartments([cytosol])
-    met2 =Metabolite("y")
+    met2 = Metabolite("y")
     met5 = Metabolite("b")
     model.add_metabolites([met2, met5])
     met2.compartment = Compartment("c")
@@ -21,7 +22,7 @@ def test_compartment_setter():
     assert met.compartment is None
     assert met2.compartment is cytosol
     assert met3.compartment is cytosol
-    assert isinstance(met4.compartment,Compartment)
+    assert isinstance(met4.compartment, Compartment)
     assert met5.compartment is cytosol
     with pytest.raises(TypeError):
-        Metabolite("c", compartment="Sane Compartment, Not!")
+        Metabolite("c", compartment="Sane Compartment, Not!")