def test_pfile_array_remove_particles(fieldset, mode, tmpdir, npart=10):
filepath = tmpdir.join("pfile_array_remove_particles")
pset = ParticleSet(fieldset, pclass=ptype[mode],
lon=np.linspace(0, 1, npart, dtype=np.float32),
lat=0.5*np.ones(npart, dtype=np.float32))
pfile = pset.ParticleFile(filepath)
pfile.write(pset, 0)
pset.remove(3)
pfile.write(pset, 1)
def test_pfile_array_remove_particles(fieldset, mode, tmpdir, npart=10):
filepath = tmpdir.join("pfile_array_remove_particles.nc")
pset = ParticleSet(fieldset,
pclass=ptype[mode],
lon=np.linspace(0, 1, npart),
lat=0.5 * np.ones(npart))
pfile = pset.ParticleFile(filepath)
pfile.write(pset, 0)
pset.remove(3)
pfile.write(pset, 1)
ncfile = close_and_compare_netcdffiles(filepath, pfile)
ncfile.close()
def test_pset_multi_execute_delete(fieldset, mode, npart=10, n=5):
def AddLat(particle, fieldset, time):
particle.lat += 0.1
pset = ParticleSet(fieldset, pclass=ptype[mode],
lon=np.linspace(0, 1, npart, dtype=np.float32),
lat=np.zeros(npart, dtype=np.float32))
k_add = pset.Kernel(AddLat)
for _ in range(n):
pset.execute(k_add, runtime=1., dt=1.0)
pset.remove(-1)
assert np.allclose([p.lat - n*0.1 for p in pset], np.zeros(npart - n), rtol=1e-12)
def test_pfile_array_remove_all_particles(fieldset, mode, tmpdir, npart=10):
filepath = tmpdir.join("pfile_array_remove_particles")
pset = ParticleSet(fieldset, pclass=ptype[mode],
lon=np.linspace(0, 1, npart),
lat=0.5*np.ones(npart))
pfile = pset.ParticleFile(filepath)
pfile.write(pset, 0)
for _ in range(npart):
pset.remove(-1)
pfile.write(pset, 1)
pfile.write(pset, 2)
def test_pset_remove_particle(fieldset, mode, npart=100):
lon = np.linspace(0, 1, npart, dtype=np.float32)
lat = np.linspace(1, 0, npart, dtype=np.float32)
pset = ParticleSet(fieldset, lon=lon, lat=lat, pclass=ptype[mode])
for ilon, ilat in zip(lon[::-1], lat[::-1]):
p = pset.remove(pset[-1])
assert (p.lon == ilon)
assert (p.lat == ilat)
assert (pset.size == 0)
def test_pset_remove_index(fieldset, mode, npart=100):
lon = np.linspace(0, 1, npart)
lat = np.linspace(1, 0, npart)
pset = ParticleSet(fieldset, lon=lon, lat=lat, pclass=ptype[mode], lonlatdepth_dtype=np.float64)
for ilon, ilat in zip(lon[::-1], lat[::-1]):
p = pset.remove(-1)
assert(p.lon == ilon)
assert(p.lat == ilat)
assert(pset.size == 0)