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_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_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_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_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_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_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_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_value_size(): model = BmiHeat() model.initialize() z = model.get_value_ptr("plate_surface__temperature") assert model.get_grid_size(0) == z.size