def test_add_single_constraint(self):
model = ConcreteModel()
model.X = Var(within=Binary)
opt = SolverFactory("cplex", solver_io="python")
opt._set_instance(model)
self.assertEqual(opt._solver_model.linear_constraints.get_num(), 0)
model.C = Constraint(expr=model.X == 1)
con_interface = opt._solver_model.linear_constraints
with unittest.mock.patch.object(
con_interface, "add",
wraps=con_interface.add) as wrapped_add_call:
opt._add_constraint(model.C)
self.assertEqual(wrapped_add_call.call_count, 1)
self.assertEqual(
wrapped_add_call.call_args,
({
"lin_expr": [[[0], (1, )]],
"names": ["x2"],
"range_values": [0.0],
"rhs": [1.0],
"senses": ["E"],
}, ),
)
self.assertEqual(opt._solver_model.linear_constraints.get_num(), 1)
