def test_all_vars():
vars = ['longitude','latitude','mass','age','flag','id']
trajectory = nc_particles.nc_particle_file(data_file)
particle_count = trajectory.particle_count
data = trajectory.get_all_timesteps(variables = vars)
for key in vars:
d = data[key]
assert len(d) == trajectory.num_times
for i in range(len(d)):
assert len(d[i]) == particle_count[i]
def test_all_vars():
vars = ['longitude', 'latitude', 'mass', 'age', 'flag', 'id']
trajectory = nc_particles.nc_particle_file(data_file)
particle_count = trajectory.particle_count
data = trajectory.get_all_timesteps(variables=vars)
for key in vars:
d = data[key]
assert len(d) == trajectory.num_times
for i in range(len(d)):
assert len(d[i]) == particle_count[i]
def test_flag():
trajectory = nc_particles.nc_particle_file(data_file)
flag = trajectory.get_all_timesteps(variables=['flag'])['flag']
print flag
def test_num_LEs(): # should be ten at the end...
trajectory = nc_particles.nc_particle_file(data_file)
lat = trajectory.get_all_timesteps(variables=['latitude'])['latitude']
assert len(lat[-1]) == 10
def test_can_open():
trajectory = nc_particles.nc_particle_file(data_file)
def test_flag():
trajectory = nc_particles.nc_particle_file(data_file)
flag = trajectory.get_all_timesteps(variables=['flag'])['flag']
print flag
def test_num_LEs(): # should be ten at the end...
trajectory = nc_particles.nc_particle_file(data_file)
lat = trajectory.get_all_timesteps(variables=['latitude'])['latitude']
assert len(lat[-1]) == 10
def test_can_open():
trajectory = nc_particles.nc_particle_file(data_file)
