def test_initialize_InitRiseVelFromDist_uniform():
'Test initialize data_arrays with uniform dist'
data_arrays = mock_append_data_arrays(rise_vel_array, num_elems)
fcn = InitRiseVelFromDist()
fcn.initialize(num_elems, None, data_arrays)
assert_dataarray_shape_size(rise_vel_array, data_arrays, num_elems)
assert np.all(0 != data_arrays['rise_vel'])
assert np.all(data_arrays['rise_vel'] <= 1)
assert np.all(data_arrays['rise_vel'] >= 0)
def test_initialize_InitRiseVelFromDist_normal():
"""
test initialize data_arrays with normal dist
assume normal distribution works fine - so statistics (mean, var) are not
tested
"""
num_elems = 1000
data_arrays = mock_append_data_arrays(rise_vel_array, num_elems)
fcn = InitRiseVelFromDist('normal')
fcn.initialize(num_elems, None, data_arrays)
assert_dataarray_shape_size(rise_vel_array, data_arrays, num_elems)
assert np.all(0 != data_arrays['rise_vel'])
