Skip to content

Commit 75c84fa

Browse files
Fixing test_grid unit tests to not use transpose
1 parent ddd647f commit 75c84fa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_grids.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -63,7 +63,7 @@ def temp_func(lon, lat):
6363
temp1_data = np.empty((time_g1.size, lat_g1.size, lon_g1.size), dtype=np.float32)
6464
for i in range(lon_g1.size):
6565
for j in range(lat_g1.size):
66-
temp1_data[i, j, :] = temp_func(lon_g1[i], lat_g1[j])
66+
temp1_data[:, j, i] = temp_func(lon_g1[i], lat_g1[j])
6767
temp1_field = Field("temp1", temp1_data, grid=grid_1)
6868

6969
other_fields = {}
@@ -347,7 +347,7 @@ def test_curvilinear_grids():
347347

348348
u_data = np.ones((2, y.size, x.size), dtype=np.float32)
349349
v_data = np.zeros((2, y.size, x.size), dtype=np.float32)
350-
u_data[:, :, 0] = lon[:, :] + lat[:, :]
350+
u_data[0, :, :] = lon[:, :] + lat[:, :]
351351
u_field = Field("U", u_data, grid=grid)
352352
v_field = Field("V", v_data, grid=grid)
353353
fieldset = FieldSet(u_field, v_field)

0 commit comments

Comments
 (0)