Add unit tests to ensure dataset copy works

Author: xylar

conda_package/tests/test_conversion.py

@@ -4,7 +4,7 @@
 import numpy as np
 
 from mpas_tools.io import write_netcdf
-from mpas_tools.mesh.conversion import convert, cull, mask
+from mpas_tools.mesh.conversion import _masks_to_int, convert, cull, mask
 from mpas_tools.mesh.spherical import recompute_angle_edge
 
 from .util import get_test_data_file
@@ -47,6 +47,27 @@ def test_conversion_angle_edge():
     assert np.max(np.abs(angle_diff)) < 1.0e-10
 
 
+def test_masks_to_int_dataset_copy():
+    ds_in = xarray.Dataset(
+        data_vars={
+            'regionCellMasks': (
+                ('nCells', 'nRegions'),
+                np.array([[True, False], [False, True]]),
+            ),
+            'cullCell': (('nCells',), np.array([False, True])),
+            'xCell': (('nCells',), np.array([1.0, 2.0])),
+        },
+        attrs={'meshName': 'unit-test'},
+    )
+
+    ds_out = _masks_to_int(ds_in)
+
+    assert ds_out.regionCellMasks.dtype == np.int32
+    assert ds_out.cullCell.dtype == np.int32
+    assert ds_out.attrs == ds_in.attrs
+    assert np.array_equal(ds_out.xCell.values, ds_in.xCell.values)
+
+
 if __name__ == '__main__':
     test_conversion()
     test_conversion_angle_edge()

conda_package/tests/test_viz_transects.py

@@ -1,11 +1,13 @@
 #!/usr/bin/env python
+import numpy as np
 import xarray as xr
 
 from mpas_tools.cime.constants import constants
 from mpas_tools.mesh.conversion import convert, cull
 from mpas_tools.planar_hex import make_planar_hex_mesh
 from mpas_tools.translate import center
 from mpas_tools.viz.mesh_to_triangles import mesh_to_triangles
+from mpas_tools.viz.transect.horiz import _fix_periodic_tris
 from mpas_tools.viz.transect import (
     find_planar_transect_cells_and_weights,
     find_spherical_transect_cells_and_weights,
@@ -130,6 +132,24 @@ def test_find_planar_transect_cells_and_weights():
         assert var in ds_transect.data_vars
 
 
+def test_fix_periodic_tris_dataset_copy():
+    ds_tris = xr.Dataset(
+        data_vars={
+            'xNode': (
+                ('nTriangles', 'nNodes'),
+                np.array([[0.0, 8.0, -8.0]]),
+            ),
+        }
+    )
+
+    ds_new = _fix_periodic_tris(ds_tris, 'xNode', period=10.0)
+
+    assert ds_new.sizes['nTriangles'] == 3
+    assert np.allclose(ds_new.xNode.values[0, :], [0.0, -2.0, 2.0])
+    assert np.allclose(ds_new.xNode.values[1, :], [10.0, 8.0, 12.0])
+    assert np.allclose(ds_new.xNode.values[2, :], [-10.0, -12.0, -8.0])
+
+
 def _get_triangles():
     ds_mesh = xr.open_dataset(get_test_data_file('mesh.QU.1920km.151026.nc'))
     earth_radius = constants['SHR_CONST_REARTH']