def test_find_drizzle( drizzle_cat_file, is_rain, falling, droplet, cold, melting, insect, radar, lidar, clutter, molecular, attenuated, v_sigma, ): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.is_rain = is_rain obj.category_bits["falling"] = falling obj.category_bits["droplet"] = droplet obj.category_bits["cold"] = cold obj.category_bits["melting"] = melting obj.category_bits["insect"] = insect obj.quality_bits["radar"] = radar obj.quality_bits["lidar"] = lidar obj.quality_bits["clutter"] = clutter obj.quality_bits["molecular"] = molecular obj.quality_bits["attenuated"] = attenuated obj.is_v_sigma = v_sigma expected = np.array([[1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0]]) testing.assert_array_almost_equal(obj._find_drizzle(), expected)
def test_find_would_be_drizzle(drizzle_cat_file, is_rain, warm, falling, melting, insect, radar, clutter, molecular): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.is_rain = is_rain obj.warm_liquid = warm obj.category_bits["falling"] = falling obj.category_bits["melting"] = melting obj.category_bits["insect"] = insect obj.quality_bits["radar"] = radar obj.quality_bits["clutter"] = clutter obj.quality_bits["molecular"] = molecular expected = np.array([[0, 1, 1, 0], [0, 1, 1, 0], [0, 1, 1, 0], [0, 1, 1, 0]]) testing.assert_array_almost_equal(obj._find_would_be_drizzle(), expected)
def test_find_would_be_drizzle(drizzle_cat_file, is_rain, warm, falling, melting, insect, radar, clutter, molecular): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.is_rain = is_rain obj.warm_liquid = warm obj.category_bits['falling'] = falling obj.category_bits['melting'] = melting obj.category_bits['insect'] = insect obj.quality_bits['radar'] = radar obj.quality_bits['clutter'] = clutter obj.quality_bits['molecular'] = molecular compare = np.array([[0, 1, 1, 0], [0, 1, 1, 0], [0, 1, 1, 0], [0, 1, 1, 0]]) testing.assert_array_almost_equal(obj._find_would_be_drizzle(), compare)
def test_find_drizzle(drizzle_cat_file, is_rain, falling, droplet, cold, melting, insect, radar, lidar, clutter, molecular, attenuated, v_sigma): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.is_rain = is_rain obj.category_bits['falling'] = falling obj.category_bits['droplet'] = droplet obj.category_bits['cold'] = cold obj.category_bits['melting'] = melting obj.category_bits['insect'] = insect obj.quality_bits['radar'] = radar obj.quality_bits['lidar'] = lidar obj.quality_bits['clutter'] = clutter obj.quality_bits['molecular'] = molecular obj.quality_bits['attenuated'] = attenuated obj.is_v_sigma = v_sigma compare = np.array([[1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0], [1, 1, 0, 0]]) testing.assert_array_almost_equal(obj._find_drizzle(), compare)
def class_objects(drizzle_source_file, drizzle_cat_file): drizzle_source = drizzle.DrizzleSource(drizzle_source_file) drizzle_class = drizzle.DrizzleClassification(drizzle_cat_file) spectral_w = drizzle.SpectralWidth(drizzle_cat_file) return [drizzle_source, drizzle_class, spectral_w]
def test_find_cold_rain(drizzle_cat_file): obj = drizzle.DrizzleClassification(drizzle_cat_file) expected = np.array([0, 1]) testing.assert_array_almost_equal(obj._find_cold_rain(), expected)
def test_find_warm_liquid(drizzle_cat_file): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.category_bits["droplet"] = np.array([0, 0, 0, 1, 1, 1, 0], dtype=bool) obj.category_bits["cold"] = np.array([1, 1, 0, 0, 1, 0, 1], dtype=bool) expected = np.array([0, 0, 0, 1, 0, 1, 0], dtype=bool) testing.assert_array_almost_equal(obj._find_warm_liquid(), expected)
def test_find_v_sigma(drizzle_cat_file): obj = drizzle.DrizzleClassification(drizzle_cat_file) expected = np.array([[1, 0, 1], [1, 1, 1]], dtype=bool) testing.assert_array_almost_equal(obj._find_v_sigma(drizzle_cat_file), expected)
def test_find_warm_liquid(drizzle_cat_file): obj = drizzle.DrizzleClassification(drizzle_cat_file) obj.category_bits['droplet'] = np.array([0, 0, 0, 1, 1, 1, 0], dtype=bool) obj.category_bits['cold'] = np.array([1, 1, 0, 0, 1, 0, 1], dtype=bool) compare = np.array([0, 0, 0, 1, 0, 1, 0], dtype=bool) testing.assert_array_almost_equal(obj._find_warm_liquid(), compare)