def test_variables_management(self):
builder = MPModelBuilder()
x = builder.add_binary_variable('x')
self.assert_equal('x', x.get_name())
y = builder.add_binary_variable('y')
self.assert_equal('y', y.get_name())
z = builder.add_binary_variable('z')
self.assert_equal('z', z.get_name())
self.assert_equal(3, builder.get_num_variables())
self.assert_equal([x, y, z], builder.get_variables())
def test_build_simple_2x2_model(self):
builder = MPModelBuilder()
x1, x2 = builder.x1(), builder.x2()
builder.set_objective(2 * x1 + 3 * x2)
builder.set_constraints([4 * x1 + 6 * x2 <= 5,
2 * x1 + 2 * x2 <= 4])
for constraint in builder.get_constraints():
self.assert_equal(set([id(var) for var in constraint.get_variables()]),
set([id(x1), id(x2)]))
self.assert_equal(2, builder.get_num_variables())
self.assert_equal(2, builder.get_num_constraints())
for var in builder.get_variables():
var.set_type(MPVariable.BINARY)
self.assert_true(builder.is_binary())
