Skip to content

Commit 5a68537

Browse files
committed
Rename gridadapter.py to xgrid.py
1 parent 4491677 commit 5a68537

3 files changed

Lines changed: 20 additions & 20 deletions

File tree

parcels/field.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -18,7 +18,6 @@
1818
VectorType,
1919
assert_valid_mesh,
2020
)
21-
from parcels.gridadapter import XGrid
2221
from parcels.tools.converters import (
2322
UnitConverter,
2423
unitconverters_map,
@@ -32,6 +31,7 @@
3231
)
3332
from parcels.uxgrid import UxGrid
3433
from parcels.xgcm.grid import Grid
34+
from parcels.xgrid import XGrid
3535

3636
from ._index_search import _search_time_index
3737

Lines changed: 16 additions & 16 deletions
Original file line numberDiff line numberDiff line change
@@ -37,16 +37,16 @@ class XGrid:
3737
"""
3838

3939
def __init__(self, grid: xgcm.Grid, mesh="flat"):
40-
self.grid = grid
40+
self.xgcm_grid = grid
4141
self.mesh = mesh
4242

4343
# ! Not ideal... Triggers computation on a throwaway item. If adapter is still needed in codebase, and this is prohibitively expensive, perhaps store GridAdapter on Field object instead of Grid
4444
self.lonlat_minmax = np.array(
4545
[
46-
np.nanmin(self.grid._ds["lon"]),
47-
np.nanmax(self.grid._ds["lon"]),
48-
np.nanmin(self.grid._ds["lat"]),
49-
np.nanmax(self.grid._ds["lat"]),
46+
np.nanmin(self.xgcm_grid._ds["lon"]),
47+
np.nanmax(self.xgcm_grid._ds["lon"]),
48+
np.nanmin(self.xgcm_grid._ds["lat"]),
49+
np.nanmax(self.xgcm_grid._ds["lat"]),
5050
]
5151
)
5252

@@ -59,10 +59,10 @@ def lon(self):
5959
TODO v4: Evaluate
6060
"""
6161
try:
62-
_ = self.grid.axes["X"]
62+
_ = self.xgcm_grid.axes["X"]
6363
except KeyError:
6464
return np.zeros(1)
65-
return self.grid._ds["lon"].values
65+
return self.xgcm_grid._ds["lon"].values
6666

6767
@property
6868
def lat(self):
@@ -73,10 +73,10 @@ def lat(self):
7373
TODO v4: Evaluate
7474
"""
7575
try:
76-
_ = self.grid.axes["Y"]
76+
_ = self.xgcm_grid.axes["Y"]
7777
except KeyError:
7878
return np.zeros(1)
79-
return self.grid._ds["lat"].values
79+
return self.xgcm_grid._ds["lat"].values
8080

8181
@property
8282
def depth(self):
@@ -87,15 +87,15 @@ def depth(self):
8787
TODO v4: Evaluate
8888
"""
8989
try:
90-
_ = self.grid.axes["Z"]
90+
_ = self.xgcm_grid.axes["Z"]
9191
except KeyError:
9292
return np.zeros(1)
93-
return self.grid._ds["depth"].values
93+
return self.xgcm_grid._ds["depth"].values
9494

9595
@property
9696
def _datetimes(self):
9797
try:
98-
axis = self.grid.axes["T"]
98+
axis = self.xgcm_grid.axes["T"]
9999
except KeyError:
100100
return np.zeros(1)
101101
return get_time(axis)
@@ -106,19 +106,19 @@ def time(self):
106106

107107
@property
108108
def xdim(self):
109-
return get_dimensionality(self.grid.axes.get("X"))
109+
return get_dimensionality(self.xgcm_grid.axes.get("X"))
110110

111111
@property
112112
def ydim(self):
113-
return get_dimensionality(self.grid.axes.get("Y"))
113+
return get_dimensionality(self.xgcm_grid.axes.get("Y"))
114114

115115
@property
116116
def zdim(self):
117-
return get_dimensionality(self.grid.axes.get("Z"))
117+
return get_dimensionality(self.xgcm_grid.axes.get("Z"))
118118

119119
@property
120120
def tdim(self):
121-
return get_dimensionality(self.grid.axes.get("T"))
121+
return get_dimensionality(self.xgcm_grid.axes.get("T"))
122122

123123
@property
124124
def time_origin(self):
Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,8 @@
77
from parcels import xgcm
88
from parcels._datasets.structured.generic import T, X, Y, Z, datasets
99
from parcels.grid import Grid as OldGrid
10-
from parcels.gridadapter import XGrid
1110
from parcels.tools.converters import TimeConverter
11+
from parcels.xgrid import XGrid
1212

1313
GridTestCase = namedtuple("GridTestCase", ["Grid", "attr", "expected"])
1414

@@ -38,7 +38,7 @@ def assert_equal(actual, expected):
3838

3939

4040
@pytest.mark.parametrize("ds, attr, expected", test_cases)
41-
def test_grid_adapter_properties_ground_truth(ds, attr, expected):
41+
def test_xgrid_properties_ground_truth(ds, attr, expected):
4242
adapter = XGrid(xgcm.Grid(ds, periodic=False))
4343
actual = getattr(adapter, attr)
4444
assert_equal(actual, expected)
@@ -60,7 +60,7 @@ def test_grid_adapter_properties_ground_truth(ds, attr, expected):
6060
],
6161
)
6262
@pytest.mark.parametrize("ds", datasets.values())
63-
def test_grid_adapter_against_old(ds, attr):
63+
def test_xgrid_against_old(ds, attr):
6464
adapter = XGrid(xgcm.Grid(ds, periodic=False))
6565

6666
grid = OldGrid.create_grid(

0 commit comments

Comments
 (0)