def testBasicBasis(self): # this should work as it's in the examples lp = LP() lp.addConstraint( (0, 1), "<", 3) lp.addConstraint( (1, 1), "<", 3) lp.setMaximize() lp.setObjective([1,1]) lp.solve(guess = [3,3]) self.assert_(lp.getInfo("Iterations") == 0, lp.getInfo("Iterations"))
def testBasicBasis(self): # this should work as it's in the examples lp = LP() lp.addConstraint((0, 1), "<", 3) lp.addConstraint((1, 1), "<", 3) lp.setMaximize() lp.setObjective([1, 1]) lp.solve(guess=[3, 3]) self.assert_(lp.getInfo("Iterations") == 0, lp.getInfo("Iterations"))
def checkUB(self, opts, ub): # these are indices to bound indices = {} indices["t"] = (0,3) indices["N"] = "a" indices["l"] = [0,1,2] indices["a"] = ar([0,1,2],dtype=uint) indices["f"] = ar([0,1,2],dtype=float64) ubvalues = {} ubvalues["s"] = ub ubvalues["l"] = [ub, ub, ub] ubvalues["a"] = ar([ub, ub, ub]) lp = LP() if opts[0] == "N": lp.getIndexBlock(indices["N"], 3) lp.setObjective([1,1,1,1,1,1]) lp.addConstraint( ((3,6), [[1,0,0],[0,1,0],[0,0,1]]), "<=", 10) lp.setMaximize() lp.setLowerBound(indices[opts[0]], None) lp.setUpperBound(indices[opts[0]], ubvalues[opts[1]]) for num_times in range(2): # make sure it's same anser second time lp.solve() self.assertAlmostEqual(lp.getObjectiveValue(), ub*3 + 10 * 3) v = lp.getSolution() self.assert_(len(v) == 6) self.assertAlmostEqual(v[0], ub) self.assertAlmostEqual(v[1], ub) self.assertAlmostEqual(v[2], ub) self.assertAlmostEqual(v[3], 10) self.assertAlmostEqual(v[4], 10) self.assertAlmostEqual(v[5], 10)
def checkUB(self, opts, ub): # these are indices to bound indices = {} indices["t"] = (0, 3) indices["N"] = "a" indices["l"] = [0, 1, 2] indices["a"] = ar([0, 1, 2], dtype=uint) indices["f"] = ar([0, 1, 2], dtype=float64) ubvalues = {} ubvalues["s"] = ub ubvalues["l"] = [ub, ub, ub] ubvalues["a"] = ar([ub, ub, ub]) lp = LP() if opts[0] == "N": lp.getIndexBlock(indices["N"], 3) lp.setObjective([1, 1, 1, 1, 1, 1]) lp.addConstraint(((3, 6), [[1, 0, 0], [0, 1, 0], [0, 0, 1]]), "<=", 10) lp.setMaximize() lp.setLowerBound(indices[opts[0]], None) lp.setUpperBound(indices[opts[0]], ubvalues[opts[1]]) for num_times in range(2): # make sure it's same anser second time lp.solve() self.assertAlmostEqual(lp.getObjectiveValue(), ub * 3 + 10 * 3) v = lp.getSolution() self.assert_(len(v) == 6) self.assertAlmostEqual(v[0], ub) self.assertAlmostEqual(v[1], ub) self.assertAlmostEqual(v[2], ub) self.assertAlmostEqual(v[3], 10) self.assertAlmostEqual(v[4], 10) self.assertAlmostEqual(v[5], 10)