Skip to content

Commit cf146a6

Browse files
authored
Test variable substitution for KernelBridge (#353)
* Test variable substitution for KernelBridge * Fix * Fix format * Fix * Fix format
1 parent b9a03eb commit cf146a6

File tree

1 file changed

+27
-8
lines changed

1 file changed

+27
-8
lines changed

test/Bridges/Variable/kernel.jl

Lines changed: 27 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -22,29 +22,48 @@ function test_runtests()
2222
MOI.Bridges.runtests(
2323
SumOfSquares.Bridges.Variable.KernelBridge,
2424
model -> begin
25-
MOI.add_constrained_variables(
25+
p, _ = MOI.add_constrained_variables(
2626
model,
2727
SumOfSquares.WeightedSOSCone{
2828
MOI.PositiveSemidefiniteConeTriangle,
2929
}(
3030
MB.SubBasis{MB.Monomial}([
31-
x^4,
32-
x^3 * y,
33-
x^2 * y^2,
34-
x * y^3,
3531
y^4,
32+
x * y^3,
33+
x^2 * y^2,
34+
x^3 * y,
35+
x^4,
3636
]),
37-
[MB.SubBasis{MB.Monomial}([x^2, y^2, x * y])],
37+
[MB.SubBasis{MB.Monomial}([y^2, x * y, x^2])],
3838
[MB.algebra_element(1.0 * x^0 * y^0)],
3939
),
4040
)
41+
a = float.(1:length(p))
42+
MOI.add_constraint(
43+
model,
44+
MOI.Utilities.vectorize([a' * p]),
45+
MOI.Zeros(1),
46+
)
4147
end,
4248
model -> begin
43-
Q, _ = MOI.add_constrained_variables(
49+
q, _ = MOI.add_constrained_variables(
4450
model,
4551
MOI.PositiveSemidefiniteConeTriangle(3),
4652
)
47-
end,
53+
a = float.(1:length(q))
54+
MOI.add_constraint(
55+
model,
56+
MOI.Utilities.vectorize([
57+
1.0 * q[1] +
58+
4.0 * q[2] +
59+
3.0 * (1.0q[3] + 2.0q[4]) +
60+
8.0 * q[5] +
61+
5.0 * q[6],
62+
]),
63+
MOI.Zeros(1),
64+
)
65+
end;
66+
cannot_unbridge = true,
4867
)
4968
return
5069
end

0 commit comments

Comments
 (0)