diff --git a/desc/equilibrium/coords.py b/desc/equilibrium/coords.py index 253bbb8a69..96699c9d0e 100644 --- a/desc/equilibrium/coords.py +++ b/desc/equilibrium/coords.py @@ -110,6 +110,7 @@ def periodic(x): # do surface average to get iota once if "iota" in profiles and profiles["iota"] is None: profiles["iota"] = eq.get_profile("iota", params=params) + params["i_l"] = profiles["iota"].params @functools.partial(jit, static_argnums=1) def compute(y, basis): diff --git a/tests/test_equilibrium.py b/tests/test_equilibrium.py index c0ba9bb023..04413817b5 100644 --- a/tests/test_equilibrium.py +++ b/tests/test_equilibrium.py @@ -48,7 +48,7 @@ def test_compute_theta_coords(): def test_map_coordinates(): """Test root finding for (rho,theta,zeta) for common use cases.""" # finding coordinates along a single field line - eq = get("W7-X") + eq = get("NCSX") eq.change_resolution(3, 3, 3, 6, 6, 6) n = 100 coords = np.array([np.ones(n), np.zeros(n), np.linspace(0, 10 * np.pi, n)]).T