earthengine-api 1.5.13rc0__py3-none-any.whl → 1.7.4__py3-none-any.whl

ee/tests/collection_test.py

@@ -11,7 +11,7 @@ from ee import apitestcase
 
 class CollectionTestCase(apitestcase.ApiTestCase):
 
-  def testSortAndLimit(self):
+  def test_sort_and_limit(self):
     """Verifies the behavior of the sort() and limit() methods."""
     collection = ee.Collection(ee.Function(), {})
 
@@ -41,12 +41,16 @@ class CollectionTestCase(apitestcase.ApiTestCase):
         'ascending': False
     }, reverse_sorted_collection.args)
 
-  def testFilter(self):
+  def test_filter(self):
     """Verifies the behavior of filter() method."""
     collection = ee.Collection(ee.Function(), {})
 
     # We don't allow empty filters.
     self.assertRaises(Exception, collection.filter)
+    with self.assertRaisesRegex(ee.EEException, 'Empty filters.'):
+      collection.filter(None)  # pytype: disable=wrong-arg-types
+    with self.assertRaisesRegex(ee.EEException, 'Empty filters.'):
+      collection.filter('')
 
     filtered = collection.filter(ee.Filter.eq('foo', 1))
     self.assertEqual(ee.ApiFunction.lookup('Collection.filter'), filtered.func)
@@ -56,7 +60,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     }, filtered.args)
     self.assertIsInstance(filtered, ee.Collection)
 
-  def testFilterShortcuts(self):
+  def test_filter_shortcuts(self):
     """Verifies the behavior of the various filtering shortcut methods."""
     collection = ee.Collection(ee.Function(), {})
     geom = {'type': 'Polygon', 'coordinates': [[[1, 2], [3, 4]]]}
@@ -75,7 +79,38 @@ class CollectionTestCase(apitestcase.ApiTestCase):
         collection.filter(ee.Filter.eq('foo', 13)),
         collection.filterMetadata('foo', 'equals', 13))
 
-  def testMapping(self):
+  def test_load_table(self):
+    """Verifies Collection.loadTable()."""
+    table_id = 'a/table/id'
+    geometry_column = 'geom'
+    version = 123
+    result = ee.Collection.loadTable(
+        tableId=table_id, geometryColumn=geometry_column, version=version
+    )
+    self.assertEqual(ee.ApiFunction.lookup('Collection.loadTable'), result.func)
+    self.assertEqual(
+        {
+            'tableId': ee.String(table_id),
+            'geometryColumn': ee.String(geometry_column),
+            'version': ee.Number(version),
+        },
+        result.args,
+    )
+    self.assertIsInstance(result, ee.FeatureCollection)
+
+    result2 = ee.Collection.loadTable(tableId=table_id)
+    self.assertEqual(ee.ApiFunction.lookup('Collection.loadTable'), result2.func)
+    self.assertEqual(
+        {
+            'tableId': ee.String(table_id),
+            'geometryColumn': None,
+            'version': None,
+        },
+        result2.args,
+    )
+    self.assertIsInstance(result2, ee.FeatureCollection)
+
+  def test_mapping(self):
     """Verifies the behavior of the map() method."""
     collection = ee.ImageCollection('foo')
     algorithm = lambda img: img.select('bar')
@@ -96,7 +131,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     self.assertEqual(expected_function.serialize(),
                      mapped.args['baseAlgorithm'].serialize())
 
-  def testNestedMapping(self):
+  def test_nested_mapping(self):
     """Verifies that nested map() calls produce distinct variables."""
     collection = ee.FeatureCollection('foo')
     result = collection.map(lambda x: collection.map(lambda y: [x, y]))
@@ -115,7 +150,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     self.assertEqual('_MAPPING_VAR_0_0',
                      inner_result.args['baseAlgorithm']._body[1].varName)
 
-  def testIteration(self):
+  def test_iteration(self):
     """Verifies the behavior of the iterate() method."""
     collection = ee.ImageCollection('foo')
     first = ee.Image(0)
@@ -140,7 +175,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     self.assertEqual(expected_function.serialize(),
                      result.args['function'].serialize())
 
-  def testNestedFunctions(self):
+  def test_nested_functions(self):
     """Verifies that nested function calls produce distinct variables."""
     fc = ee.FeatureCollection('fc')
     def f0(feat):
@@ -159,7 +194,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
         '0_0, 1_0, 1_1, 2_0, 2_1, 3_0, 3_1, 4_0',
         ', '.join(sorted(set(mapped_vars))))
 
-  def testUnboundArguments(self):
+  def test_unbound_arguments(self):
     fc = ee.FeatureCollection('fc')
     with self.assertRaisesRegex(
         Exception, 'User-defined methods must return a value'):
@@ -172,11 +207,11 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     ):
       fc.map(lambda x: x.serialize())
 
-  def testInitOptVarName(self):
+  def test_init_opt_var_name(self):
     result = ee.Collection(func=None, args=None, opt_varName='test').serialize()
     self.assertIn('"argumentReference": "test"', result)
 
-  def testFilterDateOptParams(self):
+  def test_filter_date_opt_params(self):
     result = (
         ee.Collection(func=None, args=None, varName='test')
         .filterDate(0, opt_end=42)
@@ -184,7 +219,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     )
     self.assertIn('"end": {"constantValue": 42}', result)
 
-  def testLimitOptParams(self):
+  def test_limit_opt_params(self):
     result = (
         ee.Collection(func=None, args=None, varName='test')
         .limit(0, opt_property='abc', opt_ascending=True)
@@ -193,7 +228,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     self.assertIn('"key": {"constantValue": "abc"}', result)
     self.assertIn('"ascending": {"constantValue": true}', result)
 
-  def testSortOptParams(self):
+  def test_sort_opt_params(self):
     result = (
         ee.Collection(func=None, args=None, varName='test')
         .sort('abc', opt_ascending=True)
@@ -201,7 +236,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
     )
     self.assertIn('"ascending": {"constantValue": true}', result)
 
-  def testMapOptParams(self):
+  def test_map_opt_params(self):
     a_func = lambda x: ee.Image(0)
     result = (
         ee.Collection(func=None, args=None, varName='test')

ee/tests/confusionmatrix_test.py

@@ -2,7 +2,7 @@
 """Tests for the ee.ConfusionMatrix module."""
 
 import json
-from typing import Any, Dict
+from typing import Any
 
 import unittest
 import ee
@@ -26,8 +26,8 @@ MATRIX = {
 
 
 def make_expression_graph(
-    function_invocation_value: Dict[str, Any],
-) -> Dict[str, Any]:
+    function_invocation_value: dict[str, Any],
+) -> dict[str, Any]:
   return {
       'result': '0',
       'values': {'0': {'functionInvocationValue': function_invocation_value}},