def test_initialize_InitRiseVelFromDropletDist_weibull():
'Test initialize data_arrays with Weibull dist'
num_elems = 10
data_arrays = mock_append_data_arrays(rise_vel_diameter_array, num_elems)
substance = OilProps('oil_conservative')
spill = Spill(Release())
# (.001*.2) / (.693 ** (1 / 1.8)) - smaller droplet test case, in mm
# so multiply by .001
fcn = InitRiseVelFromDropletSizeFromDist(WeibullDistribution(alpha=1.8,
lambda_=.000248
))
fcn.initialize(num_elems, spill, data_arrays, substance)
assert_dataarray_shape_size(rise_vel_array, data_arrays, num_elems)
assert np.all(0 != data_arrays['rise_vel'])
assert np.all(0 != data_arrays['droplet_diameter'])
def test_initialize_InitRiseVelFromDropletDist_weibull_with_min_max():
'Test initialize data_arrays with Weibull dist'
num_elems = 1000
data_arrays = mock_append_data_arrays(rise_vel_diameter_array, num_elems)
substance = OilProps('oil_conservative')
spill = Spill(Release())
# (.001*3.8) / (.693 ** (1 / 1.8)) - larger droplet test case, in mm
# so multiply by .001
fcn = InitRiseVelFromDropletSizeFromDist(WeibullDistribution(min_=0.002,
max_=0.004,
alpha=1.8,
lambda_=.00456
))
fcn.initialize(num_elems, spill, data_arrays, substance)
# test for the larger droplet case above
assert np.all(data_arrays['droplet_diameter'] >= .002)
# test for the larger droplet case above
assert np.all(data_arrays['droplet_diameter'] <= .004)
