Use a utility to find test data files

Author: xylar

conda_package/tests/test_conversion.py

@@ -5,26 +5,26 @@
 from mpas_tools.io import write_netcdf
 from mpas_tools.mesh.conversion import convert, cull, mask
 
+from .util import get_test_data_file
+
 matplotlib.use('Agg')
 import xarray
 from geometric_features import read_feature_collection
 
 
 def test_conversion():
     dsMesh = xarray.open_dataset(
-        'mesh_tools/mesh_conversion_tools/test/mesh.QU.1920km.151026.nc'
+        get_test_data_file('mesh.QU.1920km.151026.nc')
     )
     dsMesh = convert(dsIn=dsMesh)
     write_netcdf(dsMesh, 'mesh.nc')
 
-    dsMask = xarray.open_dataset(
-        'mesh_tools/mesh_conversion_tools/test/land_mask_final.nc'
-    )
+    dsMask = xarray.open_dataset(get_test_data_file('land_mask_final.nc'))
     dsCulled = cull(dsIn=dsMesh, dsMask=dsMask)
     write_netcdf(dsCulled, 'culled_mesh.nc')
 
     fcMask = read_feature_collection(
-        'mesh_tools/mesh_conversion_tools/test/Arctic_Ocean.geojson'
+        get_test_data_file('Arctic_Ocean.geojson')
     )
     dsMask = mask(dsMesh=dsMesh, fcMask=fcMask)
     write_netcdf(dsMask, 'antarctic_mask.nc')

conda_package/tests/test_depth.py

@@ -13,14 +13,16 @@
     write_time_varying_zmid,
 )
 
+from .util import get_test_data_file
+
 
 def create_3d_mesh():
     outFileName = 'test_depth_mesh.nc'
     if os.path.exists(outFileName):
         dsMesh = xarray.open_dataset(outFileName)
     else:
         dsMesh = xarray.open_dataset(
-            'mesh_tools/mesh_conversion_tools/test/mesh.QU.1920km.151026.nc'
+            get_test_data_file('mesh.QU.1920km.151026.nc')
         )
         nCells = dsMesh.sizes['nCells']
         nVertLevels = 10

conda_package/tests/test_mesh_mask.py

@@ -18,6 +18,8 @@
     compute_projection_grid_region_masks,
 )
 
+from .util import get_test_data_file
+
 
 def test_compute_mpas_region_masks():
     ds_mesh, _ = _get_mesh()
@@ -205,9 +207,7 @@ def _get_pool():
 
 
 def _get_mesh():
-    ds_mesh = xr.open_dataset(
-        'mesh_tools/mesh_conversion_tools/test/mesh.QU.1920km.151026.nc'
-    )
+    ds_mesh = xr.open_dataset(get_test_data_file('mesh.QU.1920km.151026.nc'))
     earth_radius = constants['SHR_CONST_REARTH']
     ds_mesh.attrs['sphere_radius'] = earth_radius
     for coord in [

conda_package/tests/test_viz_transects.py

@@ -12,6 +12,8 @@
     make_triangle_tree,
 )
 
+from .util import get_test_data_file
+
 
 def test_mesh_to_triangles():
     _, ds_tris = _get_triangles()
@@ -129,9 +131,7 @@ def test_find_planar_transect_cells_and_weights():
 
 
 def _get_triangles():
-    ds_mesh = xr.open_dataset(
-        'mesh_tools/mesh_conversion_tools/test/mesh.QU.1920km.151026.nc'
-    )
+    ds_mesh = xr.open_dataset(get_test_data_file('mesh.QU.1920km.151026.nc'))
     earth_radius = constants['SHR_CONST_REARTH']
     ds_mesh.attrs['sphere_radius'] = earth_radius
     for coord in [

conda_package/tests/util.py

@@ -0,0 +1,31 @@
+import os
+
+
+def get_test_data_file(filename):
+    """
+    Get the full path to a data file in the tests/data directory.
+
+    Parameters
+    ----------
+    filename : str
+        The name of the data file.
+
+    Returns
+    -------
+    str
+        The full relative path to the data file.
+    """
+
+    local_path = os.path.join(
+        'mesh_tools', 'mesh_conversion_tools', 'test', filename
+    )
+    repo_path = os.path.join('..', '..', local_path)
+    if os.path.exists(local_path):
+        return local_path
+    elif os.path.exists(repo_path):
+        return repo_path
+
+    raise FileNotFoundError(
+        f"Data file '{filename}' not found in expected locations: "
+        f'{local_path} or {repo_path}'
+    )