Boozer test fixes

hiddenSymmetries · Oct 31, 2023 · 917eef3 · 917eef3
1 parent ec471a7
commit 917eef3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/mhd/test_boozer.py b/tests/mhd/test_boozer.py
@@ -184,7 +184,7 @@ def test_boozer_circular_tokamak(self):
             # registered surface:
             # Register a QP target at s = 1:
             s = 0.9999
-            Quasisymmetry(b, s, 0, 1)
+            Quasisymmetry(b, s, 0, 1).J()
             self.assertEqual(b.s, {0.5, s, 1.0})
             np.testing.assert_allclose(b.bx.compute_surfs, [7, 15])
             self.assertEqual(b.s_to_index, {0.5: 0, 1.0: 1, s: 1})
@@ -205,7 +205,7 @@ def test_boozer_circular_tokamak(self):
     def test_boozer_li383(self):
         v = Vmec(os.path.join(TEST_DIR, "wout_li383_low_res_reference.nc"))
         b = Boozer(v, mpol=32, ntor=16)
-        Quasisymmetry(b, [0.0, 1.0], 1, 0)
+        Quasisymmetry(b, [0.0, 1.0], 1, 0).J()
         np.testing.assert_allclose(b.bx.compute_surfs, [0, 14])
         self.assertEqual(b.s_to_index, {0.0: 0, 1.0: 1})
         bmnc = b.bx.bmnc_b