def test_vec2scal():
data = io.create_sample_dataset()
data.piv.vec2scal()
data.piv.vec2scal(property='curl')
data.piv.vec2scal(property='ke')
assert len(data.attrs['variables']) == 5
assert data.attrs['variables'][-1] == 'ke'
def test_set_get_dt():
""" tests setting the new dt """
data = io.create_sample_dataset()
assert data.attrs['dt'] == 1.0
assert data.piv.get_dt == 1.0
data.piv.set_dt(2.0)
assert data.attrs['dt'] == 2.0
def test_vec2scal():
data = io.create_sample_dataset()
data.piv.vec2scal()
data.piv.vec2scal(property="curl")
data.piv.vec2scal(property="ke")
assert len(data.attrs["variables"]) == 5
assert data.attrs["variables"][-1] == "ke"
def test_set_get_dt():
""" tests setting the new dt """
data = io.create_sample_dataset()
assert data.attrs['dt'] == 1.0
assert data.piv.get_dt == 1.0
data.piv.set_dt(2.0)
assert data.attrs['dt'] == 2.0
def test_crop():
""" tests crop """
_c = _a.piv.crop([5, 15, -5, -15])
assert _c.u.shape == (32, 32, 1)
_c = io.create_sample_dataset()
_c = _c.sel(x=slice(35, 70), y=slice(30, 90))
assert _c.u.shape == (2, 2, 5) # note the last dimension is preserved
def test_crop():
""" tests crop """
_c = _a.piv.crop([.6, 19.,-.6,-19.])
assert _c.u.shape == (59,59,1)
_c = io.create_sample_dataset()
_c = _c.sel(x = slice(30,90),y=slice(60,80))
assert _c.u.shape == (2,2,5) # note the last dimension is preserved
def test_crop():
""" tests crop """
_c = _a.piv.crop([.6, 19., -.6, -19.])
assert _c.u.shape == (59, 59, 1)
_c = io.create_sample_dataset()
_c = _c.sel(x=slice(30, 90), y=slice(60, 80))
assert _c.u.shape == (2, 2, 5) # note the last dimension is preserved
def load_data(data_selectbox='Demo'):
if data_selectbox == 'Demo':
data = io.create_sample_dataset()
elif data_selectbox == 'Jet':
data = io.load_directory(data_path, basename='Run*', ext='.vec')
elif data_selectbox == 'Canopy':
data = io.load_directory(os.path.join(data_path, 'urban_canopy'),
ext='.vc7')
return data
def test_shear():
""" tests shear estimate """
data = io.create_sample_field()
data.piv.shear()
assert data['w'][0,0] == 0.0
data = io.create_sample_dataset()
data.piv.vec2scal(property='ke')
data.piv.vec2scal(property='tke') # now defined
assert data.attrs['variables'][-1] == 'tke'
_a.piv.vec2scal(property='curl')
def test_multiply():
""" tests subtraction """
data = io.create_sample_dataset()
tmp = data * 3.5
assert tmp['u'][0, 0, 0] == 3.5
def test_subtract():
""" tests subtraction """
data = io.create_sample_dataset()
tmp = data - data
assert tmp['u'][0, 0, 0] == 0.0
def test_subtract():
""" tests subtraction """
data = io.create_sample_dataset()
tmp = data - data
assert tmp['u'][0,0,0] == 0.0
def test_vec2scal():
data = io.create_sample_dataset()
data.piv.vec2scal()
# tests that first data['w'] exists and then
# the first value is 0.0
assert data['w'][0,0,0] == 0.0
def test_mean():
data = io.create_sample_dataset(10)
assert np.allclose(data.piv.average.u.median(), 4.5)
def test_tke():
""" tests TKE """
data = io.create_sample_dataset()
data.piv.vec2scal(property="ke")
data.piv.vec2scal(property="tke") # now defined
assert data.attrs["variables"][-1] == "tke"
def test_mean():
data = io.create_sample_dataset(10)
assert data.piv.average.u.median() == 4.0
def test_mean():
data = io.create_sample_dataset(10)
assert data.piv.average.u.median() == 4.0
def test_add():
data = io.create_sample_dataset()
tmp = data + data
assert tmp['u'][0,0,0] == 2.0
def test_vec2scal():
data = io.create_sample_dataset()
data.piv.vec2scal()
# tests that first data['w'] exists and then
# the first value is 0.0
assert data['w'][0, 0, 0] == 0.0
def test_multiply():
""" tests subtraction """
data = io.create_sample_dataset()
tmp = data * 3.5
assert tmp['u'][0,0,0] == 3.5
def test_add():
data = io.create_sample_dataset()
tmp = data + data
assert tmp['u'][0, 0, 0] == 2.0
def test_create_sample_dataset(n=3):
data = io.create_sample_dataset(n=n)
assert data.dims["t"] == 3
assert np.allclose(data["t"], np.arange(3))
def test_create_sample_dataset(n=3):
data = io.create_sample_dataset(n=n)
assert data.dims['t'] == 3
assert np.allclose(data['t'],np.arange(3))