edit to unit test

hiddenSymmetries · May 17, 2024 · 8be1c50 · 8be1c50
1 parent 6890be8
commit 8be1c50
Showing 1 changed file with 7 additions and 4 deletions.
diff --git a/tests/geo/test_boozersurface.py b/tests/geo/test_boozersurface.py
@@ -365,15 +365,18 @@ def test_run_code(self):
         # this second time should not actually run
         boozer_surface.run_code(boozer_surface.res['iota'], G=boozer_surface.res['G'])
 
+        for c in bs.coils:
+            c.current.fix_all()
+
+        boozer_surface.need_to_run_code=True
+        # run without providing value of G
+        boozer_surface.run_code(boozer_surface.res['iota'])
+
         bs, boozer_surface = get_boozer_surface(boozer_type='exact')
         boozer_surface.run_code(boozer_surface.res['iota'], G=boozer_surface.res['G'])
 
         # this second time should not actually run
         boozer_surface.run_code(boozer_surface.res['iota'], G=boozer_surface.res['G'])
-
-        boozer_surface.need_to_run_code=True
-        # run without providing value of G
-        boozer_surface.run_code(boozer_surface.res['iota'])
 
     def test_convergence_cpp_and_notcpp_same(self):
         """