Add missing key to stability fun test

PlasmaControl · Oct 6, 2024 · ce64c53 · ce64c53
1 parent 5e72593
commit ce64c53
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_stability_funs.py b/tests/test_stability_funs.py
@@ -416,6 +416,7 @@ def test_ballooning_geometry(tmpdir_factory):
             "g^ra",
             "g^rr",
             "cvdrift",
+            "gbdrift",
             "cvdrift0",
             "|B|",
             "B^zeta",
@@ -448,11 +449,9 @@ def test_ballooning_geometry(tmpdir_factory):
         sign_psi = psi_s / np.abs(psi_s)
         sign_iota = iotas / np.abs(iotas)
 
-        modB = data["|B|"]
         x = Lref * np.sqrt(psi)
         shat = -x / iotas * shears / Lref
 
-        psi_r = data["psi_r"]
         grad_alpha = data["grad(alpha)"]
 
         g_sup_rr = data["g^rr"]