Exemple #1
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
Exemple #2
0
def test_dim():
    """Test `Dimension` behavior."""
    ds = Dataset()
    dim = ds.createDimension('x', 5)
    assert dim.size == 5
    assert dim.group() is ds
    assert str(dim) == "<class 'metpy.io.cdm.Dimension'>: name = x, size = 5"
Exemple #3
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
Exemple #4
0
def test_dim():
    r'Test Dimension behavior'
    ds = Dataset()
    dim = ds.createDimension('x', 5)
    assert dim.size == 5
    assert dim.group() is ds
    assert str(dim) == "<class 'metpy.io.cdm.Dimension'>: name = x, size = 5"
Exemple #5
0
def test_add_group():
    """Test adding a group."""
    ds = Dataset()
    grp = ds.createGroup("myGroup")
    assert grp.name == "myGroup"
    assert "myGroup" in ds.groups

    assert str(ds) == "root\nGroups:\nmyGroup"
Exemple #6
0
def test_add_group():
    r'Test adding a group'
    ds = Dataset()
    grp = ds.createGroup('myGroup')
    assert grp.name == 'myGroup'
    assert 'myGroup' in ds.groups

    assert str(ds) == 'root\nGroups:\nmyGroup'
Exemple #7
0
def test_add_group():
    """Test adding a group."""
    ds = Dataset()
    grp = ds.createGroup('myGroup')
    assert grp.name == 'myGroup'
    assert 'myGroup' in ds.groups

    assert str(ds) == 'root\nGroups:\nmyGroup'
Exemple #8
0
def test_remove_attr():
    """Test removing an attribute."""
    ds = Dataset()
    ds.maker = 'me'
    assert 'maker' in ds.ncattrs()

    del ds.maker
    assert not hasattr(ds, 'maker')
    assert 'maker' not in ds.ncattrs()
Exemple #9
0
def test_remove_attr():
    """Test removing an attribute."""
    ds = Dataset()
    ds.maker = "me"
    assert "maker" in ds.ncattrs()

    del ds.maker
    assert not hasattr(ds, "maker")
    assert "maker" not in ds.ncattrs()
Exemple #10
0
def test_remove_attr():
    r'Test removing an attribute'
    ds = Dataset()
    ds.maker = 'me'
    assert 'maker' in ds.ncattrs()

    del ds.maker
    assert not hasattr(ds, 'maker')
    assert 'maker' not in ds.ncattrs()
Exemple #11
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)
Exemple #12
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)
Exemple #13
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)
Exemple #14
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)
Exemple #15
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)")
Exemple #16
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)')
Exemple #17
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)')
Exemple #18
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")
Exemple #19
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')
Exemple #20
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')
Exemple #21
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')
Exemple #22
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')
Exemple #23
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"
    )