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