예제 #1
0
    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'
예제 #2
0
    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'
예제 #3
0
    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))
예제 #4
0
    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))
예제 #5
0
    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]
예제 #6
0
    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]