def test_calc_dia(class_objects):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
beta_z = np.array([1, 2, 3])
expected = (drizzle.gamma(3) / drizzle.gamma(7) * 3.67**4 / beta_z)**(1 /
4)
testing.assert_array_almost_equal(obj._calc_dia(beta_z), expected)
def test_calc_beta_z_ratio(class_objects):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
obj._data.beta = np.array([[1, 1, 2], [1, 1, 3]])
obj._data.z = np.array([[2, 2, 1], [1, 1, 1]])
expected = 2 / np.pi * obj._data.beta / obj._data.z
testing.assert_array_almost_equal(obj._calc_beta_z_ratio(), expected)
def test_is_converged(class_objects):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
ind = (1, 2)
dia_init = np.array([[1, 3, 2], [3, 1, 2]])
dia = 1
expected = False
assert obj._is_converged(ind, dia, dia_init) == expected
def test_update_result_tables(class_objects, key, value):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
ind = (0, 1)
dia = 10
lut = (0, 1)
obj._update_result_tables(ind, dia, lut)
testing.assert_almost_equal(obj.params[key][ind], value)
def test_find_lut_indices(class_objects):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
ind = (1, 2)
dia_init = np.array([[1, 3, 2], [3, 1, 2]])
n_dia = 1
n_width = 2
ind_d = bisect_left(obj._data.mie['Do'], dia_init[ind], hi=n_dia - 1)
ind_w = bisect_left(obj._width_lut[:, ind_d], -obj._width_ht[ind], hi=n_width - 1)
compare = (ind_w, ind_d)
testing.assert_almost_equal(obj._find_lut_indices
(ind, dia_init, n_dia, n_width), compare)
def params_objects(class_objects):
d_source, d_class, s_width = class_objects
return drizzle.DrizzleSolver(d_source, d_class, s_width)
def test_init_variables(class_objects, key):
d_source, d_class, s_width = class_objects
obj = drizzle.DrizzleSolver(d_source, d_class, s_width)
result, x = obj._init_variables()
assert key in result.keys()