add QuadraticFlux obj

- doesn't pass due to jax bug, maybe PR #1002 will fix it?
PlasmaControl · Jul 4, 2024 · 680848b · 680848b
1 parent 7ba2a73
commit 680848b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/tests/test_examples.py b/tests/test_examples.py
@@ -1351,7 +1351,10 @@ def test_optimize_with_all_coil_types(DummyCoilSet, DummyMixedCoilSet):
 
     def test(c, method):
         target = 11
-        obj = ObjectiveFunction(CoilLength(c, target=target))
+        obj = ObjectiveFunction(
+            CoilLength(c, target=target),
+            QuadraticFlux(eq=Equilibrium(), field=c, vacuum=True),
+        )
         optimizer = Optimizer(method)
         (c,), _ = optimizer.optimize(c, obj, maxiter=200, ftol=0, xtol=1e-15)
         flattened_coils = tree_leaves(