Resolves #1008

PlasmaControl · Apr 30, 2024 · 1d97cab · 1d97cab
1 parent 8b2fd29
commit 1d97cab
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 1 deletion.
diff --git 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
@@ -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("DSHAPE_current")
     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