Skip to content

Commit e3a38d0

Browse files
committed
Fix remaining tests
1 parent b78de00 commit e3a38d0

2 files changed

Lines changed: 11 additions & 18 deletions

File tree

tests/test_index_search.py

Lines changed: 4 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -3,23 +3,17 @@
33
import xgcm
44

55
import parcels.tutorial
6-
from parcels import Field, XGrid
6+
from parcels import XGrid
7+
from parcels._core.fieldset import FieldSet
78
from parcels._core.index_search import _latlon_rad_to_xyz, _search_indices_curvilinear_2d
89
from parcels._datasets.structured.generic import datasets
9-
from parcels.interpolators import XLinear
1010

1111

1212
@pytest.fixture
1313
def field_cone():
1414
ds = datasets["2d_left_unrolled_cone"]
15-
grid = XGrid.from_dataset(ds, mesh="flat")
16-
field = Field(
17-
name="test_field",
18-
data=ds["data_g"],
19-
grid=grid,
20-
interp_method=XLinear,
21-
)
22-
return field
15+
fieldset = FieldSet.from_sgrid_conventions(ds, mesh="flat")
16+
return fieldset.data_g
2317

2418

2519
def test_grid_indexing_fpoints(field_cone):

tests/test_particleset_execute.py

Lines changed: 7 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -17,7 +17,6 @@
1717
UxGrid,
1818
Variable,
1919
VectorField,
20-
XGrid,
2120
)
2221
from parcels._core.utils.time import timedelta_to_float
2322
from parcels._datasets.structured.generated import simple_UV_dataset
@@ -27,8 +26,6 @@
2726
Ux_Velocity,
2827
UxConstantFaceConstantZC,
2928
UxLinearNodeLinearZF,
30-
XLinear,
31-
XLinear_Velocity,
3229
)
3330
from parcels.kernels import AdvectionEE, AdvectionRK2, AdvectionRK4, AdvectionRK4_3D, AdvectionRK45
3431
from tests.common_kernels import DoNothing
@@ -40,11 +37,13 @@ def fieldset_no_time_interval() -> FieldSet:
4037
# i.e., no time variation
4138
ds = datasets_structured["ds_2d_left"].isel(time=0).drop_vars("time")
4239

43-
grid = XGrid.from_dataset(ds, mesh="flat")
44-
U = Field("U", ds["U_A_grid"], grid, interp_method=XLinear)
45-
V = Field("V", ds["V_A_grid"], grid, interp_method=XLinear)
46-
UV = VectorField("UV", U, V, vector_interp_method=XLinear_Velocity)
47-
return FieldSet([U, V, UV])
40+
ds = ds[["U_A_grid", "V_A_grid", "grid"]].rename(
41+
{
42+
"U_A_grid": "U",
43+
"V_A_grid": "V",
44+
}
45+
)
46+
return FieldSet.from_sgrid_conventions(ds, mesh="flat")
4847

4948

5049
@pytest.fixture

0 commit comments

Comments
 (0)