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))
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))
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))
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)
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))
def test_grid_var_shape_fail(): model = BmiHeat() model.initialize() model.get_grid_shape(invalid_grid_id)
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)
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)
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))