def test_local_solver_sa_ising():
model = LogicalModel(mtype="ising")
s = model.variables("s", shape=(2,))
model.add_interaction(s[0], coefficient=1.0)
model.add_interaction(s[1], coefficient=2.0)
model.add_interaction((s[0], s[1]), coefficient=-3.0)
model.offset(10.0)
solver = LocalSolver()
sampleset1 = solver.solve(model.to_physical(), seed=12345)
assert sampleset1.variables == ["s[0]", "s[1]"]
assert len(sampleset1.record) == 1
# Check the ground state
assert np.array_equal(sampleset1.record[0].sample, [-1, 1])
assert sampleset1.record[0].energy == 6.0
assert sampleset1.record[0].num_occurrences == 1
# Check the second solve with the same solver instance
sampleset2 = solver.solve(model.to_physical(), seed=54321)
assert sampleset2.variables == ["s[0]", "s[1]"]
assert len(sampleset2.record) == 1
# Check the ground state
assert np.array_equal(sampleset2.record[0].sample, [-1, 1])
assert sampleset2.record[0].energy == 6.0
assert sampleset2.record[0].num_occurrences == 1
def test_local_solver_sa_qubo():
model = LogicalModel(mtype="qubo")
x = model.variables("x", shape=(2,))
model.add_interaction(x[0], coefficient=1.0)
model.add_interaction(x[1], coefficient=2.0)
model.add_interaction((x[0], x[1]), coefficient=-5.0)
model.offset(10.0)
solver = LocalSolver(exact=False)
sampleset = solver.solve(model.to_physical(), seed=12345)
assert sampleset.variables == ["x[0]", "x[1]"]
assert len(sampleset.record) == 1
# Check the ground state
assert np.array_equal(sampleset.record[0].sample, [0, 1])
assert sampleset.record[0].energy == 8.0
assert sampleset.record[0].num_occurrences == 1
def test_sawatabi_solver_qubo():
model = LogicalModel(mtype="qubo")
x = model.variables("x", shape=(2,))
model.add_interaction(x[0], coefficient=1.0)
model.add_interaction(x[1], coefficient=2.0)
model.add_interaction((x[0], x[1]), coefficient=-5.0)
model.offset(10.0)
solver = SawatabiSolver()
sampleset = solver.solve(model.to_physical(), num_reads=1, num_sweeps=10, cooling_rate=0.55, seed=12345)
assert sampleset.variables == ["x[0]", "x[1]"]
assert len(sampleset.record) == 1
# Check the ground state
record = sorted(sampleset.record, key=lambda r: r.energy) # sort by energy
assert np.array_equal(record[0].sample, [0, 1])
assert record[0].energy == 8.0
def test_sawatabi_solver_ising():
model = LogicalModel(mtype="ising")
s = model.variables("s", shape=(2,))
model.add_interaction(s[0], coefficient=1.0)
model.add_interaction(s[1], coefficient=2.0)
model.add_interaction((s[0], s[1]), coefficient=-3.0)
model.offset(10.0)
solver = SawatabiSolver()
sampleset = solver.solve(model.to_physical(), num_reads=2, num_sweeps=10, cooling_rate=0.5, initial_temperature=10.0, seed=12345)
assert sampleset.variables == ["s[0]", "s[1]"]
assert len(sampleset.record) == 1
# Check the ground state
assert np.array_equal(sampleset.record[0].sample, [-1, 1])
assert sampleset.record[0].energy == 6.0
assert sampleset.record[0].num_occurrences == 2