earthengine-api 1.6.11__py3-none-any.whl → 1.6.12rc0__py3-none-any.whl

ee/tests/_state_test.py

@@ -0,0 +1,49 @@
+#!/usr/bin/env python3
+"""Tests for ee._state."""
+
+from absl.testing import parameterized
+
+import unittest
+from ee import _state
+
+
+class StateTest(parameterized.TestCase):
+
+  def setUp(self):
+    super().setUp()
+    _state.reset_state()
+
+  @parameterized.named_parameters(
+      ('global_mode', False),
+  )
+  def test_get_state(self, use_context_mode: bool):
+    state = _state.get_state()
+
+    self.assertIsInstance(state, _state.EEState)
+    self.assertEqual(state, _state.EEState())
+
+  @parameterized.named_parameters(
+      ('global_mode', False),
+  )
+  def test_update_state(self, use_context_mode: bool):
+    state = _state.get_state()
+
+    # Modify the state and verify the global state has been updated.
+    state.cloud_api_user_project = 'my-project'
+
+    self.assertEqual(_state.get_state().cloud_api_user_project, 'my-project')
+
+  @parameterized.named_parameters(
+      ('global_mode', False),
+  )
+  def test_reset_state(self, use_context_mode: bool):
+    state = _state.get_state()
+    state.cloud_api_user_project = 'my-project'
+
+    _state.reset_state()
+
+    self.assertEqual(_state.get_state(), _state.EEState())
+
+
+if __name__ == '__main__':
+  unittest.main()

ee/tests/apifunction_test.py

@@ -10,7 +10,7 @@ from ee import apitestcase
 
 class ApiFunctionTest(apitestcase.ApiTestCase):
 
-  def testAddFunctions(self):
+  def test_add_functions(self):
     """Verifies that addition of static and instance API functions."""
 
     # Check instance vs static functions, and trampling of
@@ -48,7 +48,7 @@ class ApiFunctionTest(apitestcase.ApiTestCase):
     self.assertTrue(hasattr(TestClass, 'pre_addBands'))
     self.assertFalse(hasattr(TestClass, '_pre_addBands'))
 
-  def testAddFunctions_Inherited(self):
+  def test_add_functions_inherited(self):
     """Verifies that inherited non-client functions can be overridden."""
 
     class Base:
@@ -66,7 +66,7 @@ class ApiFunctionTest(apitestcase.ApiTestCase):
     self.assertNotEqual(Base.addBands, Child.addBands)
     # pytype: enable=attribute-error
 
-  def testEq(self):
+  def test_eq(self):
     a_signature = {'hello': 'world', 'args': []}
     b_signature = {
         'hello': 'world',
@@ -93,14 +93,14 @@ class ApiFunctionTest(apitestcase.ApiTestCase):
         ee.ApiFunction(name='test', signature=a_signature), a_signature
     )
 
-  def testInitOptParams(self):
+  def test_init_opt_params(self):
     signature = {'hello': 'world', 'args': []}
     self.assertEqual(
         ee.ApiFunction(name='test', signature=signature),
         ee.ApiFunction(name='test', opt_signature=signature),
     )
 
-  def testImportApiOptParams(self):
+  def test_import_api_opt_params(self):
     args = dict(
         target=ee.Dictionary, prefix='Dictionary', type_name='Dictionary'
     )

ee/tests/batch_test.py

@@ -8,6 +8,7 @@ from unittest import mock
 
 import unittest
 import ee
+from ee import _state
 from ee import apitestcase
 from ee import batch
 from ee import data
@@ -55,15 +56,25 @@ class TaskTest(unittest.TestCase):
 
   def setUp(self):
     super().setUp()
-    data.setCloudApiUserProject('test-project')
+    mock.patch.object(
+        _state,
+        'get_state',
+        return_value=_state.EEState(
+            initialized=True, cloud_api_user_project='test-project'
+        ),
+    ).start()
+
+  def tearDown(self):
+    super().tearDown()
+    mock.patch.stopall()
 
