This repository has been archived by the owner on Apr 4, 2023. It is now read-only.
-
Notifications
You must be signed in to change notification settings - Fork 0
Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
- Loading branch information
1 parent
93f160f
commit 6ec0830
Showing
3 changed files
with
89 additions
and
138 deletions.
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
Original file line number | Diff line number | Diff line change |
---|---|---|
@@ -1,192 +1,138 @@ | ||
function __test_basic_examples(::Type{S}) where {S<:AbstractSampler} | ||
function __test_basic_examples(sampler::Type{S}) where {S<:AbstractSampler} | ||
n = 3 | ||
|
||
# ~ QUBO Matrix | ||
Q = [-1 2 2;2 -1 2;2 2 -1] | ||
Q = [ | ||
-1 2 2 | ||
2 -1 2 | ||
2 2 -1 | ||
] | ||
|
||
# ~ Boolean States | ||
↑ = 0.0 | ||
↓ = 1.0 | ||
↑, ↓ = 0, 1 | ||
|
||
Test.@testset "Basic ~ Bool ~ Min" begin | ||
model = JuMP.Model(S) | ||
|
||
# -*- Build Model -*- # | ||
model = JuMP.Model(sampler) | ||
JuMP.@variable(model, x[1:n], Bin) | ||
JuMP.@objective(model, Min, x' * Q * x) | ||
|
||
# -*- Run -*- # | ||
JuMP.optimize!(model) | ||
|
||
Test.@test JuMP.result_count(model) == 2^n # ~ 8 | ||
|
||
for i = 1:3 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↑, ↑, ↓] || xi ≈ [↑, ↓, ↑] || xi ≈ [↓, ↑, ↑] | ||
Test.@test yi ≈ -1.0 | ||
end | ||
|
||
let i = 4 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↑, ↑, ↑] | ||
Test.@test yi ≈ 0.0 | ||
end | ||
|
||
for i = 5:7 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↑, ↓, ↓] || xi ≈ [↓, ↓, ↑] || xi ≈ [↓, ↑, ↓] | ||
Test.@test yi ≈ 2.0 | ||
end | ||
Test.@test JuMP.result_count(model) > 0 | ||
|
||
let i = 8 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
for i = 1:JuMP.result_count(model) | ||
xi = JuMP.value.(x; result = i) | ||
yi = JuMP.objective_value(model; result = i) | ||
|
||
Test.@test xi ≈ [↓, ↓, ↓] | ||
Test.@test yi ≈ 9.0 | ||
if xi ≈ [↑, ↑, ↓] || xi ≈ [↑, ↓, ↑] || xi ≈ [↓, ↑, ↑] | ||
Test.@test yi ≈ -1.0 | ||
elseif xi ≈ [↑, ↑, ↑] | ||
Test.@test yi ≈ 0.0 | ||
elseif xi ≈ [↑, ↓, ↓] || xi ≈ [↓, ↓, ↑] || xi ≈ [↓, ↑, ↓] | ||
Test.@test yi ≈ 2.0 | ||
elseif xi ≈ [↓, ↓, ↓] | ||
Test.@test yi ≈ 9.0 | ||
else | ||
Test.@test false | ||
end | ||
end | ||
end | ||
|
||
Test.@testset "Basic ~ Bool ~ Max" begin | ||
model = JuMP.Model(S) | ||
|
||
# -*- Build Model -*- # | ||
model = JuMP.Model(sampler) | ||
JuMP.@variable(model, x[1:n], Bin) | ||
JuMP.@objective(model, Max, x' * Q * x) | ||
|
||
# -*- Run -*- # | ||
JuMP.optimize!(model) | ||
|
||
Test.@test JuMP.result_count(model) == 2^n # ~ 8 | ||
|
||
let i = 1 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↓, ↓, ↓] | ||
Test.@test yi ≈ 9.0 | ||
end | ||
|
||
for i = 2:4 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↑, ↓, ↓] || xi ≈ [↓, ↓, ↑] || xi ≈ [↓, ↑, ↓] | ||
Test.@test yi ≈ 2.0 | ||
end | ||
|
||
let i = 5 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test xi ≈ [↑, ↑, ↑] | ||
Test.@test yi ≈ 0.0 | ||
end | ||
Test.@test JuMP.result_count(model) > 0 | ||
|
||
for i = 6:8 | ||
xi = JuMP.value.(x; result=i) | ||
yi = JuMP.objective_value(model; result=i) | ||
for i = 1:JuMP.result_count(model) | ||
xi = JuMP.value.(x; result = i) | ||
yi = JuMP.objective_value(model; result = i) | ||
|
||
Test.@test xi ≈ [↑, ↑, ↓] || xi ≈ [↑, ↓, ↑] || xi ≈ [↓, ↑, ↑] | ||
Test.@test yi ≈ -1.0 | ||
if xi ≈ [↓, ↓, ↓] | ||
Test.@test yi ≈ 9.0 | ||
elseif xi ≈ [↑, ↓, ↓] || xi ≈ [↓, ↓, ↑] || xi ≈ [↓, ↑, ↓] | ||
Test.@test yi ≈ 2.0 | ||
elseif xi ≈ [↑, ↑, ↑] | ||
Test.@test yi ≈ 0.0 | ||
elseif xi ≈ [↑, ↑, ↓] || xi ≈ [↑, ↓, ↑] || xi ≈ [↓, ↑, ↑] | ||
Test.@test yi ≈ -1.0 | ||
else | ||
Test.@test false | ||
end | ||
end | ||
end | ||
|
||
# ~ Ising Hamiltonian | ||
J = [0 4 4; 0 0 4; 0 0 0] | ||
h = [-1;-1;-1] | ||
h = [-1; -1; -1] | ||
|
||
# ~ Spin states | ||
↑ = -1.0 | ||
↓ = 1.0 | ||
↑, ↓ = -1, 1 | ||
|
||
Test.@testset "Basic ~ Spin ~ Min" begin | ||
model = JuMP.Model(S) | ||
|
||
# -*- Build Model -*- # | ||
model = JuMP.Model(sampler) | ||
JuMP.@variable(model, s[1:n], Anneal.Spin) | ||
JuMP.@objective(model, Min, s' * J * s + h' * s) | ||
|
||
# -*- Run -*- # | ||
JuMP.optimize!(model) | ||
|
||
Test.@test JuMP.result_count(model) == 2^n # ~ 8 | ||
|
||
for i = 1:3 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↓, ↓, ↑] || si ≈ [↓, ↑, ↓] || si ≈ [↑, ↓, ↓] | ||
Test.@test Hi ≈ -5.0 | ||
end | ||
|
||
for i = 4:6 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↑, ↑, ↓] || si ≈ [↑, ↓, ↑] || si ≈ [↓, ↑, ↑] | ||
Test.@test Hi ≈ -3.0 | ||
end | ||
|
||
let i = 7 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↓, ↓, ↓] | ||
Test.@test Hi ≈ 9.0 | ||
end | ||
Test.@test JuMP.result_count(model) > 0 | ||
|
||
let i = 8 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
for i = 1:JuMP.result_count(model) | ||
si = JuMP.value.(s; result = i) | ||
Hi = JuMP.objective_value(model; result = i) | ||
|
||
Test.@test si ≈ [↑, ↑, ↑] | ||
Test.@test Hi ≈ 15.0 | ||
if si ≈ [↓, ↓, ↑] || si ≈ [↓, ↑, ↓] || si ≈ [↑, ↓, ↓] | ||
Test.@test Hi ≈ -5.0 | ||
elseif si ≈ [↑, ↑, ↓] || si ≈ [↑, ↓, ↑] || si ≈ [↓, ↑, ↑] | ||
Test.@test Hi ≈ -3.0 | ||
elseif si ≈ [↓, ↓, ↓] | ||
Test.@test Hi ≈ 9.0 | ||
elseif si ≈ [↑, ↑, ↑] | ||
Test.@test Hi ≈ 15.0 | ||
else | ||
Test.@test false | ||
end | ||
end | ||
end | ||
|
||
Test.@testset "Basic ~ Spin ~ Max" begin | ||
model = JuMP.Model(S) | ||
|
||
# -*- Build Model -*- # | ||
model = JuMP.Model(sampler) | ||
JuMP.@variable(model, s[1:n], Anneal.Spin) | ||
JuMP.@objective(model, Max, s' * J * s + h' * s) | ||
|
||
# -*- Run -*- # | ||
JuMP.optimize!(model) | ||
|
||
Test.@test JuMP.result_count(model) == 2^n # ~ 8 | ||
|
||
let i = 1 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↑, ↑, ↑] | ||
Test.@test Hi ≈ 15.0 | ||
end | ||
|
||
let i = 2 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↓, ↓, ↓] | ||
Test.@test Hi ≈ 9.0 | ||
end | ||
|
||
for i = 3:5 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
|
||
Test.@test si ≈ [↑, ↑, ↓] || si ≈ [↑, ↓, ↑] || si ≈ [↓, ↑, ↑] | ||
Test.@test Hi ≈ -3.0 | ||
end | ||
Test.@test JuMP.result_count(model) > 0 | ||
|
||
for i = 6:8 | ||
si = JuMP.value.(s; result=i) | ||
Hi = JuMP.objective_value(model; result=i) | ||
for i = 1:JuMP.result_count(model) | ||
si = JuMP.value.(s; result = i) | ||
Hi = JuMP.objective_value(model; result = i) | ||
|
||
Test.@test si ≈ [↓, ↓, ↑] || si ≈ [↓, ↑, ↓] || si ≈ [↑, ↓, ↓] | ||
Test.@test Hi ≈ -5.0 | ||
if si ≈ [↑, ↑, ↑] | ||
Test.@test Hi ≈ 15.0 | ||
elseif si ≈ [↓, ↓, ↓] | ||
Test.@test Hi ≈ 9.0 | ||
elseif si ≈ [↑, ↑, ↓] || si ≈ [↑, ↓, ↑] || si ≈ [↓, ↑, ↑] | ||
Test.@test Hi ≈ -3.0 | ||
elseif si ≈ [↓, ↓, ↑] || si ≈ [↓, ↑, ↓] || si ≈ [↑, ↓, ↓] | ||
Test.@test Hi ≈ -5.0 | ||
else | ||
Test.@test false | ||
end | ||
end | ||
end | ||
|
||
nothing | ||
return nothing | ||
end |
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
6ec0830
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
@JuliaRegistrator register
6ec0830
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Registration pull request created: JuliaRegistries/General/70931
After the above pull request is merged, it is recommended that a tag is created on this repository for the registered package version.
This will be done automatically if the Julia TagBot GitHub Action is installed, or can be done manually through the github interface, or via: