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_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_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)