-  def testStartWithoutConfig(self):
+  def test_start_without_config(self):
     task = batch.Task('an id', 'a task type', 'a state')
     self.assertIsNone(task.config)
     with self.assertRaisesRegex(ee.EEException, 'Task config'):
       task.start()
 
-  def testStartUnknownTaskType(self):
+  def test_start_unknown_task_type(self):
     task_type = 'bad task type'
     task = batch.Task('an id', task_type, 'a state', {'some': 'value'})
     with self.assertRaisesRegex(
@@ -71,7 +82,7 @@ class TaskTest(unittest.TestCase):
     ):
       task.start()
 
-  def testStatusWithId(self):
+  def test_status_with_id(self):
     name = 'projects/test-project/operations/test_1'
     task = batch.Task('an id', 'a task type', 'a state', name=name)
     with mock.patch.object(
@@ -82,7 +93,7 @@ class TaskTest(unittest.TestCase):
           m.call_args.args[0], 'projects/test-project/operations/test_1'
       )
 
-  def testStatusWithName(self):
+  def test_status_with_name(self):
     task = batch.Task(
         None,
         'a task type',
@@ -97,7 +108,7 @@ class TaskTest(unittest.TestCase):
           m.call_args.args[0], 'projects/test-project/operations/test_1'
       )
 
-  def testStatusWithIdStateUnknown(self):
+  def test_status_with_id_state_unknown(self):
     name = 'projects/test-project/operations/an id'
     task = batch.Task('an id', 'a task type', 'a state', name=name)
     with mock.patch.object(
@@ -108,11 +119,11 @@ class TaskTest(unittest.TestCase):
           m.call_args.args[0], 'projects/test-project/operations/an id'
       )
 
-  def testStatusWithoutIdOrName(self):
+  def test_status_without_id_or_name(self):
     task = batch.Task(None, 'a task type', 'a state')
     self.assertEqual('UNSUBMITTED', task.status()['state'])
 
-  def testActive(self):
+  def test_active(self):
     name = 'projects/test-project/operations/an id'
     task = batch.Task('an id', 'a task type', 'a state', name=name)
     with mock.patch.object(
@@ -120,18 +131,18 @@ class TaskTest(unittest.TestCase):
     ):
       self.assertTrue(task.active())
 
-  def testNotActive(self):
+  def test_not_active(self):
     task = batch.Task('an id', 'a task type', 'a state')
     with mock.patch.object(
         data, 'getOperation', return_value=SUCCEEDED_OPERATION
     ):
       self.assertFalse(task.active())
 
-  def testReprWithoutConfig(self):
+  def test_repr_without_config(self):
     task = batch.Task('an id', 'a task type', 'a state')
     self.assertEqual('<Task "an id">', task.__repr__())
 
-  def testReprWithConfig(self):
+  def test_repr_with_config(self):
     an_id = None
     task_type = 'a task type'
     state = 'a state'
@@ -143,7 +154,7 @@ class TaskTest(unittest.TestCase):
         f'<Task {task_type}: {description} ({state})>', task.__repr__()
     )
 
-  def testReprWithIdAndConfig(self):
+  def test_repr_with_id_and_config(self):
     an_id = 'an id'
     task_type = 'a task type'
     state = 'a state'
@@ -158,23 +169,23 @@ class TaskTest(unittest.TestCase):
 
 class ExportTest(unittest.TestCase):
 
-  def testExportCannotInit(self):
+  def test_export_cannot_init(self):
     with self.assertRaises(AssertionError):
       batch.Export()
 
-  def testExportImageCannotInit(self):
+  def test_export_image_cannot_init(self):
     with self.assertRaises(AssertionError):
       batch.Export.image.__init__('something')
 
-  def testExportMapCannotInit(self):
+  def test_export_map_cannot_init(self):
     with self.assertRaises(AssertionError):
       batch.Export.map.__init__('something')
 
