def test_get_initial_value():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
assert_array_less(z0, 1.)
assert_array_less(0., z0)
def test_get_initial_value():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ptr("plate_surface__temperature")
assert_array_less(z0, 1.0)
assert_array_less(0.0, z0)
def test_get_initial_value():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
assert_array_less(z0, 1.)
assert_array_less(0., z0)
def test_get_value_at_indices():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
z1 = model.get_value_at_indices('plate_surface__temperature', [0, 2, 4])
assert_array_almost_equal(z0.take((0, 2, 4)), z1)
def test_get_value_copy():
model = BmiHeat()
model.initialize()
z0 = model.get_value('plate_surface__temperature')
z1 = model.get_value('plate_surface__temperature')
assert_is_not(z0, z1)
assert_array_almost_equal(z0, z1)
def test_get_value_copy():
model = BmiHeat()
model.initialize()
z0 = model.get_value('plate_surface__temperature')
z1 = model.get_value('plate_surface__temperature')
assert_is_not(z0, z1)
assert_array_almost_equal(z0, z1)
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_get_value_copy():
model = BmiHeat()
model.initialize()
z0 = model.get_value("plate_surface__temperature")
z1 = model.get_value("plate_surface__temperature")
assert z0 is not z1
assert_array_almost_equal(z0, z1)
def test_grid_var_names():
model = BmiHeat()
model.initialize()
names = model.get_input_var_names()
assert_tuple_equal(names, ('plate_surface__temperature',))
names = model.get_output_var_names()
assert_tuple_equal(names, ('plate_surface__temperature',))
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)
assert_equal(model.get_grid_shape(0), (7, 5))
def test_update():
model = BmiHeat()
model.initialize()
for inc in xrange(10):
model.update()
assert_almost_equal(model.get_current_time(),
(inc + 1) * model.get_time_step())
def test_get_value_pointer():
model = BmiHeat()
model.initialize()
dest1 = np.empty(model.get_grid_size(0), dtype=float)
z0 = model.get_value_ptr("plate_surface__temperature")
z1 = model.get_value("plate_surface__temperature", dest1)
assert z0 is not z1
assert_array_almost_equal(z0.flatten(), z1)
for _ in range(10):
model.update()
assert z0 is model.get_value_ptr("plate_surface__temperature")
def test_get_value_pointer():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ptr("plate_surface__temperature")
z1 = model.get_value("plate_surface__temperature")
assert z0 is not z1
assert_array_almost_equal(z0, z1)
for _ in range(10):
model.update()
assert z0 is model.get_value_ptr("plate_surface__temperature")
def test_initialize_defaults():
model = BmiHeat()
model.initialize()
assert_almost_equal(model.get_current_time(), 0.)
assert_array_less(model.get_value('plate_surface__temperature'), 1.)
assert_array_less(0., model.get_value('plate_surface__temperature'))
def test_get_value_reference():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
z1 = model.get_value('plate_surface__temperature')
assert_is_not(z0, z1)
assert_array_almost_equal(z0, z1)
for _ in range(10):
model.update()
assert_is(z0, model.get_value_ref('plate_surface__temperature'))
def test_set_value_at_indices():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
model.set_value_at_indices('plate_surface__temperature', [-1, -1, -1],
[0, 2, 4])
new_z = model.get_value_ref('plate_surface__temperature')
assert_array_almost_equal(new_z.take((0, 2, 4)), -1.)
def test_set_value_at_indices():
model = BmiHeat()
model.initialize()
model.set_value_at_indices("plate_surface__temperature", [0, 2, 4],
[-1, -1, -1])
new_z = model.get_value_ptr("plate_surface__temperature")
assert_array_almost_equal(new_z.take((0, 2, 4)), -1.0)
def test_set_value():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
z1 = np.zeros_like(z0) - 1
model.set_value('plate_surface__temperature', z1)
new_z = model.get_value_ref('plate_surface__temperature')
assert_is(new_z, z0)
assert_is_not(new_z, z1)
assert_array_almost_equal(new_z, z1)
def test_get_value_reference():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ref('plate_surface__temperature')
z1 = model.get_value('plate_surface__temperature')
assert_is_not(z0, z1)
assert_array_almost_equal(z0, z1)
for _ in xrange(10):
model.update()
assert_is(z0, model.get_value_ref('plate_surface__temperature'))
def test_set_value():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ptr("plate_surface__temperature")
z1 = np.zeros_like(z0) - 1
model.set_value("plate_surface__temperature", z1)
new_z = model.get_value_ptr("plate_surface__temperature")
assert new_z is z0
assert new_z is not z1
assert_array_almost_equal(new_z, z1)
def test_initialize_defaults():
model = BmiHeat()
model.initialize()
assert_almost_equal(model.get_current_time(), 0.0)
z0 = model.get_value_ptr("plate_surface__temperature")
assert_array_less(z0, 1.0)
assert_array_less(0.0, z0)
def test_get_value_at_indices():
model = BmiHeat()
model.initialize()
z0 = model.get_value_ptr("plate_surface__temperature")
z1 = model.get_value_at_indices("plate_surface__temperature", [0, 2, 4])
assert_array_almost_equal(z0.take((0, 2, 4)), z1)
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_grid_var_names():
model = BmiHeat()
model.initialize()
names = model.get_input_var_names()
assert_tuple_equal(names, ('plate_surface__temperature', ))
names = model.get_output_var_names()
assert_tuple_equal(names, ('plate_surface__temperature', ))
def test_grid_var_names():
model = BmiHeat()
model.initialize()
names = model.get_input_var_names()
assert names == ("plate_surface__temperature", )
names = model.get_output_var_names()
assert names == ("plate_surface__temperature", )
def test_grid_var_item_count():
model = BmiHeat()
model.initialize()
count = model.get_input_item_count()
assert count == 1
count = model.get_output_item_count()
assert count == 1
def test_get_value_copy():
model = BmiHeat()
model.initialize()
dest0 = np.empty(model.get_grid_size(0), dtype=float)
dest1 = np.empty(model.get_grid_size(0), dtype=float)
z0 = model.get_value("plate_surface__temperature", dest0)
z1 = model.get_value("plate_surface__temperature", dest1)
assert z0 is not z1
assert_array_almost_equal(z0, z1)
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_type():
model = BmiHeat()
model.initialize()
assert_equal(model.get_var_type('plate_surface__temperature'), 'float64')
def test_grid_var_rank_fail():
model = BmiHeat()
model.initialize()
model.get_grid_rank(invalid_grid_id)
def test_finalize():
model = BmiHeat()
model.initialize()
model.update()
model.finalize()
def test_component_name():
model = BmiHeat()
name = model.get_component_name()
assert_equal(name, 'The 2D Heat Equation')
assert_is(model.get_component_name(), name)
def test_update_until():
model = BmiHeat()
model.initialize()
model.update_until(10.1)
assert_almost_equal(model.get_current_time(), 10.1)
def test_grid_var_origin():
model = BmiHeat()
model.initialize()
assert_equal(model.get_grid_origin(grid_id), (0., 0.))
def test_grid_var_rank():
model = BmiHeat()
model.initialize()
assert_equal(model.get_grid_rank(grid_id), 2)
def test_grid_var_spacing():
model = BmiHeat()
model.initialize()
assert_equal(model.get_grid_spacing(grid_id), (1., 1.))
def test_end_time():
model = BmiHeat()
model.initialize()
assert_almost_equal(model.get_end_time(), np.finfo('d').max)
def test_value_nbytes():
model = BmiHeat()
model.initialize()
z = model.get_value_ref('plate_surface__temperature')
assert_equal(model.get_var_nbytes('plate_surface__temperature'), z.nbytes)
def test_grid_type():
model = BmiHeat()
model.initialize()
assert_equal(model.get_grid_type(grid_id), 'uniform_rectilinear_grid')
def test_grid_var_shape_fail():
model = BmiHeat()
model.initialize()
model.get_grid_shape(invalid_grid_id)
def test_value_size():
model = BmiHeat()
model.initialize()
z = model.get_value_ptr("plate_surface__temperature")
assert model.get_grid_size(0) == z.size
def test_grid_var_shape():
model = BmiHeat()
model.initialize()
assert_equal(model.get_grid_shape(grid_id), (10, 20))
def test_value_nbytes():
model = BmiHeat()
model.initialize()
z = model.get_value_ptr("plate_surface__temperature")
assert model.get_var_nbytes("plate_surface__temperature") == z.nbytes
def test_grid_id():
model = BmiHeat()
model.initialize()
assert_equal(model.get_var_grid('plate_surface__temperature'), grid_id)
def test_start_time():
model = BmiHeat()
model.initialize()
assert_almost_equal(model.get_start_time(), 0.0)
def test_value_size():
model = BmiHeat()
model.initialize()
z = model.get_value_ref('plate_surface__temperature')
assert_equal(model.get_grid_size(0), z.size)
