def test_append_iwc_including_rain(iwc_source_file, iwc_cat_file): from cloudnetpy.products.iwc import _append_iwc_including_rain ice_class = _IceClassification(iwc_cat_file) ice_class.is_ice = np.array([0, 0, 1]) ice_data = IwcSource(iwc_source_file) _append_iwc_including_rain(ice_data, ice_class) assert 'iwc_inc_rain' in ice_data.data.keys()
def test_find_would_be_ice(falling, cold, melting, insect, result, iwc_cat_file): obj = _IceClassification(iwc_cat_file) obj.category_bits['falling'] = falling obj.category_bits['cold'] = cold obj.category_bits['melting'] = melting obj.category_bits['insect'] = insect testing.assert_array_equal(obj._find_would_be_ice().data, result)
def test_find_uncorrected_ice(is_ice, attenuated, corrected, result, iwc_cat_file): obj = _IceClassification(iwc_cat_file) obj.quality_bits['attenuated'] = attenuated obj.quality_bits['corrected'] = corrected obj.is_ice = is_ice testing.assert_array_equal(obj._find_uncorrected_ice().data, result)
def test_append_iwc(iwc_source_file, iwc_cat_file): from cloudnetpy.products.iwc import _append_iwc ice_class = _IceClassification(iwc_cat_file) ice_class.ice_above_rain = np.array([0, 0, 1]) ice_data = IwcSource(iwc_source_file) ice_data.data['iwc_inc_rain'] = [1, 1, 1] _append_iwc(ice_data, ice_class) assert 'iwc' in ice_data.data.keys()
def test_iwc_error(iwc_source_file, iwc_cat_file): from cloudnetpy.products.iwc import _append_iwc_error ice_data = IwcSource(iwc_source_file) ice_class = _IceClassification(iwc_cat_file) ice_class.uncorrected_ice = np.array([1, 0, 1]) ice_class.is_ice = np.array([0, 0, 1]) ice_class.ice_above_rain = np.array([1, 1, 0]) _append_iwc_error(ice_data, ice_class) assert 'iwc_error' in ice_data.data.keys()
def test_append_iwc_including_rain_indices(iwc_source_file, iwc_cat_file): from cloudnetpy.products.iwc import _append_iwc_including_rain ice_class = _IceClassification(iwc_cat_file) ice_class.is_ice = np.array([0, 0, 1]) ice_data = IwcSource(iwc_source_file) _append_iwc_including_rain(ice_data, ice_class) x = ice_data.data['iwc_inc_rain'][:] x = x.mask assert np.alltrue(x[1:])
def test_append_iwc_indices(iwc_source_file, iwc_cat_file): from cloudnetpy.products.iwc import _append_iwc ice_class = _IceClassification(iwc_cat_file) ice_class.ice_above_rain = np.array([0, 0, 1]) ice_data = IwcSource(iwc_source_file) ice_data.data['iwc_inc_rain'] = [1, 1, 1] _append_iwc(ice_data, ice_class) x = ice_data.data['iwc'][:] x = x.mask assert np.alltrue(x[:2])
def test_find_cold_above_rain(cold, is_rain, melting, result, iwc_cat_file): obj = _IceClassification(iwc_cat_file) obj.category_bits['cold'] = cold obj.category_bits['melting'] = melting obj.is_rain = is_rain testing.assert_array_equal(obj._find_cold_above_rain().data, result)
def test_find_ice_above_rain(is_ice, is_rain, result, iwc_cat_file): obj = _IceClassification(iwc_cat_file) obj.is_ice = is_ice obj.is_rain = is_rain testing.assert_array_equal(obj._find_ice_above_rain().data, result)