-  def testExportTableCannotInit(self):
+  def test_export_table_cannot_init(self):
     with self.assertRaises(AssertionError):
       batch.Export.table.__init__('something')
 
-  def testExportVideoCannotInit(self):
+  def test_export_video_cannot_init(self):
     with self.assertRaises(AssertionError):
       batch.Export.video.__init__('something')
 
@@ -191,7 +202,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
     self.start_call_params = None
     self.update_call_params = None
 
-  def testTaskStartCloudApi(self):
+  def test_task_start_cloud_api(self):
     """Verifies that Task.start() calls the server appropriately."""
     mock_cloud_api_resource = mock.MagicMock()
     mock_cloud_api_resource.projects().table().export().execute.return_value = {
@@ -206,7 +217,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
       self.assertTrue(export_args[1]['body']['requestId'])
       self.assertEqual(export_args[1]['body']['description'], 'bar')
 
-  def testTaskCancelCloudApi(self):
+  def test_task_cancel_cloud_api(self):
     mock_cloud_api_resource = mock.MagicMock()
     mock_cloud_api_resource.projects().operations().list().execute.return_value = {
         'operations': [{
@@ -227,7 +238,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           cancel_args[1]['name'], 'projects/earthengine-legacy/operations/TEST1'
       )
 
-  def testExportImageTrivialRegionCloudApi(self):
+  def test_export_image_trivial_region_cloud_api(self):
     """Verifies the task created by Export.image() with a trivial region."""
     with apitestcase.UsingCloudApi():
       region = [0, 0, 1, 0, 1, 1]
@@ -262,7 +273,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task.config,
       )
 
-  def testExportImageCloudApi(self):
+  def test_export_image_cloud_api(self):
     """Verifies the task created by Export.image()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -305,7 +316,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task.config,
       )
 
-  def testExportImageCloudApiInvalidSkipEmptyTiles(self):
+  def test_export_image_cloud_api_invalid_skip_empty_tiles(self):
     """Verifies errors are thrown when incorrectly specifying skipEmptyTiles."""
     with apitestcase.UsingCloudApi():
       with self.assertRaisesRegex(
@@ -318,7 +329,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             skipEmptyTiles=True,
         )
 
-  def testExportImageWithTfRecordCloudApi(self):
+  def test_export_image_with_tf_record_cloud_api(self):
     """Verifies the task created by Export.image()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -377,7 +388,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task.config,
       )
 
-  def testExportImageToAssetCloudApi(self):
+  def test_export_image_to_asset_cloud_api(self):
     """Verifies the Asset export task created by Export.image.toAsset()."""
     with apitestcase.UsingCloudApi():
       config = dict(
@@ -441,7 +452,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_ordered.config,
       )
 
-  def testExportImageToAssetCloudApi_withTileSize(self):
+  def test_export_image_to_asset_cloud_api_with_tile_size(self):
     """Verifies the Asset export task created by Export.image.toAsset()."""
     with apitestcase.UsingCloudApi():
       config = dict(
@@ -479,7 +490,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_ordered.config,
       )
 
-  def testExportImageToCloudStorageCloudApi(self):
+  def test_export_image_to_cloud_storage_cloud_api(self):
     """Verifies the Cloud Storage export task created by Export.image()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -573,7 +584,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_with_priority.config,
       )
 
-  def testExportImageToGoogleDriveCloudApi(self):
+  def test_export_image_to_google_drive_cloud_api(self):
     """Verifies the Drive destined task created by Export.image.toDrive()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -679,7 +690,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           drive_task_with_priority.config,
       )
 
-  def testExportMapToCloudStorageCloudApi(self):
+  def test_export_map_to_cloud_storage_cloud_api(self):
     """Verifies the task created by Export.map.toCloudStorage()."""
     with apitestcase.UsingCloudApi():
       config = dict(
@@ -812,7 +823,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_with_priority.config,
       )
 
