示例#1
0
def test_initialize_from_file_like():
    config = StringIO(yaml.dump({"shape": [7, 5]}))
    model = BmiHeat()
    model.initialize(config)

    ndim = model.get_grid_rank(0)
    shape = np.empty(ndim, dtype=np.int32)

    assert_array_equal(model.get_grid_shape(0, shape), (7, 5))
示例#2
0
def test_initialize_from_file_like():
    from StringIO import StringIO
    import yaml

    config = StringIO(yaml.dump({'shape': (7, 5)}))
    model = BmiHeat()
    model.initialize(config)

    assert_equal(model.get_grid_shape(0), (7, 5))
示例#3
0
def test_initialize_from_file():
    import os
    import yaml
    import tempfile

    with tempfile.NamedTemporaryFile('w', delete=False) as fp:
        fp.write(yaml.dump({'shape': (7, 5)}))
        name = fp.name

    model = BmiHeat()
    model.initialize(name)

    os.remove(name)

    assert_equal(model.get_grid_shape(0), (7, 5))
示例#4
0
def test_initialize_from_file():
    import os
    import yaml
    import tempfile

    with tempfile.NamedTemporaryFile("w", delete=False) as fp:
        fp.write(yaml.dump({"shape": [7, 5]}))
        name = fp.name

    model = BmiHeat()
    model.initialize(name)

    os.remove(name)

    assert model.get_grid_shape(0) == (7, 5)
示例#5
0
def test_initialize_from_file():
    import os
    import yaml
    import tempfile

    with tempfile.NamedTemporaryFile("w", delete=False) as fp:
        fp.write(yaml.dump({"shape": [7, 5]}))
        name = fp.name

    model = BmiHeat()
    model.initialize(name)

    os.remove(name)

    ndim = model.get_grid_rank(0)
    shape = np.empty(ndim, dtype=np.int32)

    assert_array_equal(model.get_grid_shape(0, shape), (7, 5))
示例#6
0
def test_grid_var_shape_fail():
    model = BmiHeat()
    model.initialize()
    model.get_grid_shape(invalid_grid_id)
示例#7
0
def test_grid_var_shape():
    model = BmiHeat()
    model.initialize()
    assert_equal(model.get_grid_shape(grid_id), (10, 20))
def test_grid_var_shape_fail():
    model = BmiHeat()
    model.initialize()
    with pytest.raises(KeyError):
        model.get_grid_shape(invalid_grid_id)
def test_grid_var_shape():
    model = BmiHeat()
    model.initialize()
    assert model.get_grid_shape(grid_id) == (10, 20)
示例#10
0
def test_initialize_from_file_like():
    config = StringIO(yaml.dump({"shape": [7, 5]}))
    model = BmiHeat()
    model.initialize(config)

    assert model.get_grid_shape(0) == (7, 5)
示例#11
0
def test_grid_var_shape():
    model = BmiHeat()
    model.initialize()
    ndim = model.get_grid_rank(0)
    shape = np.empty(ndim, dtype=np.int32)
    assert_array_equal(model.get_grid_shape(grid_id, shape), (10, 20))
from heat import BmiHeat

bmiheatModel = BmiHeat()
bmiheatModel.initialize()

ipts = bmiheatModel.get_input_var_names()
opts = bmiheatModel.get_output_var_names()

for ipt in ipts:
    print(bmiheatModel.get_var_nbytes(ipt))
    print(bmiheatModel.get_var_type(ipt))
    grid = bmiheatModel.get_var_grid(ipt)
    print(bmiheatModel.get_grid_type(grid))
    print(bmiheatModel.get_grid_shape(grid))
    print(bmiheatModel.get_grid_spacing(grid))
    print(bmiheatModel.get_grid_origin(grid))

for opt in opts:
    print(bmiheatModel.get_var_nbytes(opt))
    print(bmiheatModel.get_var_type(opt))
    grid = bmiheatModel.get_var_grid(opt)
    print(bmiheatModel.get_grid_type(grid))
    print(bmiheatModel.get_grid_shape(grid))
    print(bmiheatModel.get_grid_spacing(grid))
    print(bmiheatModel.get_grid_origin(grid))
示例#13
0
def test_grid_var_shape_fail():
    model = BmiHeat()
    model.initialize()
    model.get_grid_shape(invalid_grid_id)
示例#14
0
def test_grid_var_shape():
    model = BmiHeat()
    model.initialize()
    assert_equal(model.get_grid_shape(grid_id), (10, 20))