Add test case for rotation with tglfnn-ukaea in sim_test

hamelphi · Torax team · commit 5c7009afdc4e · 2026-02-12T12:16:00.000-08:00
PiperOrigin-RevId: 866504545
diff --git a/torax/tests/sim_test.py b/torax/tests/sim_test.py
@@ -258,6 +258,13 @@ 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',
diff --git a/torax/tests/test_data/test_iterhybrid_predictor_corrector_tglfnn_ukaea_rotation.nc b/torax/tests/test_data/test_iterhybrid_predictor_corrector_tglfnn_ukaea_rotation.nc
diff --git a/torax/tests/test_data/test_iterhybrid_predictor_corrector_tglfnn_ukaea_rotation.py b/torax/tests/test_data/test_iterhybrid_predictor_corrector_tglfnn_ukaea_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'] = 1.0

Original file line number	Diff line number	Diff line change
`@@ -32,3 +32,7 @@`
`32`	`32`	`0.0: 1.8 * _omega_ref,`
`33`	`33`	`1.0: 0.0,`
`34`	`34`	`}`
	`35`	`+`
	`36`	`+# Reduce the final time since numerical difference across platforms can`
	`37`	`+# cause significant divergences over time.`
	`38`	`+CONFIG['numerics']['t_final'] = 1.0`