rework main block example to have a constraint, and do an assert on it.

peterdsharpe · Oct 24, 2023 · 70ede74 · 70ede74
1 parent 5aa1d7b
commit 70ede74
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/aerosandbox/optimization/opti.py b/aerosandbox/optimization/opti.py
@@ -1430,8 +1430,11 @@ def show_infeasibilities(self, tol: float = 1e-3) -> None:
     f = (a - x) ** 2 + b * (y - x ** 2) ** 2
     opti.minimize(f)
 
+    opti.subject_to([
+        x ** 2 + y ** 2 <= 1
+    ])
+
     # Optimize
     sol = opti.solve()
 
-    for i in [x, y]:
-        assert sol.value(i) == pytest.approx(1, abs=1e-4)
+    assert sol([x, y]) == pytest.approx([0.7864, 0.6177], abs=1e-3)