-  def testExportMapToCloudStorageCloudApi_WithV1Parameters(self):
+  def test_export_map_to_cloud_storage_cloud_api_with_v1_parameters(self):
     """Verifies Export.map.toCloudStorage() tasks with v1 parameters."""
     with apitestcase.UsingCloudApi():
       config = dict(
@@ -868,7 +879,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_keyed.config,
       )
 
-  def testExportMapToCloudStorageCloudApi_WithV1AlphaParameters(self):
+  def test_export_map_to_cloud_storage_cloud_api_with_v1alpha_parameters(self):
     """Verifies Export.map.toCloudStorage() tasks with v1alpha parameters."""
     with apitestcase.UsingCloudApi():
       task_keyed = ee.batch.Export.map.toCloudStorage(
@@ -897,7 +908,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_keyed.config,
       )
 
-  def testExportTableCloudApi(self):
+  def test_export_table_cloud_api(self):
     """Verifies the task created by Export.table()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table(
@@ -922,7 +933,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task.config,
       )
 
-  def testExportTableCloudApiBogusParameter(self):
+  def test_export_table_cloud_api_bogus_parameter(self):
     """Verifies that bogus parameters are rejected."""
     with apitestcase.UsingCloudApi():
       with self.assertRaisesRegex(
@@ -932,7 +943,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             ee.FeatureCollection('drive test FC'), framesPerSecond=30
         )
 
-  def testExportTableSelectorsCloudApi(self):
+  def test_export_table_selectors_cloud_api(self):
     """Verifies that table export accepts a list or tuple of selectors."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toCloudStorage(
@@ -955,7 +966,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
       )
       self.assertEqual(['ab', 'cd', 'ef'], task.config['selectors'])
 
-  def testExportTableToCloudStorageCloudApi(self):
+  def test_export_table_to_cloud_storage_cloud_api(self):
     """Verifies the Cloud Storage task created by Export.table()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toCloudStorage(
@@ -1006,7 +1017,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_with_priority.config,
       )
 
-  def testExportTableToGoogleDriveCloudApi(self):
+  def test_export_table_to_google_drive_cloud_api(self):
     """Verifies the Drive destined task created by Export.table.toDrive()."""
     with apitestcase.UsingCloudApi():
       test_collection = ee.FeatureCollection('foo')
@@ -1093,7 +1104,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
       )
       self.assertEqual(expected_config_with_priority, task_with_priority.config)
 
-  def testExportTableToAssetCloudApi(self):
+  def test_export_table_to_asset_cloud_api(self):
     """Verifies the export task created by Export.table.toAsset()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toAsset(
@@ -1141,7 +1152,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_with_priority.config,
       )
 
-  def testExportTableWithFileFormatCloudApi(self):
+  def test_export_table_with_file_format_cloud_api(self):
     """Verifies the task created by Export.table() given a file format."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toCloudStorage(
@@ -1168,7 +1179,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task.config,
       )
 
-  def testExportTableToFeatureViewCloudApi(self):
+  def test_export_table_to_feature_view_cloud_api(self):
     """Verifies the export task created by Export.table.toFeatureView()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toFeatureView(
@@ -1231,7 +1242,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           task_with_priority.config,
       )
 
