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))