Add examples test to collected tests (when not run with pytest).

PiperOrigin-RevId: 732211424
google-deepmind · Feb 28, 2025 · c620238 · c620238
1 parent 0554b3c
commit c620238
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/torax/examples/tests/examples.py b/torax/examples/tests/examples.py
@@ -16,6 +16,7 @@
 import importlib
 import os
 
+from absl.testing import absltest
 from absl.testing import parameterized
 from torax.config import build_sim
 from torax.tests.test_lib import paths
@@ -41,3 +42,7 @@ def test_build_sim_from_config(self, config_name_no_py: str):
         python_config_module, PYTHON_CONFIG_PACKAGE
     )
     _ = build_sim.build_sim_from_config(config_module.CONFIG)
+
+
+if __name__ == '__main__':
+  absltest.main()