def test_construction(self): curr_file = os.path.join(s_data, 'staggered_sine_channel.nc') u = Variable.from_netCDF(filename=curr_file, varname='u_rho') v = Variable.from_netCDF(filename=curr_file, varname='v_rho') gvp = VectorVariable(name='velocity', units='m/s', time=u.time, variables=[u, v]) assert gvp.name == 'velocity' assert gvp.units == 'm/s' assert gvp.varnames[0] == 'u_rho'
def test_at(self): curr_file = os.path.join(s_data, 'staggered_sine_channel.nc') u = Variable.from_netCDF(filename=curr_file, varname='u_rho') v = Variable.from_netCDF(filename=curr_file, varname='v_rho') points = np.array(([0, 0, 0], [np.pi, 1, 0], [2 * np.pi, 0, 0])) time = dt.datetime.now() assert all(u.at(points, time) == [1, 1, 1]) print np.cos(points[:, 0] / 2) / 2 assert all(np.isclose(v.at(points, time), np.cos(points[:, 0] / 2) / 2))
def test_construction(self): data = sinusoid['u'][:] grid = PyGrid.from_netCDF(dataset=sinusoid) time = None u = Variable(name='u', units='m/s', data=data, grid=grid, time=time, data_file='staggered_sine_channel.nc', grid_file='staggered_sine_channel.nc') curr_file = os.path.join(s_data, 'staggered_sine_channel.nc') k = Variable.from_netCDF(filename=curr_file, varname='u', name='u') assert k.name == u.name assert k.units == 'm/s' # fixme: this was failing # assert k.time == u.time assert k.data[0, 0] == u.data[0, 0]