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)