Skip to content

Commit 4d03c0e

Browse files
Adding unit tests that seemed to break on VirtualShip
But work on v4-dev itself. Need to explore further
1 parent 0e236db commit 4d03c0e

File tree

2 files changed

+28
-0
lines changed

2 files changed

+28
-0
lines changed

tests/test_interpolation.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -91,6 +91,12 @@ def test_raw_2d_interpolation(field, func, t, z, y, x, expected):
9191
np.testing.assert_equal(value, expected)
9292

9393

94+
def test_scalar_field_eval(field):
95+
UV = VectorField("UV", field, field)
96+
97+
UV.eval(np.timedelta64(2, "s"), 2, 1.5, 0.5)
98+
99+
94100
@pytest.mark.parametrize(
95101
"func, t, z, y, x, expected",
96102
[

tests/test_particleset_execute.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -150,6 +150,28 @@ def test_particleset_endtime_type(fieldset, endtime, expectation):
150150
pset.execute(endtime=endtime, dt=np.timedelta64(10, "m"), pyfunc=DoNothing)
151151

152152

153+
def test_particleset_run_to_endtime(fieldset):
154+
starttime = fieldset.time_interval.left
155+
endtime = fieldset.time_interval.right
156+
157+
def SampleU(particles, fieldset): # pragma: no cover
158+
_ = fieldset.U[particles]
159+
160+
pset = ParticleSet(fieldset, lon=[0.2], lat=[5.0], time=[starttime])
161+
pset.execute(SampleU, endtime=endtime, dt=np.timedelta64(1, "D"))
162+
assert pset[0].time + pset[0].dt == endtime
163+
164+
165+
def test_particleset_interpolate_domainedge(fieldset):
166+
def SampleU(particles, fieldset): # pragma: no cover
167+
particles.dlon = fieldset.U[particles]
168+
169+
print(fieldset.U.grid.lon)
170+
pset = ParticleSet(fieldset, lon=fieldset.U.grid.lon[0], lat=fieldset.U.grid.lat[0])
171+
pset.execute(SampleU, runtime=np.timedelta64(1, "D"), dt=np.timedelta64(1, "D"))
172+
assert np.isfinite(pset[0].dlon)
173+
174+
153175
@pytest.mark.parametrize(
154176
"dt", [np.timedelta64(1, "s"), np.timedelta64(1, "ms"), np.timedelta64(10, "ms"), np.timedelta64(1, "ns")]
155177
)

0 commit comments

Comments
 (0)