google-deepmind · copybara-service · May 19, 2025
diff --git a/docs/contribution_tips.rst b/docs/contribution_tips.rst
@@ -130,16 +130,16 @@ Further filtering is possible, for example running only the ``test_face_to_cell`
 
 Which runs any test containing the string expression ``face_to_cell``.
 
-Where appropiate, please add tests for your changes.
+Where appropriate, please add tests for your changes.
 
 An important class of test is the sim test. These are integration tests running
-the configs in the ``torax/tests/test_data/`` directory, and comparing to the ground-truth
+the configs in the ``torax/_src/test_data/`` directory, and comparing to the ground-truth
 ``.nc`` TORAX outputs found in the same directory. Sim tests can be triggered separately
 by a command (from the TORAX root directory) such as:
 
 .. code-block:: console
 
-  pytest -n <num_workers> torax/tests/sim_test.py
+  pytest -n <num_workers> torax/_src/integration_tests/sim_test.py
 
 If any sim tests fail, they write their output to the ``/tmp/torax_failed_sim_test_outputs/<test_name>.nc``.
 This is useful for debugging, and also to stage new output files for replacing the ground-truth files,
@@ -163,7 +163,7 @@ the TORAX repository. Using ``test_qlknnheat`` as an example:
 
 .. code-block:: console
 
-  plot_torax --outfile torax/tests/test_data/test_qlknnheat.nc /tmp/torax_failed_sim_test_outputs/test_qlknnheat.nc
+  plot_torax --outfile torax/_src/test_data/test_qlknnheat.nc /tmp/torax_failed_sim_test_outputs/test_qlknnheat.nc
 
 If it is deemed that the new outputs should replace the ground-truth files,
 they can be copied over using the following command, again with this example working

diff --git a/pyproject.toml b/pyproject.toml
@@ -96,11 +96,11 @@ exclude = ["*tests*"]
 
 [tool.pytest.ini_options]
 addopts = [
-    "--ignore=torax/tests/test_data",
+    "--ignore=torax/_src/test_data",
     "--ignore=torax/tests/scripts",
     "--ignore-glob=*lib*",
     # Leads to an ImportError in collection for `pytest` without qualikiz_tools.
     "--ignore=torax/_src/transport_model/qualikiz_wrapper.py",
 ]
-testpaths = "**/tests"
+testpaths = ["**/tests", "**/integration_tests"]
 python_files = "*.py"
diff --git a/pytest.ini b/pytest.ini
@@ -1,2 +1,2 @@
 [pytest]
-python_files = torax/**/tests/*_test.py
+python_files = torax/**/*tests/*_test.py
diff --git a/torax/tests/persistent_cache_test.py → ...ntegration_tests/persistent_cache_test.py b/torax/tests/persistent_cache_test.py → ...ntegration_tests/persistent_cache_test.py
@@ -38,7 +38,7 @@ def test_persistent_cache(self):
     cache = cache.full_path
 
     flags = [
-        '--config=tests/test_data/test_iterhybrid_rampup_short.py',
+        '--config=_src/test_data/test_iterhybrid_rampup_short.py',
         f'--jax_compilation_cache_dir={cache}',
         '--jax_persistent_cache_min_entry_size_bytes=-1',
         '--jax_persistent_cache_min_compile_time_secs=0.0',

diff --git a/torax/tests/sim_no_compile_test.py → .../integration_tests/sim_no_compile_test.py b/torax/tests/sim_no_compile_test.py → .../integration_tests/sim_no_compile_test.py
diff --git a/torax/tests/sim_test.py → torax/_src/integration_tests/sim_test.py b/torax/tests/sim_test.py → torax/_src/integration_tests/sim_test.py
diff --git a/torax/tests/sim_time_dependence_test.py → ...gration_tests/sim_time_dependence_test.py b/torax/tests/sim_time_dependence_test.py → ...gration_tests/sim_time_dependence_test.py
diff --git a/.../tests/test_data/compilation_benchmark.py → ...x/_src/test_data/compilation_benchmark.py b/.../tests/test_data/compilation_benchmark.py → ...x/_src/test_data/compilation_benchmark.py
diff --git a/torax/tests/test_data/default_config.py → torax/_src/test_data/default_config.py b/torax/tests/test_data/default_config.py → torax/_src/test_data/default_config.py
diff --git a/...t_data/test_all_transport_fusion_qlknn.nc → ...t_data/test_all_transport_fusion_qlknn.nc b/...t_data/test_all_transport_fusion_qlknn.nc → ...t_data/test_all_transport_fusion_qlknn.nc
diff --git a/...t_data/test_all_transport_fusion_qlknn.py → ...t_data/test_all_transport_fusion_qlknn.py b/...t_data/test_all_transport_fusion_qlknn.py → ...t_data/test_all_transport_fusion_qlknn.py
diff --git a/.../tests/test_data/test_bohmgyrobohm_all.nc → ...x/_src/test_data/test_bohmgyrobohm_all.nc b/.../tests/test_data/test_bohmgyrobohm_all.nc → ...x/_src/test_data/test_bohmgyrobohm_all.nc
diff --git a/.../tests/test_data/test_bohmgyrobohm_all.py → ...x/_src/test_data/test_bohmgyrobohm_all.py b/.../tests/test_data/test_bohmgyrobohm_all.py → ...x/_src/test_data/test_bohmgyrobohm_all.py
diff --git a/...est_bremsstrahlung_time_dependent_Zimp.nc → ...est_bremsstrahlung_time_dependent_Zimp.nc b/...est_bremsstrahlung_time_dependent_Zimp.nc → ...est_bremsstrahlung_time_dependent_Zimp.nc
diff --git a/...est_bremsstrahlung_time_dependent_Zimp.py → ...est_bremsstrahlung_time_dependent_Zimp.py b/...est_bremsstrahlung_time_dependent_Zimp.py → ...est_bremsstrahlung_time_dependent_Zimp.py
diff --git a/...s/test_data/test_changing_config_after.nc → ...c/test_data/test_changing_config_after.nc b/...s/test_data/test_changing_config_after.nc → ...c/test_data/test_changing_config_after.nc
diff --git a/...s/test_data/test_changing_config_after.py → ...c/test_data/test_changing_config_after.py b/...s/test_data/test_changing_config_after.py → ...c/test_data/test_changing_config_after.py
diff --git a/.../test_data/test_changing_config_before.nc → .../test_data/test_changing_config_before.nc b/.../test_data/test_changing_config_before.nc → .../test_data/test_changing_config_before.nc
diff --git a/.../test_data/test_changing_config_before.py → .../test_data/test_changing_config_before.py b/.../test_data/test_changing_config_before.py → .../test_data/test_changing_config_before.py
diff --git a/torax/tests/test_data/test_chease.nc → torax/_src/test_data/test_chease.nc b/torax/tests/test_data/test_chease.nc → torax/_src/test_data/test_chease.nc
diff --git a/torax/tests/test_data/test_chease.py → torax/_src/test_data/test_chease.py b/torax/tests/test_data/test_chease.py → torax/_src/test_data/test_chease.py
diff --git a/torax/tests/test_data/test_crank_nicolson.py → torax/_src/test_data/test_crank_nicolson.py b/torax/tests/test_data/test_crank_nicolson.py → torax/_src/test_data/test_crank_nicolson.py
diff --git a/torax/tests/test_data/test_eqdsk.nc → torax/_src/test_data/test_eqdsk.nc b/torax/tests/test_data/test_eqdsk.nc → torax/_src/test_data/test_eqdsk.nc
diff --git a/torax/tests/test_data/test_eqdsk.py → torax/_src/test_data/test_eqdsk.py b/torax/tests/test_data/test_eqdsk.py → torax/_src/test_data/test_eqdsk.py
diff --git a/torax/tests/test_data/test_fixed_dt.nc → torax/_src/test_data/test_fixed_dt.nc b/torax/tests/test_data/test_fixed_dt.nc → torax/_src/test_data/test_fixed_dt.nc
diff --git a/torax/tests/test_data/test_fixed_dt.py → torax/_src/test_data/test_fixed_dt.py b/torax/tests/test_data/test_fixed_dt.py → torax/_src/test_data/test_fixed_dt.py
diff --git a/torax/tests/test_data/test_implicit.nc → torax/_src/test_data/test_implicit.nc b/torax/tests/test_data/test_implicit.nc → torax/_src/test_data/test_implicit.nc
diff --git a/torax/tests/test_data/test_implicit.py → torax/_src/test_data/test_implicit.py b/torax/tests/test_data/test_implicit.py → torax/_src/test_data/test_implicit.py
diff --git a/...est_data/test_implicit_short_optimizer.nc → ...est_data/test_implicit_short_optimizer.nc b/...est_data/test_implicit_short_optimizer.nc → ...est_data/test_implicit_short_optimizer.nc
diff --git a/...est_data/test_implicit_short_optimizer.py → ...est_data/test_implicit_short_optimizer.py b/...est_data/test_implicit_short_optimizer.py → ...est_data/test_implicit_short_optimizer.py
diff --git a/...sts/test_data/test_iterbaseline_mockup.nc → ...src/test_data/test_iterbaseline_mockup.nc b/...sts/test_data/test_iterbaseline_mockup.nc → ...src/test_data/test_iterbaseline_mockup.nc
diff --git a/...sts/test_data/test_iterbaseline_mockup.py → ...src/test_data/test_iterbaseline_mockup.py b/...sts/test_data/test_iterbaseline_mockup.py → ...src/test_data/test_iterbaseline_mockup.py
diff --git a/...sts/test_data/test_iterhybrid_makenans.py → ...src/test_data/test_iterhybrid_makenans.py b/...sts/test_data/test_iterhybrid_makenans.py → ...src/test_data/test_iterhybrid_makenans.py
diff --git a/...tests/test_data/test_iterhybrid_mockup.nc → .../_src/test_data/test_iterhybrid_mockup.nc b/...tests/test_data/test_iterhybrid_mockup.nc → .../_src/test_data/test_iterhybrid_mockup.nc
diff --git a/...tests/test_data/test_iterhybrid_mockup.py → .../_src/test_data/test_iterhybrid_mockup.py b/...tests/test_data/test_iterhybrid_mockup.py → .../_src/test_data/test_iterhybrid_mockup.py
diff --git a/...ta/test_iterhybrid_predictor_corrector.nc → ...ta/test_iterhybrid_predictor_corrector.nc b/...ta/test_iterhybrid_predictor_corrector.nc → ...ta/test_iterhybrid_predictor_corrector.nc
diff --git a/...ta/test_iterhybrid_predictor_corrector.py → ...ta/test_iterhybrid_predictor_corrector.py b/...ta/test_iterhybrid_predictor_corrector.py → ...ta/test_iterhybrid_predictor_corrector.py
diff --git a/...hybrid_predictor_corrector_clip_inputs.nc → ...hybrid_predictor_corrector_clip_inputs.nc b/...hybrid_predictor_corrector_clip_inputs.nc → ...hybrid_predictor_corrector_clip_inputs.nc
diff --git a/...hybrid_predictor_corrector_clip_inputs.py → ...hybrid_predictor_corrector_clip_inputs.py b/...hybrid_predictor_corrector_clip_inputs.py → ...hybrid_predictor_corrector_clip_inputs.py
diff --git a/...r_constant_fraction_impurity_radiation.nc → ...r_constant_fraction_impurity_radiation.nc b/...r_constant_fraction_impurity_radiation.nc → ...r_constant_fraction_impurity_radiation.nc
diff --git a/...r_constant_fraction_impurity_radiation.py → ...r_constant_fraction_impurity_radiation.py b/...r_constant_fraction_impurity_radiation.py → ...r_constant_fraction_impurity_radiation.py
diff --git a/...erhybrid_predictor_corrector_cyclotron.nc → ...erhybrid_predictor_corrector_cyclotron.nc b/...erhybrid_predictor_corrector_cyclotron.nc → ...erhybrid_predictor_corrector_cyclotron.nc
diff --git a/...erhybrid_predictor_corrector_cyclotron.py → ...erhybrid_predictor_corrector_cyclotron.py b/...erhybrid_predictor_corrector_cyclotron.py → ...erhybrid_predictor_corrector_cyclotron.py
diff --git a/...erhybrid_predictor_corrector_ec_linliu.nc → ...erhybrid_predictor_corrector_ec_linliu.nc b/...erhybrid_predictor_corrector_ec_linliu.nc → ...erhybrid_predictor_corrector_ec_linliu.nc
diff --git a/...erhybrid_predictor_corrector_ec_linliu.py → ...erhybrid_predictor_corrector_ec_linliu.py b/...erhybrid_predictor_corrector_ec_linliu.py → ...erhybrid_predictor_corrector_ec_linliu.py
diff --git a/...t_iterhybrid_predictor_corrector_eqdsk.nc → ...t_iterhybrid_predictor_corrector_eqdsk.nc b/...t_iterhybrid_predictor_corrector_eqdsk.nc → ...t_iterhybrid_predictor_corrector_eqdsk.nc
diff --git a/...t_iterhybrid_predictor_corrector_eqdsk.py → ...t_iterhybrid_predictor_corrector_eqdsk.py b/...t_iterhybrid_predictor_corrector_eqdsk.py → ...t_iterhybrid_predictor_corrector_eqdsk.py
diff --git a/...or_corrector_mavrin_impurity_radiation.nc → ...or_corrector_mavrin_impurity_radiation.nc b/...or_corrector_mavrin_impurity_radiation.nc → ...or_corrector_mavrin_impurity_radiation.nc
diff --git a/...or_corrector_mavrin_impurity_radiation.py → ...or_corrector_mavrin_impurity_radiation.py b/...or_corrector_mavrin_impurity_radiation.py → ...or_corrector_mavrin_impurity_radiation.py
diff --git a/...terhybrid_predictor_corrector_qualikiz.nc → ...terhybrid_predictor_corrector_qualikiz.nc b/...terhybrid_predictor_corrector_qualikiz.nc → ...terhybrid_predictor_corrector_qualikiz.nc
diff --git a/...terhybrid_predictor_corrector_qualikiz.py → ...terhybrid_predictor_corrector_qualikiz.py b/...terhybrid_predictor_corrector_qualikiz.py → ...terhybrid_predictor_corrector_qualikiz.py
diff --git a/...ctor_corrector_set_pped_tpedratio_nped.nc → ...ctor_corrector_set_pped_tpedratio_nped.nc b/...ctor_corrector_set_pped_tpedratio_nped.nc → ...ctor_corrector_set_pped_tpedratio_nped.nc
diff --git a/...ctor_corrector_set_pped_tpedratio_nped.py → ...ctor_corrector_set_pped_tpedratio_nped.py b/...ctor_corrector_set_pped_tpedratio_nped.py → ...ctor_corrector_set_pped_tpedratio_nped.py
diff --git a/...ictor_corrector_timedependent_isotopes.nc → ...ictor_corrector_timedependent_isotopes.nc b/...ictor_corrector_timedependent_isotopes.nc → ...ictor_corrector_timedependent_isotopes.nc
diff --git a/...ictor_corrector_timedependent_isotopes.py → ...ictor_corrector_timedependent_isotopes.py b/...ictor_corrector_timedependent_isotopes.py → ...ictor_corrector_timedependent_isotopes.py
diff --git a/...terhybrid_predictor_corrector_tungsten.nc → ...terhybrid_predictor_corrector_tungsten.nc b/...terhybrid_predictor_corrector_tungsten.nc → ...terhybrid_predictor_corrector_tungsten.nc
diff --git a/...terhybrid_predictor_corrector_tungsten.py → ...terhybrid_predictor_corrector_tungsten.py b/...terhybrid_predictor_corrector_tungsten.py → ...terhybrid_predictor_corrector_tungsten.py
diff --git a/...hybrid_predictor_corrector_zeffprofile.nc → ...hybrid_predictor_corrector_zeffprofile.nc b/...hybrid_predictor_corrector_zeffprofile.nc → ...hybrid_predictor_corrector_zeffprofile.nc
diff --git a/...hybrid_predictor_corrector_zeffprofile.py → ...hybrid_predictor_corrector_zeffprofile.py b/...hybrid_predictor_corrector_zeffprofile.py → ...hybrid_predictor_corrector_zeffprofile.py
diff --git a/...edictor_corrector_zeffprofile_qualikiz.nc → ...edictor_corrector_zeffprofile_qualikiz.nc b/...edictor_corrector_zeffprofile_qualikiz.nc → ...edictor_corrector_zeffprofile_qualikiz.nc
diff --git a/...edictor_corrector_zeffprofile_qualikiz.py → ...edictor_corrector_zeffprofile_qualikiz.py b/...edictor_corrector_zeffprofile_qualikiz.py → ...edictor_corrector_zeffprofile_qualikiz.py
diff --git a/...tests/test_data/test_iterhybrid_rampup.nc → .../_src/test_data/test_iterhybrid_rampup.nc b/...tests/test_data/test_iterhybrid_rampup.nc → .../_src/test_data/test_iterhybrid_rampup.nc
diff --git a/...tests/test_data/test_iterhybrid_rampup.py → .../_src/test_data/test_iterhybrid_rampup.py b/...tests/test_data/test_iterhybrid_rampup.py → .../_src/test_data/test_iterhybrid_rampup.py
diff --git a/...st_data/test_iterhybrid_rampup_restart.py → ...st_data/test_iterhybrid_rampup_restart.py b/...st_data/test_iterhybrid_rampup_restart.py → ...st_data/test_iterhybrid_rampup_restart.py
diff --git a/...t_data/test_iterhybrid_rampup_sawtooth.nc → ...t_data/test_iterhybrid_rampup_sawtooth.nc b/...t_data/test_iterhybrid_rampup_sawtooth.nc → ...t_data/test_iterhybrid_rampup_sawtooth.nc
diff --git a/...t_data/test_iterhybrid_rampup_sawtooth.py → ...t_data/test_iterhybrid_rampup_sawtooth.py b/...t_data/test_iterhybrid_rampup_sawtooth.py → ...t_data/test_iterhybrid_rampup_sawtooth.py
diff --git a/...test_data/test_iterhybrid_rampup_short.py → ...test_data/test_iterhybrid_rampup_short.py b/...test_data/test_iterhybrid_rampup_short.py → ...test_data/test_iterhybrid_rampup_short.py
diff --git a/...ests/test_data/test_ne_qlknn_deff_veff.nc → ..._src/test_data/test_ne_qlknn_deff_veff.nc b/...ests/test_data/test_ne_qlknn_deff_veff.nc → ..._src/test_data/test_ne_qlknn_deff_veff.nc
diff --git a/...ests/test_data/test_ne_qlknn_deff_veff.py → ..._src/test_data/test_ne_qlknn_deff_veff.py b/...ests/test_data/test_ne_qlknn_deff_veff.py → ..._src/test_data/test_ne_qlknn_deff_veff.py
diff --git a/...sts/test_data/test_ne_qlknn_defromchie.nc → ...src/test_data/test_ne_qlknn_defromchie.nc b/...sts/test_data/test_ne_qlknn_defromchie.nc → ...src/test_data/test_ne_qlknn_defromchie.nc
diff --git a/...sts/test_data/test_ne_qlknn_defromchie.py → ...src/test_data/test_ne_qlknn_defromchie.py b/...sts/test_data/test_ne_qlknn_defromchie.py → ...src/test_data/test_ne_qlknn_defromchie.py
diff --git a/...ts/test_data/test_particle_sources_cgm.nc → ...rc/test_data/test_particle_sources_cgm.nc b/...ts/test_data/test_particle_sources_cgm.nc → ...rc/test_data/test_particle_sources_cgm.nc
diff --git a/...ts/test_data/test_particle_sources_cgm.py → ...rc/test_data/test_particle_sources_cgm.py b/...ts/test_data/test_particle_sources_cgm.py → ...rc/test_data/test_particle_sources_cgm.py
diff --git a/...test_prescribed_generic_current_source.nc → ...test_prescribed_generic_current_source.nc b/...test_prescribed_generic_current_source.nc → ...test_prescribed_generic_current_source.nc
diff --git a/...test_prescribed_generic_current_source.py → ...test_prescribed_generic_current_source.py b/...test_prescribed_generic_current_source.py → ...test_prescribed_generic_current_source.py
diff --git a/..._data/test_prescribed_timedependent_ne.nc → ..._data/test_prescribed_timedependent_ne.nc b/..._data/test_prescribed_timedependent_ne.nc → ..._data/test_prescribed_timedependent_ne.nc
diff --git a/..._data/test_prescribed_timedependent_ne.py → ..._data/test_prescribed_timedependent_ne.py b/..._data/test_prescribed_timedependent_ne.py → ..._data/test_prescribed_timedependent_ne.py
diff --git a/torax/tests/test_data/test_psi_and_heat.nc → torax/_src/test_data/test_psi_and_heat.nc b/torax/tests/test_data/test_psi_and_heat.nc → torax/_src/test_data/test_psi_and_heat.nc
diff --git a/torax/tests/test_data/test_psi_and_heat.py → torax/_src/test_data/test_psi_and_heat.py b/torax/tests/test_data/test_psi_and_heat.py → torax/_src/test_data/test_psi_and_heat.py
diff --git a/torax/tests/test_data/test_psi_heat_dens.nc → torax/_src/test_data/test_psi_heat_dens.nc b/torax/tests/test_data/test_psi_heat_dens.nc → torax/_src/test_data/test_psi_heat_dens.nc
diff --git a/torax/tests/test_data/test_psi_heat_dens.py → torax/_src/test_data/test_psi_heat_dens.py b/torax/tests/test_data/test_psi_heat_dens.py → torax/_src/test_data/test_psi_heat_dens.py
diff --git a/...st_data/test_psichease_ip_chease_vloop.nc → ...st_data/test_psichease_ip_chease_vloop.nc b/...st_data/test_psichease_ip_chease_vloop.nc → ...st_data/test_psichease_ip_chease_vloop.nc
diff --git a/...st_data/test_psichease_ip_chease_vloop.py → ...st_data/test_psichease_ip_chease_vloop.py b/...st_data/test_psichease_ip_chease_vloop.py → ...st_data/test_psichease_ip_chease_vloop.py
diff --git a/..._psichease_ip_parameters_vloop_varying.nc → ..._psichease_ip_parameters_vloop_varying.nc b/..._psichease_ip_parameters_vloop_varying.nc → ..._psichease_ip_parameters_vloop_varying.nc
diff --git a/..._psichease_ip_parameters_vloop_varying.py → ..._psichease_ip_parameters_vloop_varying.py b/..._psichease_ip_parameters_vloop_varying.py → ..._psichease_ip_parameters_vloop_varying.py
diff --git a/...st_data/test_psichease_prescribed_johm.nc → ...st_data/test_psichease_prescribed_johm.nc b/...st_data/test_psichease_prescribed_johm.nc → ...st_data/test_psichease_prescribed_johm.nc
diff --git a/...st_data/test_psichease_prescribed_johm.py → ...st_data/test_psichease_prescribed_johm.py b/...st_data/test_psichease_prescribed_johm.py → ...st_data/test_psichease_prescribed_johm.py
diff --git a/...st_data/test_psichease_prescribed_jtot.nc → ...st_data/test_psichease_prescribed_jtot.nc b/...st_data/test_psichease_prescribed_jtot.nc → ...st_data/test_psichease_prescribed_jtot.nc
diff --git a/...st_data/test_psichease_prescribed_jtot.py → ...st_data/test_psichease_prescribed_jtot.py b/...st_data/test_psichease_prescribed_jtot.py → ...st_data/test_psichease_prescribed_jtot.py
diff --git a/...a/test_psichease_prescribed_jtot_vloop.nc → ...a/test_psichease_prescribed_jtot_vloop.nc b/...a/test_psichease_prescribed_jtot_vloop.nc → ...a/test_psichease_prescribed_jtot_vloop.nc
diff --git a/...a/test_psichease_prescribed_jtot_vloop.py → ...a/test_psichease_prescribed_jtot_vloop.py b/...a/test_psichease_prescribed_jtot_vloop.py → ...a/test_psichease_prescribed_jtot_vloop.py
diff --git a/...test_data/test_semiimplicit_convection.nc → ...test_data/test_semiimplicit_convection.nc b/...test_data/test_semiimplicit_convection.nc → ...test_data/test_semiimplicit_convection.nc
diff --git a/...test_data/test_semiimplicit_convection.py → ...test_data/test_semiimplicit_convection.py b/...test_data/test_semiimplicit_convection.py → ...test_data/test_semiimplicit_convection.py
diff --git a/torax/tests/test_data/test_timedependence.nc → torax/_src/test_data/test_timedependence.nc b/torax/tests/test_data/test_timedependence.nc → torax/_src/test_data/test_timedependence.nc
diff --git a/torax/tests/test_data/test_timedependence.py → torax/_src/test_data/test_timedependence.py b/torax/tests/test_data/test_timedependence.py → torax/_src/test_data/test_timedependence.py
diff --git a/torax/_src/test_utils/paths.py b/torax/_src/test_utils/paths.py
@@ -24,6 +24,6 @@ def test_data_dir() -> pathlib.Path:
   """Absolute path of "test_data" directory."""
   src_dir = pathlib.Path(absltest.TEST_SRCDIR.value)
   torax_dir = 'torax/'
-  path = src_dir.joinpath(torax_dir, 'tests/test_data')
+  path = src_dir.joinpath(torax_dir, '_src/test_data')
   assert path.is_dir(), f'Path {path} is not a directory.'
   return path