add test

Author: philipc2

test/test_cross_sections.py

@@ -3,6 +3,7 @@
 import numpy as np
 from pathlib import Path
 import os
+import xarray as xr
 
 import numpy.testing as nt
 
@@ -13,82 +14,89 @@
 quad_hex_node_data = current_path / 'meshfiles' / "ugrid" / "quad-hexagon" / 'random-node-data.nc'
 cube_sphere_grid = current_path / "meshfiles" / "ugrid" / "outCSne30" / "outCSne30.ug"
 
+
+csne8_grid = current_path / "meshfiles" / "scrip" / "ne30pg2" / "grid.nc"
+csne8_data = current_path / "meshfiles" / "scrip" / "ne30pg2" / "data.nc"
+
+
+
+
 from uxarray.grid.intersections import constant_lat_intersections_face_bounds
 
 
 
 
 
-def test_constant_lat_cross_section_grid():
+def test_constant_lat_subset_grid():
     uxgrid = ux.open_grid(quad_hex_grid_path)
 
-    grid_top_two = uxgrid.cross_section.constant_latitude(lat=0.1)
+    grid_top_two = uxgrid.subset.constant_latitude(lat=0.1)
     assert grid_top_two.n_face == 2
 
-    grid_bottom_two = uxgrid.cross_section.constant_latitude(lat=-0.1)
+    grid_bottom_two = uxgrid.subset.constant_latitude(lat=-0.1)
     assert grid_bottom_two.n_face == 2
 
-    grid_all_four = uxgrid.cross_section.constant_latitude(lat=0.0)
+    grid_all_four = uxgrid.subset.constant_latitude(lat=0.0)
     assert grid_all_four.n_face == 4
 
     with pytest.raises(ValueError):
-        uxgrid.cross_section.constant_latitude(lat=10.0)
+        uxgrid.subset.constant_latitude(lat=10.0)
 
-def test_constant_lon_cross_section_grid():
+def test_constant_lon_subset_grid():
     uxgrid = ux.open_grid(quad_hex_grid_path)
 
-    grid_left_two = uxgrid.cross_section.constant_longitude(lon=-0.1)
+    grid_left_two = uxgrid.subset.constant_longitude(lon=-0.1)
     assert grid_left_two.n_face == 2
 
-    grid_right_two = uxgrid.cross_section.constant_longitude(lon=0.2)
+    grid_right_two = uxgrid.subset.constant_longitude(lon=0.2)
     assert grid_right_two.n_face == 2
 
     with pytest.raises(ValueError):
-        uxgrid.cross_section.constant_longitude(lon=10.0)
+        uxgrid.subset.constant_longitude(lon=10.0)
 
-def test_constant_lat_cross_section_uxds():
+def test_constant_lat_subset_uxds():
     uxds = ux.open_dataset(quad_hex_grid_path, quad_hex_data_path)
     uxds.uxgrid.normalize_cartesian_coordinates()
 
-    da_top_two = uxds['t2m'].cross_section.constant_latitude(lat=0.1)
+    da_top_two = uxds['t2m'].subset.constant_latitude(lat=0.1)
     np.testing.assert_array_equal(da_top_two.data, uxds['t2m'].isel(n_face=[1, 2]).data)
 
-    da_bottom_two = uxds['t2m'].cross_section.constant_latitude(lat=-0.1)
+    da_bottom_two = uxds['t2m'].subset.constant_latitude(lat=-0.1)
     np.testing.assert_array_equal(da_bottom_two.data, uxds['t2m'].isel(n_face=[0, 3]).data)
 
-    da_all_four = uxds['t2m'].cross_section.constant_latitude(lat=0.0)
+    da_all_four = uxds['t2m'].subset.constant_latitude(lat=0.0)
     np.testing.assert_array_equal(da_all_four.data, uxds['t2m'].data)
 
     with pytest.raises(ValueError):
-        uxds['t2m'].cross_section.constant_latitude(lat=10.0)
+        uxds['t2m'].subset.constant_latitude(lat=10.0)
 
-def test_constant_lon_cross_section_uxds():
+def test_constant_lon_subset_uxds():
     uxds = ux.open_dataset(quad_hex_grid_path, quad_hex_data_path)
     uxds.uxgrid.normalize_cartesian_coordinates()
 
-    da_left_two = uxds['t2m'].cross_section.constant_longitude(lon=-0.1)
+    da_left_two = uxds['t2m'].subset.constant_longitude(lon=-0.1)
     np.testing.assert_array_equal(da_left_two.data, uxds['t2m'].isel(n_face=[0, 2]).data)
 
-    da_right_two = uxds['t2m'].cross_section.constant_longitude(lon=0.2)
+    da_right_two = uxds['t2m'].subset.constant_longitude(lon=0.2)
     np.testing.assert_array_equal(da_right_two.data, uxds['t2m'].isel(n_face=[1, 3]).data)
 
     with pytest.raises(ValueError):
-        uxds['t2m'].cross_section.constant_longitude(lon=10.0)
+        uxds['t2m'].subset.constant_longitude(lon=10.0)
 
 def test_north_pole():
     uxgrid = ux.open_grid(cube_sphere_grid)
     lats = [89.85, 89.9, 89.95, 89.99]
 
     for lat in lats:
