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