Пример #1
0
    def test_variable_size_check():
        r'Test Variable checking size of passed array'
        ds = Dataset()
        xdim = ds.createDimension('x', 2)
        ydim = ds.createDimension('y', 3)

        # Create array with dims flipped
        arr = np.empty((ydim.size, xdim.size), dtype='f4')

        ds.createVariable('data', 'f4', ('x', 'y'), wrap_array=arr)
Пример #2
0
def test_variable_size_check():
    r'Test Variable checking size of passed array'
    ds = Dataset()
    xdim = ds.createDimension('x', 2)
    ydim = ds.createDimension('y', 3)

    # Create array with dims flipped
    arr = np.empty((ydim.size, xdim.size), dtype='f4')

    with pytest.raises(ValueError):
        ds.createVariable('data', 'f4', ('x', 'y'), wrap_array=arr)
Пример #3
0
def test_variable_size_check():
    """Test `Variable` checking size of passed array."""
    ds = Dataset()
    xdim = ds.createDimension('x', 2)
    ydim = ds.createDimension('y', 3)

    # Create array with dims flipped
    arr = np.empty((ydim.size, xdim.size), dtype='f4')

    with pytest.raises(ValueError):
        ds.createVariable('data', 'f4', ('x', 'y'), wrap_array=arr)
Пример #4
0
def test_variable_size_check():
    """Test `Variable` checking size of passed array."""
    ds = Dataset()
    xdim = ds.createDimension("x", 2)
    ydim = ds.createDimension("y", 3)

    # Create array with dims flipped
    arr = np.empty((ydim.size, xdim.size), dtype="f4")

    with pytest.raises(ValueError):
        ds.createVariable("data", "f4", ("x", "y"), wrap_array=arr)
Пример #5
0
def test_group():
    """Test `Group`/`Dataset` behavior."""
    ds = Dataset()
    ds.createDimension('x', 5)
    ds.createVariable('data', 'f4', ('x',), 5)
    ds.conventions = 'CF-1.5'

    assert 'x' in ds.dimensions
    assert 'data' in ds.variables
    assert 'conventions' in ds.ncattrs()

    assert str(ds) == ('root\n\nDimensions:\n'
                       "<class 'metpy.io.cdm.Dimension'>: name = x, size = 5\n\n"
                       "Variables:\n<class 'metpy.io.cdm.Variable'>: float32 data(x)\n\t"
                       'shape = 5\n\nAttributes:\n\tconventions: CF-1.5')
Пример #6
0
def test_group():
    """Test `Group`/`Dataset` behavior."""
    ds = Dataset()
    ds.createDimension('x', 5)
    ds.createVariable('data', 'f4', ('x',), 5)
    ds.conventions = 'CF-1.5'

    assert 'x' in ds.dimensions
    assert 'data' in ds.variables
    assert 'conventions' in ds.ncattrs()

    assert str(ds) == ('root\n\nDimensions:\n'
                       "<class 'metpy.io.cdm.Dimension'>: name = x, size = 5\n\n"
                       "Variables:\n<class 'metpy.io.cdm.Variable'>: float32 data(x)\n\t"
                       'shape = 5\n\nAttributes:\n\tconventions: CF-1.5')
Пример #7
0
def test_var():
    """Fixture to create a dataset and variable for tests."""
    ds = Dataset()
    ds.createDimension('x', 5)
    var = ds.createVariable('data', 'f4', ('x',), 5)
    var[:] = np.arange(5)
    return var
Пример #8
0
def test_var():
    """Fixture to create a dataset and variable for tests."""
    ds = Dataset()
    ds.createDimension('x', 5)
    var = ds.createVariable('data', 'f4', ('x', ), 5)
    var[:] = np.arange(5)
    return var
Пример #9
0
def test_group():
    """Test `Group`/`Dataset` behavior."""
    ds = Dataset()
    ds.createDimension("x", 5)
    ds.createVariable("data", "f4", ("x",), 5)
    ds.conventions = "CF-1.5"

    assert "x" in ds.dimensions
    assert "data" in ds.variables
    assert "conventions" in ds.ncattrs()

    assert str(ds) == (
        "root\n\nDimensions:\n"
        "<class 'metpy.io.cdm.Dimension'>: name = x, size = 5\n\n"
        "Variables:\n<class 'metpy.io.cdm.Variable'>: float32 data(x)\n\t"
        "shape = 5\n\nAttributes:\n\tconventions: CF-1.5"
    )
Пример #10
0
def test_multidim_var():
    """Test multi-dim `Variable`."""
    ds = Dataset()
    ds.createDimension("x", 2)
    ds.createDimension("y", 3)
    var = ds.createVariable("data", "i8", ("x", "y"))

    assert var.shape == (2, 3)
    assert var.size == 6
    assert var.ndim == 2
    assert var.dtype == np.int64

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: int64 data(x, y)" "\n\tshape = (2, 3)")
Пример #11
0
def test_multidim_var():
    r'Test multi-dim Variable'
    ds = Dataset()
    ds.createDimension('x', 2)
    ds.createDimension('y', 3)
    var = ds.createVariable('data', 'i8', ('x', 'y'))

    assert var.shape == (2, 3)
    assert var.size == 6
    assert var.ndim == 2
    assert var.dtype == np.int64

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: int64 data(x, y)"
                        '\n\tshape = (2, 3)')
Пример #12
0
def test_multidim_var():
    """Test multi-dim `Variable`."""
    ds = Dataset()
    ds.createDimension('x', 2)
    ds.createDimension('y', 3)
    var = ds.createVariable('data', 'i8', ('x', 'y'))

    assert var.shape == (2, 3)
    assert var.size == 6
    assert var.ndim == 2
    assert var.dtype == np.int64

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: int64 data(x, y)"
                        '\n\tshape = (2, 3)')
Пример #13
0
def test_var():
    """Test `Variable` behavior."""
    ds = Dataset()
    ds.createDimension("x", 2)
    var = ds.createVariable("data", "f4", ("x",), 5)

    assert "data" in ds.variables
    assert var.shape == (2,)
    assert var.size == 2
    assert var.ndim == 1
    assert var.dtype == np.float32
    assert var[0] == 5

    var.units = "meters"

    assert "units" in var.ncattrs()
    assert var.units == "meters"

    assert var.group() is ds

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: float32 data(x)" "\n\tunits: meters\n\tshape = 2")
Пример #14
0
def test_var():
    r'Test Variable behavior'
    ds = Dataset()
    ds.createDimension('x', 2)
    var = ds.createVariable('data', 'f4', ('x', ), 5)

    assert 'data' in ds.variables
    assert var.shape == (2, )
    assert var.size == 2
    assert var.ndim == 1
    assert var.dtype == np.float32
    assert var[0] == 5

    var.units = 'meters'

    assert 'units' in var.ncattrs()
    assert var.units == 'meters'

    assert var.group() is ds

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: float32 data(x)"
                        '\n\tunits: meters\n\tshape = 2')
Пример #15
0
def test_var():
    """Test `Variable` behavior."""
    ds = Dataset()
    ds.createDimension('x', 2)
    var = ds.createVariable('data', 'f4', ('x',), 5)

    assert 'data' in ds.variables
    assert var.shape == (2,)
    assert var.size == 2
    assert var.ndim == 1
    assert var.dtype == np.float32
    assert var[0] == 5

    var.units = 'meters'

    assert 'units' in var.ncattrs()
    assert var.units == 'meters'

    assert var.group() is ds

    assert str(var) == ("<class 'metpy.io.cdm.Variable'>: float32 data(x)"
                        '\n\tunits: meters\n\tshape = 2')