-  def testExportTableToFeatureViewEmptyParamsCloudApi(self):
+  def test_export_table_to_feature_view_empty_params_cloud_api(self):
     """Verifies the export task created by Export.table.toFeatureView()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toFeatureView(
@@ -1263,7 +1274,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             task.config,
         )
 
-  def testExportTableToFeatureViewAllIngestionParams(self):
+  def test_export_table_to_feature_view_all_ingestion_params(self):
     """Verifies the task ingestion params created by toFeatureView()."""
     task = ee.batch.Export.table.toFeatureView(
         collection=ee.FeatureCollection('foo'),
@@ -1309,7 +1320,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
         task.config['featureViewExportOptions']['ingestionTimeParameters'],
     )
 
-  def testExportTableToFeatureViewBadRankByOneThingRule(self):
+  def test_export_table_to_feature_view_bad_rank_by_one_thing_rule(self):
     """Verifies a bad RankByOneThingRule throws an exception."""
     with self.assertRaisesRegex(
         ee.EEException, 'Ranking rule format is invalid.*'
@@ -1320,7 +1331,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           ingestionTimeParameters={'thinningRanking': 'my-attribute BAD_DIR'},
       )
 
-  def testExportTableToFeatureViewBadRankingRule(self):
+  def test_export_table_to_feature_view_bad_ranking_rule(self):
     """Verifies a bad RankingRule throws an exception."""
     with self.assertRaisesRegex(
         ee.EEException, 'Unable to build ranking rule from rules.*'
@@ -1331,7 +1342,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           ingestionTimeParameters={'thinningRanking': {'key': 'val'}},
       )
 
-  def testExportTableToFeatureViewBadIngestionTimeParams(self):
+  def test_export_table_to_feature_view_bad_ingestion_time_params(self):
     """Verifies a bad set of ingestion time params throws an exception."""
     with self.assertRaisesRegex(
         ee.EEException,
@@ -1346,7 +1357,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
           ingestionTimeParameters={'badThinningKey': {'key': 'val'}},
       )
 
-  def testExportTableToBigQueryRequiredParams(self):
+  def test_export_table_to_big_query_required_params(self):
     """Verifies the export task created by Export.table.toBigQuery()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toBigQuery(
@@ -1399,7 +1410,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             task_with_priority.config,
         )
 
-  def testExportTableToBigQueryAllParams(self):
+  def test_export_table_to_big_query_all_params(self):
     """Verifies the export task created by Export.table.toBigQuery()."""
     with apitestcase.UsingCloudApi():
       task = ee.batch.Export.table.toBigQuery(
@@ -1436,7 +1447,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             task.config,
         )
 
-  def testExportTableToBigQueryBadTableName(self):
+  def test_export_table_to_big_query_bad_table_name(self):
     """Verifies a bad table name throws an exception."""
     with apitestcase.UsingCloudApi():
       with self.assertRaisesRegex(
@@ -1459,7 +1470,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             description='foo',
         )
 
-  def testExportVideoCloudApi(self):
+  def test_export_video_cloud_api(self):
     """Verifies the task created by Export.video()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -1549,7 +1560,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
             collection, 'TestVideoName', config_with_bogus_option
         )
 
-  def testExportVideoToCloudStorageCloudApi(self):
+  def test_export_video_to_cloud_storage_cloud_api(self):
     """Verifies the task created by Export.video.toCloudStorage()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -1644,7 +1655,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
       )
       self.assertEqual(expected_config_with_priority, task_with_priority.config)
 
-  def testExportVideoToDriveCloudApi(self):
+  def test_export_video_to_drive_cloud_api(self):
     """Verifies the task created by Export.video.toDrive()."""
     with apitestcase.UsingCloudApi():
       region = ee.Geometry.Rectangle(1, 2, 3, 4)
@@ -1737,7 +1748,7 @@ class BatchTestCase(apitestcase.ApiTestCase):
       )
       self.assertEqual(expected_config_with_priority, task_with_priority.config)
 
-  def testExportWorkloadTag(self):
+  def test_export_workload_tag(self):
     """Verifies that the workload tag state is captured before start."""
     mock_cloud_api_resource = mock.MagicMock()
     mock_cloud_api_resource.projects().table().export().execute.return_value = {

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,7 +41,7 @@ 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(), {})
 
@@ -56,7 +56,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 +75,7 @@ class CollectionTestCase(apitestcase.ApiTestCase):
         collection.filter(ee.Filter.eq('foo', 13)),
         collection.filterMetadata('foo', 'equals', 13))
 
-  def testMapping(self):
+  def test_mapping(self):
     """Verifies the behavior of the map() method."""
     collection = ee.ImageCollection('foo')
     algorithm = lambda img: img.select('bar')
@@ -96,7 +96,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 +115,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 +140,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 +159,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 +172,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 +184,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 +193,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 +201,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')