def test_typedvariables(self): cqm = CQM() x = Binary('x') s = Spin('s') i = Integer('i') cqm.set_objective(x + i) cqm.add_constraint(i + s <= 1) with self.assertWarns(DeprecationWarning): self.assertEqual(cqm.variables.lower_bounds, {'x': 0.0, 'i': 0.0, 's': -1.0}) with self.assertWarns(DeprecationWarning): self.assertEqual(cqm.variables.upper_bounds, {'x': 1.0, 'i': 9007199254740991.0, 's': 1}) with self.assertWarns(DeprecationWarning): self.assertEqual(len(cqm.variables.lower_bounds), 3) with self.assertWarns(DeprecationWarning): self.assertEqual(len(cqm.variables.upper_bounds), 3) with self.assertWarns(DeprecationWarning): self.assertEqual(list(cqm.variables.vartypes), [dimod.BINARY, dimod.INTEGER, dimod.SPIN]) with self.assertWarns(DeprecationWarning): self.assertIs(cqm.variables.vartype('x'), dimod.BINARY)
def test_terms(self): cqm = CQM() a = cqm.add_variable('a', 'BINARY') b = cqm.add_variable('b', 'BINARY') c = cqm.add_variable('c', 'INTEGER') cqm.add_constraint([(a, b, 1), (b, 2.5,), (3,), (c, 1.5)], sense='<=')
def test_symbolic(self): cqm = CQM() a = Spin('a') b = Spin('b') c = Spin('c') cqm.add_constraint(2*a*b + b*c - c + 1 <= 1)
def test_duplicate(self): cqm = CQM() bqm = BQM({'a': -1}, {'ab': 1}, 1.5, 'SPIN') cqm.add_constraint(bqm <= 5, label='hello') with self.assertRaises(ValueError): cqm.add_constraint(bqm <= 5, label='hello')
def test_bqm_equality_constraint_offset(self): x, y, z = dimod.Binaries('xyz') cqm = CQM() cqm.add_constraint(x + y + z - 1 == 2) bqm, inverter = dimod.cqm_to_bqm(cqm, lagrange_multiplier=1) self.assertEqual(bqm, (x + y + z - 3)*(x + y + z - 3))
def test_later_defn(self): i0 = Integer('i') i1 = Integer('i', upper_bound=1) cqm = CQM() cqm.add_variable('i', 'INTEGER') cqm.set_objective(i0) with self.assertRaises(ValueError): cqm.add_constraint(i1 <= 1) cqm.add_variable('i', 'INTEGER')
def test_symbolic_mixed(self): cqm = CQM() x = Binary('x') s = Spin('s') i = Integer('i') cqm.add_constraint(2*i + s + x <= 2) self.assertIs(cqm.vartype('x'), dimod.BINARY) self.assertIs(cqm.vartype('s'), dimod.SPIN) self.assertIs(cqm.vartype('i'), dimod.INTEGER)
def test_qm_Ge_constraint(self): i = dimod.Integer('i', upper_bound=7) j = dimod.Integer('j', upper_bound=9) x = dimod.Binary('x') cqm = CQM() cqm.add_constraint(i + j + x >= 5) bqm, inverter = dimod.cqm_to_bqm(cqm, lagrange_multiplier=1) for bin_sample in dimod.ExactSolver().sample(bqm).lowest().samples(): int_sample = inverter(bin_sample) self.assertGreaterEqual(int_sample['i'] + int_sample['j'] + int_sample['x'], 5)
def test_bqm_Ge_constraint(self): x, y, z = dimod.Binaries('xyz') cqm = CQM() cqm.add_constraint(x + y + z >= 2) bqm, inverter = dimod.cqm_to_bqm(cqm, lagrange_multiplier=1) configs = set() for sample in dimod.ExactSolver().sample(bqm).lowest().samples(): self.assertGreaterEqual(sample['x'] + sample['y'] + sample['z'], 2) configs.add((sample['x'], sample['y'], sample['z'])) self.assertEqual(len(configs), 4)
def test_integer(self): i = dimod.Integer('i', lower_bound=-1000) j, k = dimod.Integers('jk') cqm = CQM() label_le = cqm.add_constraint(i + j*k <= 5) label_ge = cqm.add_constraint(i + j >= 1000) sample = {'i': 105, 'j': 4, 'k': 5} self.assertEqual(cqm.violations(sample), {label_le: 120.0, label_ge: 891.0}) sample = {'j': -1, 'i': 1004, 'k': 1000} self.assertEqual(cqm.violations(sample, clip=False), {label_ge: -3.0, label_le: -1.0})
def test_inconsistent(self): i0 = Integer('i') i1 = Integer('i', upper_bound=1) i2 = Integer('i', lower_bound=-2) cqm = CQM() cqm.set_objective(i0) with self.assertRaises(ValueError): cqm.add_constraint(i1 <= 1) cqm = CQM() cqm.add_constraint(i0 <= 1) with self.assertRaises(ValueError): cqm.set_objective(i2)
def test_serializable(self): i = dimod.Integer('i', upper_bound=7) j = dimod.Integer('j', upper_bound=9) x = dimod.Binary('x') cqm = CQM() cqm.add_constraint(i + j + x >= 5) bqm, inverter = dimod.cqm_to_bqm(cqm, lagrange_multiplier=1) newinverter = dimod.constrained.CQMToBQMInverter.from_dict( json.loads(json.dumps(inverter.to_dict()))) for bin_sample in dimod.ExactSolver().sample(bqm).lowest().samples(): int_sample = newinverter(bin_sample) self.assertGreaterEqual(int_sample['i'] + int_sample['j'] + int_sample['x'], 5)
def test_typical(self): i, j, k = dimod.Integers('ijk') cqm = CQM() cqm.set_objective(2 * i * i + i * k) label = cqm.add_constraint(-3 * i * i + 4 * j * j <= 5) mapping = cqm.substitute_self_loops() self.assertIn('i', mapping) self.assertIn('j', mapping) self.assertEqual(len(mapping), 2) self.assertEqual(cqm.objective.quadratic, { ('k', 'i'): 1, (mapping['i'], 'i'): 2 }) self.assertEqual(cqm.constraints[label].lhs.quadratic, { (mapping['i'], 'i'): -3.0, (mapping['j'], 'j'): 4.0 }) self.assertEqual(len(cqm.constraints), 3) for v, new in mapping.items(): self.assertIn(new, cqm.constraints) self.assertEqual(cqm.constraints[new].sense, Sense.Eq) self.assertEqual(cqm.constraints[new].lhs.linear, {v: 1, new: -1}) self.assertEqual(cqm.constraints[new].rhs, 0)
def test_functional(self): cqm = CQM() bqm = BQM({'a': -1}, {'ab': 1}, 1.5, 'SPIN') cqm.add_constraint(bqm, '<=') cqm.add_constraint(bqm, '>=') cqm.set_objective(BQM({'c': -1}, {}, 'SPIN')) cqm.add_constraint(Spin('a')*Integer('d')*5 <= 3) new = CQM.from_file(cqm.to_file()) self.assertTrue(cqm.objective.is_equal(new.objective)) self.assertEqual(set(cqm.constraints), set(new.constraints)) for label, constraint in cqm.constraints.items(): self.assertTrue(constraint.lhs.is_equal(new.constraints[label].lhs)) self.assertEqual(constraint.rhs, new.constraints[label].rhs) self.assertEqual(constraint.sense, new.constraints[label].sense)
def test_construction(self): cqm = CQM() cqm.set_objective(dimod.AdjVectorBQM({'ab': 1}, 'SPIN')) label = cqm.add_constraint(dimod.AdjVectorBQM({'ab': 1}, 'SPIN'), sense='==', rhs=1) self.assertIsInstance(cqm.objective, BQM) self.assertIsInstance(cqm.constraints[label].lhs, BQM)
def test_terms_integer_bounds(self): # bug report: https://github.com/dwavesystems/dimod/issues/943 cqm = CQM() i = Integer('i', lower_bound=-1, upper_bound=5) cqm.set_objective(i) label = cqm.add_constraint([('i', 1)], sense='<=') # failing in #943 self.assertEqual(cqm.constraints[label].lhs.lower_bound('i'), -1) self.assertEqual(cqm.constraints[label].lhs.upper_bound('i'), 5)
def test_header(self): from dimod.serialization.fileview import read_header cqm = CQM() x = Binary('x') s = Spin('s') i = Integer('i') cqm.set_objective(x + 3*i + s*x) cqm.add_constraint(x*s + x <= 5) cqm.add_constraint(i*i + i*s <= 4) header_info = read_header(cqm.to_file(), b'DIMODCQM') self.assertEqual(header_info.data, {'num_biases': 11, 'num_constraints': 2, 'num_quadratic_variables': 4, 'num_variables': 3})
def test_construction(self): cqm = CQM() with self.assertWarns(DeprecationWarning): bqm = dimod.AdjVectorBQM({'ab': 1}, 'SPIN') cqm.set_objective(bqm) label = cqm.add_constraint(bqm, sense='==', rhs=1) self.assertIsInstance(cqm.objective, dimod.QuadraticModel) self.assertIsInstance(cqm.constraints[label].lhs, BQM)
def test_deepcopy(self): from copy import deepcopy cqm = CQM() x = Binary('x') s = Spin('s') i = Integer('i') cqm.set_objective(i + s + x) constraint = cqm.add_constraint(i + s + x <= 1) new = deepcopy(cqm) self.assertTrue(new.objective.is_equal(cqm.objective)) self.assertTrue(new.constraints[constraint].lhs.is_equal(cqm.constraints[constraint].lhs))
def test_binary(self): a, b, c = dimod.Binaries('abc') cqm = CQM() cqm.add_constraint(a + b + c == 1, label='onehot') cqm.add_constraint(a*b <= 0, label='ab LE') cqm.add_constraint(c >= 1, label='c GE') sample = {'a': 0, 'b': 0, 'c': 1} # satisfying sample self.assertEqual(cqm.violations(sample), {'ab LE': 0.0, 'c GE': 0.0, 'onehot': 0.0}) self.assertEqual(cqm.violations(sample, skip_satisfied=True), {}) sample = {'a': 1, 'b': 0, 'c': 0} # violates one self.assertEqual(cqm.violations(sample), {'ab LE': 0.0, 'c GE': 1.0, 'onehot': 0.0}) self.assertEqual(cqm.violations(sample, skip_satisfied=True), {'c GE': 1.0})
def test_bqm(self): cqm = CQM() bqm = BQM({'a': -1}, {'ab': 1}, 1.5, 'SPIN') cqm.add_constraint(bqm, '<=') cqm.add_constraint(bqm, '>=') # add it again