modify test to ensure currents are saved correctly too

PlasmaControl · May 12, 2024 · f06f8ae · f06f8ae
1 parent d5f67c9
commit f06f8ae
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test_coils.py b/tests/test_coils.py
@@ -923,9 +923,9 @@ def test_save_h5_for_mixed_coilset(tmpdir_factory):
     output_dir = tmpdir_factory.mktemp("coilset_save")
     output_path = output_dir.join("DummyMixedCoilSet.h5")
 
-    tf_coil = FourierPlanarCoil(center=[2, 0, 0], normal=[0, 1, 0], r_n=[1])
+    tf_coil = FourierPlanarCoil(current=20, center=[2, 0, 0], normal=[0, 1, 0], r_n=[1])
     tf_coilset = CoilSet.linspaced_angular(tf_coil, n=4)
-    vf_coil = FourierRZCoil(R_n=3, Z_n=-1)
+    vf_coil = FourierRZCoil(current=100, R_n=3, Z_n=-1)
     vf_coilset = CoilSet.linspaced_linear(
         vf_coil, displacement=[0, 0, 2], n=3, endpoint=True
     )