add test case for rotation with tglfnn-ukaea in sim_test

PiperOrigin-RevId: 866504545

Author: hamelphi

torax/tests/sim_test.py

@@ -258,6 +258,20 @@ class SimTest(sim_test_case.SimTestCase):
           'test_iterhybrid_predictor_corrector_tglfnn_ukaea',
           'test_iterhybrid_predictor_corrector_tglfnn_ukaea.py',
       ),
+      # Predictor-corrector solver with TGLFNNukaea transport with rotation
+      (
+          'test_iterhybrid_predictor_corrector_tglfnn_ukaea_rotation',
+          'test_iterhybrid_predictor_corrector_tglfnn_ukaea_rotation.py',
+          _ALL_PROFILES,
+          1e-6,
+      ),
+      # Predictor-corrector solver with rotation
+      (
+          'test_iterhybrid_predictor_corrector_rotation',
+          'test_iterhybrid_predictor_corrector_rotation.py',
+          _ALL_PROFILES,
+          1e-6,
+      ),
       # L-mode iterhybrid variant with combined transport model and overwrites
       (
           'test_iterhybrid_predictor_corrector_Lmode_combined',

torax/tests/test_data/test_iterhybrid_predictor_corrector_rotation.nc

@@ -31,6 +31,11 @@
     1.0: 0.0,
 }
 
+# Reduce the final time since numerical difference across platforms can
+# cause significant divergences over time.
+CONFIG['numerics']['t_final'] = 3.0
+
+
 # Modify this if you want to scale up or down the effect of rotation.
 CONFIG['transport']['rotation_multiplier'] = 1.0
 CONFIG['transport']['shear_suppression_model'] = 'vandeplassche2020'

torax/tests/test_data/test_iterhybrid_predictor_corrector_rotation.py

@@ -32,3 +32,7 @@
     0.0: 1.8 * _omega_ref,
     1.0: 0.0,
 }
+
+# Reduce the final time since numerical difference across platforms can
+# cause significant divergences over time.
+CONFIG['numerics']['t_final'] = 3.0