Ejemplo n.º 1
0
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")
Ejemplo n.º 2
0
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)
Ejemplo n.º 3
0
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)
Ejemplo n.º 4
0
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")
Ejemplo n.º 5
0
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)
Ejemplo n.º 6
0
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)
Ejemplo n.º 7
0
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)
Ejemplo n.º 8
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
Ejemplo n.º 9
0
def test_value_size():
    model = BmiHeat()
    model.initialize()

    z = model.get_value_ptr("plate_surface__temperature")
    assert model.get_grid_size(0) == z.size