def testFilter(self): """Verifies the behavior of filter() method.""" collection = ee.Collection(ee.Function(), {}) # We don't allow empty filters. self.assertRaises(Exception, collection.filter) filtered = collection.filter(ee.Filter.eq('foo', 1)) self.assertEquals(ee.ApiFunction.lookup('Collection.filter'), filtered.func) self.assertEquals( { 'collection': collection, 'filter': ee.Filter.eq('foo', 1) }, filtered.args) self.assertTrue(isinstance(filtered, ee.Collection))
def testFilterShortcuts(self): """Verifies the behavior of the various filtering shortcut methods.""" collection = ee.Collection(ee.Function(), {}) geom = {'type': 'Polygon', 'coordinates': [[[1, 2], [3, 4]]]} d1 = datetime.datetime.strptime('1/1/2000', '%m/%d/%Y') d2 = datetime.datetime.strptime('1/1/2001', '%m/%d/%Y') self.assertEqual( collection.filter(ee.Filter.geometry(geom)), collection.filterBounds(geom)) self.assertEqual( collection.filter(ee.Filter.date(d1)), collection.filterDate(d1)) self.assertEqual( collection.filter(ee.Filter.date(d1, d2)), collection.filterDate( d1, d2)) self.assertEqual( collection.filter(ee.Filter.eq('foo', 13)), collection.filterMetadata('foo', 'equals', 13))