예제 #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
예제 #2
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
예제 #3
0
파일: test_cdm.py 프로젝트: metpy/MetPy
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)")
예제 #4
0
파일: test_cdm.py 프로젝트: akrherz/MetPy
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)')
예제 #5
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)')
예제 #6
0
파일: test_cdm.py 프로젝트: wholmgren/MetPy
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
파일: test_cdm.py 프로젝트: akrherz/MetPy
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')
예제 #8
0
파일: test_cdm.py 프로젝트: akrherz/MetPy
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"
예제 #9
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"
예제 #10
0
파일: test_cdm.py 프로젝트: metpy/MetPy
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"
    )
예제 #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)
예제 #12
0
파일: test_cdm.py 프로젝트: akrherz/MetPy
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)
예제 #13
0
파일: test_cdm.py 프로젝트: metpy/MetPy
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)
예제 #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)
예제 #15
0
파일: test_cdm.py 프로젝트: metpy/MetPy
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")
예제 #16
0
파일: test_cdm.py 프로젝트: akrherz/MetPy
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')
예제 #17
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')