black

Author: emma58

pyomo/contrib/preprocessing/plugins/var_aggregator.py

@@ -13,8 +13,14 @@
 
 
 from pyomo.common.collections import ComponentMap, ComponentSet
-from pyomo.core.base import (Block, Constraint, VarList, Objective, Reals,
-                             TransformationFactory)
+from pyomo.core.base import (
+    Block,
+    Constraint,
+    VarList,
+    Objective,
+    Reals,
+    TransformationFactory,
+)
 from pyomo.core.expr import ExpressionReplacementVisitor
 from pyomo.core.expr.numvalue import value
 from pyomo.core.plugins.transform.hierarchy import IsomorphicTransformation

pyomo/contrib/preprocessing/tests/test_var_aggregator.py

@@ -219,38 +219,31 @@ def test_binary_inequality(self):
         m.x = Var(domain=Binary)
         m.y = Var(domain=Binary)
         m.c = Constraint(expr=m.x == m.y)
-        m.o = Objective(expr=0.5*m.x + m.y, sense=maximize)
+        m.o = Objective(expr=0.5 * m.x + m.y, sense=maximize)
         TransformationFactory('contrib.aggregate_vars').apply_to(m)
         var_to_z = m._var_aggregator_info.var_to_z
         z = var_to_z[m.x]
         self.assertIs(var_to_z[m.y], z)
         self.assertEqual(z.domain, Binary)
         self.assertEqual(z.lb, 0)
         self.assertEqual(z.ub, 1)
-        assertExpressionsEqual(
-            self,
-            m.o.expr,
-            0.5 * z + z
-        )
+        assertExpressionsEqual(self, m.o.expr, 0.5 * z + z)
 
     def test_equality_different_domains(self):
         m = ConcreteModel()
         m.x = Var(domain=Reals, bounds=(1, 2))
         m.y = Var(domain=Binary)
         m.c = Constraint(expr=m.x == m.y)
-        m.o = Objective(expr=0.5*m.x + m.y, sense=maximize)
+        m.o = Objective(expr=0.5 * m.x + m.y, sense=maximize)
         TransformationFactory('contrib.aggregate_vars').apply_to(m)
         var_to_z = m._var_aggregator_info.var_to_z
         z = var_to_z[m.x]
         self.assertIs(var_to_z[m.y], z)
         self.assertEqual(z.lb, 1)
         self.assertEqual(z.ub, 1)
         self.assertEqual(z.domain, Binary)
-        assertExpressionsEqual(
-            self,
-            m.o.expr,
-            0.5 * z + z
-        )
+        assertExpressionsEqual(self, m.o.expr, 0.5 * z + z)
+
 
 if __name__ == '__main__':
     unittest.main()