-        cross_grid = uxgrid.cross_section.constant_latitude(lat=lat)
+        cross_grid = uxgrid.subset.constant_latitude(lat=lat)
         assert cross_grid.n_face == 4
 
 def test_south_pole():
     uxgrid = ux.open_grid(cube_sphere_grid)
     lats = [-89.85, -89.9, -89.95, -89.99]
 
     for lat in lats:
-        cross_grid = uxgrid.cross_section.constant_latitude(lat=lat)
+        cross_grid = uxgrid.subset.constant_latitude(lat=lat)
         assert cross_grid.n_face == 4
 
 def test_constant_lat():
@@ -130,39 +138,39 @@ def test_const_lat_interval_da():
     uxds = ux.open_dataset(quad_hex_grid_path, quad_hex_data_path)
     uxds.uxgrid.normalize_cartesian_coordinates()
 
-    res = uxds['t2m'].cross_section.constant_latitude_interval(lats=(-10, 10))
+    res = uxds['t2m'].subset.constant_latitude_interval(lats=(-10, 10))
 
     assert len(res) == 4
 
 
 def test_const_lat_interval_grid():
     uxgrid = ux.open_grid(quad_hex_grid_path)
 
-    res = uxgrid.cross_section.constant_latitude_interval(lats=(-10, 10))
+    res = uxgrid.subset.constant_latitude_interval(lats=(-10, 10))
 
     assert res.n_face == 4
 
-    res, indices = uxgrid.cross_section.constant_latitude_interval(lats=(-10, 10), return_face_indices=True)
+    res, indices = uxgrid.subset.constant_latitude_interval(lats=(-10, 10), return_face_indices=True)
 
     assert len(indices) == 4
 
 def test_const_lon_interva_da():
     uxds = ux.open_dataset(quad_hex_grid_path, quad_hex_data_path)
     uxds.uxgrid.normalize_cartesian_coordinates()
 
-    res = uxds['t2m'].cross_section.constant_longitude_interval(lons=(-10, 10))
+    res = uxds['t2m'].subset.constant_longitude_interval(lons=(-10, 10))
 
     assert len(res) == 4
 
 
 def test_const_lon_interval_grid():
     uxgrid = ux.open_grid(quad_hex_grid_path)
 
-    res = uxgrid.cross_section.constant_longitude_interval(lons=(-10, 10))
+    res = uxgrid.subset.constant_longitude_interval(lons=(-10, 10))
 
     assert res.n_face == 4
 
-    res, indices = uxgrid.cross_section.constant_longitude_interval(lons=(-10, 10), return_face_indices=True)
+    res, indices = uxgrid.subset.constant_longitude_interval(lons=(-10, 10), return_face_indices=True)
 
     assert len(indices) == 4
 
@@ -185,22 +193,49 @@ def test_latitude_along_arc(self):
 
 
 
-def test_double_cross_section():
+def test_double_subset():
     uxgrid = ux.open_grid(quad_hex_grid_path)
 
     # construct edges
-    sub_lat = uxgrid.cross_section.constant_latitude(0.0)
+    sub_lat = uxgrid.subset.constant_latitude(0.0)
 
-    sub_lat_lon = sub_lat.cross_section.constant_longitude(0.0)
+    sub_lat_lon = sub_lat.subset.constant_longitude(0.0)
 
     assert "n_edge" not in sub_lat_lon._ds.dims
 
     _ = uxgrid.face_edge_connectivity
     _ = uxgrid.edge_node_connectivity
     _ = uxgrid.edge_lon
 
-    sub_lat = uxgrid.cross_section.constant_latitude(0.0)
+    sub_lat = uxgrid.subset.constant_latitude(0.0)
 
-    sub_lat_lon = sub_lat.cross_section.constant_longitude(0.0)
+    sub_lat_lon = sub_lat.subset.constant_longitude(0.0)
 
     assert "n_edge" in sub_lat_lon._ds.dims
+
+
+def test_cross_section():
+    uxds = ux.open_dataset(csne8_grid, csne8_data)
+
+    # Tributary GCA
+    ss_gca = uxds['RELHUM'].cross_section(start=(-45, -45), end=(45, 45))
+    assert isinstance(ss_gca, xr.DataArray)
+
+    # Constant Latitude
+    ss_clat = uxds['RELHUM'].cross_section(lat=45)
+    assert isinstance(ss_clat, xr.DataArray)
+
+    # Constat Longitude
+    ss_clon = uxds['RELHUM'].cross_section(lon=45)
+    assert isinstance(ss_clon, xr.DataArray)
+
+    # Constat Longitude with increased samples
+    ss_clon = uxds['RELHUM'].cross_section(lon=45, steps=3)
+    assert isinstance(ss_clon, xr.DataArray)
+
+
+    with pytest.raises(ValueError):
+        _ = uxds['RELHUM'].cross_section(end=(45, 45))
+        _ = uxds['RELHUM'].cross_section(start=(45, 45))
+        _ = uxds['RELHUM'].cross_section(lon = 45, end=(45, 45))
+        _ = uxds['RELHUM'].cross_section()