def test_sawatabi_solver_n_hot_ising_with_deleting(n, s, i): # n out of (s - 1) variables should be 1 model = LogicalModel(mtype="ising") x = model.variables("x", shape=(s,)) model.add_constraint(NHotConstraint(variables=x, n=n)) model.delete_variable(x[i]) solver = SawatabiSolver() sampleset = solver.solve(model.to_physical(), seed=12345) result = np.array(sampleset.record[0].sample) assert np.count_nonzero(result == 1) == n assert np.count_nonzero(result == -1) == s - n - 1
def _create_ising_model_for_eq(): model = LogicalModel(mtype="ising") x = model.variables(name="x", shape=(4,)) z = model.variables(name="z", shape=(4,)) model.add_interaction(target=x[0], coefficient=1.1) model.add_interaction(target=(x[0], x[1]), coefficient=2.2, scale=3.3, attributes={"foo": "bar"}) model.add_interaction(target=x[2], coefficient=4.4) model.add_interaction(target=x[3], coefficient=5.5) model.remove_interaction(target=x[2]) model.fix_variable(target=x[3], value=1) model.add_constraint(NHotConstraint(variables=z, n=1)) return model
def test_logical_model_to_physical_with_n_hot_constraint_qubo(n, s): # n out of s variables should be 1 model = LogicalModel(mtype="qubo") x = model.variables("x", shape=(s,)) model.add_constraint(NHotConstraint(x, n=n)) physical = model.to_physical() for i in range(s): assert physical._raw_interactions[constants.INTERACTION_LINEAR][f"x[{i}]"] == 2 * n - 1.0 for i in range(s): for j in range(s): l1 = f"x[{i}]" l2 = f"x[{j}]" if l1 < l2: assert physical._raw_interactions[constants.INTERACTION_QUADRATIC][(l1, l2)] == -2.0
def test_local_solver_zero_or_one_hot_qubo(n): model = LogicalModel(mtype="qubo") x = model.variables("x", shape=(n,)) model.add_constraint(ZeroOrOneHotConstraint(variables=x)) solver = LocalSolver() physical = model.to_physical() for seed in [11, 22, 33, 44, 55]: sampleset = solver.solve(physical, seed=seed) result = np.array(sampleset.record[0].sample) assert np.count_nonzero(result == 1) in [0, 1] # Execution time should be within practical seconds (20 sec). assert sampleset.info["timing"]["execution_sec"] <= 20.0
def test_sawatabi_solver_n_hot_qubo(n, s): # n out of s variables should be 1 model = LogicalModel(mtype="qubo") x = model.variables("x", shape=(s,)) model.add_constraint(NHotConstraint(variables=x, n=n)) solver = SawatabiSolver() sampleset = solver.solve(model.to_physical(), seed=12345) result = np.array(sampleset.record[0].sample) assert np.count_nonzero(result == 1) == n assert np.count_nonzero(result == 0) == s - n # Execution time should be within practical seconds (20 sec). assert sampleset.info["timing"]["execution_sec"] <= 20.0
def test_local_solver_n_hot_ising(n, s): # n out of s spins should be +1 model = LogicalModel(mtype="ising") x = model.variables("x", shape=(s,)) model.add_constraint(NHotConstraint(variables=x, n=n)) solver = LocalSolver() physical = model.to_physical() for seed in [11, 22, 33, 44, 55]: sampleset = solver.solve(physical, seed=seed) result = np.array(sampleset.record[0].sample) assert np.count_nonzero(result == 1) == n assert np.count_nonzero(result == -1) == s - n # Execution time should be within practical seconds (20 sec). assert sampleset.info["timing"]["execution_sec"] <= 20.0
def test_logical_model_to_physical_with_n_hot_constraint_ising(n, s): # n out of s spins should be +1 model = LogicalModel(mtype="ising") x = model.variables("x", shape=(s,)) model.add_constraint(NHotConstraint(x, n=n)) physical = model.to_physical() for i in range(s): if s != 2 * n: assert physical._raw_interactions[constants.INTERACTION_LINEAR][f"x[{i}]"] == -0.5 * (s - 2 * n) else: assert f"x[{i}]" not in physical._raw_interactions[constants.INTERACTION_LINEAR] for i in range(s): for j in range(s): l1 = f"x[{i}]" l2 = f"x[{j}]" if l1 < l2: assert physical._raw_interactions[constants.INTERACTION_QUADRATIC][(l1, l2)] == -0.5
def test_local_solver_equality_qubo(m, n): model = LogicalModel(mtype="qubo") x = model.variables("x", shape=(m,)) y = model.variables("y", shape=(n,)) model.add_constraint(EqualityConstraint(variables_1=x, variables_2=y)) solver = LocalSolver() physical = model.to_physical() for seed in [11, 22, 33, 44, 55]: sampleset = solver.solve(physical, seed=seed) result = np.array(sampleset.record[0].sample) result_1 = result[0:m] result_2 = result[m : (m + n)] # noqa: E203 assert np.count_nonzero(result_1 == 1) == np.count_nonzero(result_2 == 1) # Execution time should be within practical seconds (20 sec). assert sampleset.info["timing"]["execution_sec"] <= 20.0
def test_logical_model_delete_dealing_with_nhot_constraints_qubo(): model = LogicalModel(mtype="qubo") x = model.variables("x", shape=(4, )) default_label = "Default N-hot Constraint" model.add_constraint(NHotConstraint(x, n=1, strength=1.0)) assert len(model.get_constraints()) == 1 assert default_label in model.get_constraints() assert model.get_constraints_by_label(default_label)._n == 1 assert len(model.get_constraints_by_label(default_label)._variables) == 4 model.delete_variable(x[0]) assert len(model.get_constraints()) == 1 assert default_label in model.get_constraints() assert model.get_constraints_by_label(default_label)._n == 1 assert len(model.get_constraints_by_label(default_label)._variables) == 3 physical = model.to_physical() assert physical._raw_interactions[ constants.INTERACTION_LINEAR]["x[1]"] == 1.0 assert physical._raw_interactions[constants.INTERACTION_QUADRATIC][( "x[1]", "x[2]")] == -2.0
def ising_z3(): model = LogicalModel(mtype="ising") z = model.variables("z", shape=(3,)) model.add_constraint(NHotConstraint(variables=z, n=1)) return model