def map(self, collection, **kwargs): """ Map the mask function over a collection :param collection: the ImageCollection :type collection: ee.ImageCollection :param renamed: whether the collection is renamed or not :type renamed: bool :param col: the EE Collection :type col: geetools.collection.Collection :return: the ImageCollection with all images masked :rtype: ee.ImageCollection """ col = kwargs.get('col') renamed = kwargs.get('renamed', False) bands = [] for band in [ 'aerosol', 'blue', 'green', 'red_edge_1', 'red_edge_2', 'red_edge_3', 'red_edge_4', 'water_vapor', 'cirrus', 'swir' ]: if renamed: bands.append(band) else: bands.append(col.get_band(band, 'name').id) if 'hollstein' in col.algorithms: f = lambda img: cloud_mask.applyHollstein(img, self.options, *bands ) return collection.map(f) else: return lambda img: img
def test_snow(): masked = cloud_mask.applyHollstein(image, ['snow']) vals = getValue(masked, p_snow, 30) assert vals.get("B1").getInfo() == None
def test_cirrus(): masked = cloud_mask.applyHollstein(image, ['cirrus']) vals = getValue(masked, p_cirrus, 30) assert vals.get("B1").getInfo() == None