Esempio n. 1
0
def test_initial_conditions():
    """initial_conditions() sets u.prev & h.prev correctly
    """
    rain.initial_conditions(u, h, ho)
    # u.prev values
    np.testing.assert_equal(u.prev, np.zeros(n_grid))
    # h.prev values
    midpoint = n_grid // 2
    np.testing.assert_equal(h.prev[:midpoint], np.zeros(midpoint))
    assert_equal(h.prev[midpoint], ho)
    np.testing.assert_equal(h.prev[midpoint + 1:], np.zeros(midpoint))
Esempio n. 2
0
def test_initial_conditions():
    """initial_conditions() sets u.prev & h.prev correctly
    """
    rain.initial_conditions(u, h, ho)
    # u.prev values
    np.testing.assert_equal(u.prev, np.zeros(n_grid))
    # h.prev values
    midpoint = n_grid // 2
    np.testing.assert_equal(h.prev[:midpoint], np.zeros(midpoint))
    assert_equal(h.prev[midpoint], ho)
    np.testing.assert_equal(h.prev[midpoint + 1:], np.zeros(midpoint))
Esempio n. 3
0
def test_Quantity_store_timestep():
    """store_timestep() method copies time step values to storage array
    """
    rain.initial_conditions(u, h, ho)
    h.store_timestep(0, 'prev')
    np.testing.assert_equal(h.store[:, 0], h.prev)
Esempio n. 4
0
def test_Quantity_store_timestep():
    """store_timestep() method copies time step values to storage array
    """
    rain.initial_conditions(u, h, ho)
    h.store_timestep(0, 'prev')
    np.testing.assert_equal(h.store[:, 0], h.prev)