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

ee/reducer.py

@@ -1,8 +1,6 @@
 """A wrapper for Reducers."""
 from __future__ import annotations
 
-from typing import Optional
-
 from ee import _arg_types
 from ee import apifunction
 from ee import computedobject
@@ -100,11 +98,11 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def autoHistogram(
       # pylint: disable=invalid-name
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
-      cumulative: Optional[_arg_types.Bool] = None,
+      cumulative: _arg_types.Bool | None = None,
   ) -> Reducer:
     """Returns an ee.Reducer that computes a histogram of the inputs.
 
@@ -201,8 +199,8 @@ class Reducer(computedobject.ComputedObject):
       self,
       reducer2: _arg_types.Reducer,
       # pylint: disable=invalid-name
-      outputPrefix: Optional[_arg_types.String] = None,
-      sharedInputs: Optional[_arg_types.Bool] = None,
+      outputPrefix: _arg_types.String | None = None,
+      sharedInputs: _arg_types.Bool | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a Reducer that runs two reducers in parallel.
@@ -289,7 +287,7 @@ class Reducer(computedobject.ComputedObject):
 
     return apifunction.ApiFunction.call_('Reducer.covariance')
 
-  def disaggregate(self, axis: Optional[_arg_types.Integer] = None) -> Reducer:
+  def disaggregate(self, axis: _arg_types.Integer | None = None) -> Reducer:
     """Returns a Reducer that separates aggregate inputs.
 
     Separates aggregate inputs (Arrays, Lists, or Dictionaries) into individual
@@ -363,7 +361,7 @@ class Reducer(computedobject.ComputedObject):
       min: _arg_types.Number,  # pylint: disable=redefined-builtin
       max: _arg_types.Number,  # pylint: disable=redefined-builtin
       steps: _arg_types.Integer,
-      cumulative: Optional[_arg_types.Bool] = None,
+      cumulative: _arg_types.Bool | None = None,
   ) -> Reducer:
     """Returns a fixed histogram reducer.
 
@@ -443,9 +441,9 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def geometricMedian(
       numX: _arg_types.Integer,  # pylint: disable=invalid-name
-      eta: Optional[_arg_types.Number] = None,
+      eta: _arg_types.Number | None = None,
       # pylint: disable-next=invalid-name
-      initialStepSize: Optional[_arg_types.Number] = None,
+      initialStepSize: _arg_types.Number | None = None,
   ) -> Reducer:
     """Returns a reducer that computes the geometric median across the inputs.
 
@@ -467,8 +465,10 @@ class Reducer(computedobject.ComputedObject):
 
   def group(
       self,
-      groupField: Optional[_arg_types.Integer] = None,
-      groupName: Optional[_arg_types.String] = None,
+      # pylint: disable=invalid-name
+      groupField: _arg_types.Integer | None = None,
+      groupName: _arg_types.String | None = None,
+      # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer groups reducer records by the value of a given input.
 
@@ -487,9 +487,9 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def histogram(
       # pylint: disable=invalid-name
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that will compute a histogram of the inputs.
@@ -512,9 +512,9 @@ class Reducer(computedobject.ComputedObject):
       # pylint: disable=invalid-name
       minPercentile: _arg_types.Number,
       maxPercentile: _arg_types.Number,
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns an interval mean reducer.
@@ -547,7 +547,7 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def kendallsCorrelation(
       # pylint: disable-next=invalid-name
-      numInputs: Optional[_arg_types.Integer] = None,
+      numInputs: _arg_types.Integer | None = None,
   ) -> Reducer:
     """Returns a reducer that computes the Kendall's Tau-b rank correlation.
 
@@ -609,7 +609,7 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def linearRegression(
       numX: _arg_types.Integer,  # pylint: disable=invalid-name
-      numY: Optional[_arg_types.Integer] = None,  # pylint: disable=invalid-name
+      numY: _arg_types.Integer | None = None,  # pylint: disable=invalid-name
   ) -> Reducer:
     """Returns a linear regression reducer.
 
@@ -635,7 +635,7 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def max(
       # pylint: disable-next=invalid-name
-      numInputs: Optional[_arg_types.Integer] = None,
+      numInputs: _arg_types.Integer | None = None,
   ) -> Reducer:
     """Returns a reducer that outputs the maximum value of its (first) input.
 
@@ -660,9 +660,9 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def median(
       # pylint: disable=invalid-name
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that will compute the median of the inputs.
@@ -687,7 +687,7 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def min(
       # pylint: disable-next=invalid-name
-      numInputs: Optional[_arg_types.Integer] = None,
+      numInputs: _arg_types.Integer | None = None,
   ) -> Reducer:
     """Returns a reducer that outputs the minimum value of its first input.
 
@@ -712,9 +712,9 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def mode(
       # pylint: disable=invalid-name
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that will compute the mode of the inputs.
@@ -760,10 +760,10 @@ class Reducer(computedobject.ComputedObject):
   def percentile(
       percentiles: _arg_types.List,
       # pylint: disable=invalid-name
-      outputNames: Optional[_arg_types.List] = None,
-      maxBuckets: Optional[_arg_types.Integer] = None,
-      minBucketWidth: Optional[_arg_types.Number] = None,
-      maxRaw: Optional[_arg_types.Integer] = None,
+      outputNames: _arg_types.List | None = None,
+      maxBuckets: _arg_types.Integer | None = None,
+      minBucketWidth: _arg_types.Number | None = None,
+      maxRaw: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that will compute the specified percentiles.
@@ -818,8 +818,8 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def ridgeRegression(
       numX: _arg_types.Integer,  # pylint: disable=invalid-name
-      numY: Optional[_arg_types.Integer] = None,  # pylint: disable=invalid-name
-      lambda_: Optional[_arg_types.Number] = None,
+      numY: _arg_types.Integer | None = None,  # pylint: disable=invalid-name
+      lambda_: _arg_types.Number | None = None,
       **kwargs,
   ) -> Reducer:
     # pylint: disable=g-doc-args
@@ -862,8 +862,8 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def robustLinearRegression(
       numX: _arg_types.Integer,  # pylint: disable=invalid-name
-      numY: Optional[_arg_types.Integer] = None,  # pylint: disable=invalid-name
-      beta: Optional[_arg_types.Number] = None,
+      numY: _arg_types.Integer | None = None,  # pylint: disable=invalid-name
+      beta: _arg_types.Number | None = None,
   ) -> Reducer:
     """Returns a robust linear regression reducer.
 
@@ -979,7 +979,7 @@ class Reducer(computedobject.ComputedObject):
   def toCollection(
       # pylint: disable=invalid-name
       propertyNames: _arg_types.List,
-      numOptional: Optional[_arg_types.Integer] = None,
+      numOptional: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that collects its inputs into a FeatureCollection.
@@ -998,8 +998,8 @@ class Reducer(computedobject.ComputedObject):
   @staticmethod
   def toList(
       # pylint: disable=invalid-name
-      tupleSize: Optional[_arg_types.Integer] = None,
-      numOptional: Optional[_arg_types.Integer] = None,
+      tupleSize: _arg_types.Integer | None = None,
+      numOptional: _arg_types.Integer | None = None,
       # pylint: enable=invalid-name
   ) -> Reducer:
     """Returns a reducer that collects its inputs into a list.

ee/serializer.py

@@ -4,7 +4,7 @@ import collections
 import datetime
 import hashlib
 import json
-from typing import Any, Dict, List, Optional, Set
+from typing import Any
 
 from ee import _cloud_api_utils
 from ee import _utils
@@ -32,23 +32,23 @@ def DatetimeToMicroseconds(date: datetime.datetime) -> int:
 
 class Serializer:
   """A serializer for EE object trees."""
-  unbound_name: Optional[str]
+  unbound_name: str | None
 
   # Whether the encoding should factor out shared subtrees.
   _is_compound: bool
   _for_cloud_api: bool
   # A list of shared subtrees as [name, value] pairs.
-  _scope: List[str]
+  _scope: list[str]
   # A lookup table from object hash to subtree names as stored in self._scope
-  _encoded: Dict[Any, Any]
+  _encoded: dict[Any, Any]
   # A lookup table from object ID as retrieved by id() to md5 hash values.
-  _hashcache: Dict[Any, Any]
+  _hashcache: dict[Any, Any]
 
   def __init__(
       self,
       is_compound: bool = True,
       for_cloud_api: bool = False,
-      unbound_name: Optional[str] = None,
+      unbound_name: str | None = None,
   ):
     """Constructs a serializer.
 
@@ -160,8 +160,8 @@ class Serializer:
           'type':
               'Dictionary',
           'value':
-              dict([(key, self._encode_value(value))
-                    for key, value in obj.items()])
+              {key: self._encode_value(value)
+                    for key, value in obj.items()}
       }
     else:
       raise ee_exception.EEException('Cannot encode object: %s' % obj)
@@ -280,7 +280,7 @@ def encode(
     obj: Any,
     is_compound: bool = True,
     for_cloud_api: bool = True,
-    unbound_name: Optional[str] = None,
+    unbound_name: str | None = None,
 ) -> Any:
   """Serialize an object to a JSON-compatible structure for API calls.
 
@@ -358,7 +358,7 @@ class _ExpressionOptimizer:
   - Collapse dicts and arrays of constants to constant dicts/arrays.
   """
 
-  def __init__(self, result: Any, values: Optional[Any] = None):
+  def __init__(self, result: Any, values: Any | None = None):
     """Builds an ExpressionOptimizer.
 
     Args:
@@ -380,12 +380,12 @@ class _ExpressionOptimizer:
   def _is_compound(self) -> bool:
     return self._values is not None
 
-  def _find_single_uses(self) -> Set[Any]:
+  def _find_single_uses(self) -> set[Any]:
     """Finds the names of all named values that are referred to only once."""
     reference_counts = collections.defaultdict(int)
     reference_counts[self._result] += 1
 
-    def _contained_reference(value: Any) -> Optional[Any]:
+    def _contained_reference(value: Any) -> Any | None:
       """Gets a contained reference from a ValueNode, if there is one."""
       if 'functionDefinitionValue' in value:
         return value['functionDefinitionValue']['body']
@@ -403,8 +403,8 @@ class _ExpressionOptimizer:
         reference_counts[reference] += 1
 
     self._visit_all_values_in_expression(increment_reference_count)
-    return set(reference for reference, count in reference_counts.items()
-               if count == 1)
+    return {reference for reference, count in reference_counts.items()
+               if count == 1}
 
   def optimize(self) -> Any:
     """Optimises the expression, returning the optimised form."""

ee/table_converter.py

@@ -1,6 +1,7 @@
 """Converters used in the table data fetching methods."""
 
-from typing import Any, Dict, Iterator, List, Optional, Type, Union
+from collections.abc import Iterator
+from typing import Any
 
 
 class TableConverter:
@@ -24,7 +25,7 @@ class PandasConverter(TableConverter):
 
   def _convert_to_records(
       self, features: Iterator[Any]
-  ) -> Iterator[Dict[str, Any]]:
+  ) -> Iterator[dict[str, Any]]:
     for feature in features:
       yield {
           'geo': feature.get('geometry'),
@@ -46,20 +47,20 @@ class GeoPandasConverter(TableConverter):
         self._materialize_features(features)
     )
 
-  def _materialize_features(self, features: Iterator[Any]) -> List[Any]:
+  def _materialize_features(self, features: Iterator[Any]) -> list[Any]:
     """Materializes the features, making several requests if necessary."""
     return list(features)
 
 
-_TABLE_DATA_CONVERTERS: Dict[str, Type[TableConverter]] = {
+_TABLE_DATA_CONVERTERS: dict[str, type[TableConverter]] = {
     'PANDAS_DATAFRAME': PandasConverter,
     'GEOPANDAS_GEODATAFRAME': GeoPandasConverter,
 }
 
 
 def from_file_format(
-    file_format: Union[str, TableConverter]
-) -> Optional[TableConverter]:
+    file_format: str | TableConverter
+) -> TableConverter | None:
   if isinstance(file_format, TableConverter):
     return file_format
   if file_format in _TABLE_DATA_CONVERTERS:

ee/terrain.py

@@ -1,8 +1,6 @@
 """A namespace for Terrain."""
 from __future__ import annotations
 
-from typing import Optional
-
 from ee import _arg_types
 from ee import apifunction
 from ee import image as ee_image
@@ -15,7 +13,7 @@ class Terrain:
 
   def __init__(self):
     raise RuntimeError(
-        self.__name__
+        self.__class__.__name__
         + ' should not be used as an object. Only direct usage of Terrain'
         ' static methods is allowed. For example, use this: '
         ' `ee.Terrain.aspect(...)`'
@@ -54,8 +52,8 @@ class Terrain:
   @staticmethod
   def fillMinima(
       image: _arg_types.Image,
-      borderValue: Optional[_arg_types.Integer] = None,
-      neighborhood: Optional[_arg_types.Integer] = None,
+      borderValue: _arg_types.Integer | None = None,
+      neighborhood: _arg_types.Integer | None = None,
   ) -> ee_image.Image:
     """Returns an ee.Image with local minima filled.
 
@@ -76,8 +74,8 @@ class Terrain:
       image: _arg_types.Image,
       azimuth: _arg_types.Number,
       zenith: _arg_types.Number,
-      neighborhoodSize: Optional[_arg_types.Integer] = None,
-      hysteresis: Optional[_arg_types.Bool] = None,
+      neighborhoodSize: _arg_types.Integer | None = None,
+      hysteresis: _arg_types.Bool | None = None,
   ) -> ee_image.Image:
     """Returns an ee.Image with the hill shadow.
 
@@ -109,8 +107,8 @@ class Terrain:
   @staticmethod
   def hillshade(
       input: _arg_types.Image,
-      azimuth: Optional[_arg_types.Number] = None,
-      elevation: Optional[_arg_types.Number] = None,
+      azimuth: _arg_types.Number | None = None,
+      elevation: _arg_types.Number | None = None,
   ) -> ee_image.Image:
     """Returns an ee.Image with a simple hillshade from a DEM.
 

ee/tests/_cloud_api_utils_test.py

@@ -16,10 +16,6 @@ from ee import ee_exception
 
 class CloudApiUtilsTest(unittest.TestCase):
 
-  def setUp(self):
-    super().setUp()
-    _cloud_api_utils.set_cloud_api_user_project('earthengine-legacy')
-
   def test_build_cloud_resource(self):
     base = 'https://earthengine.basetest'
     path = '$discovery/rest?version=v1&prettyPrint=false'
@@ -171,7 +167,10 @@ class CloudApiUtilsTest(unittest.TestCase):
   def test_convert_task_id_to_operation_name(self):
     self.assertEqual(
         'projects/earthengine-legacy/operations/taskId',
-        _cloud_api_utils.convert_task_id_to_operation_name('taskId'))
+        _cloud_api_utils.convert_task_id_to_operation_name(
+            'earthengine-legacy', 'taskId'
+        ),
+    )
 
   def test_encode_number_as_cloud_value(self):
     self.assertEqual({
@@ -414,7 +413,7 @@ class CloudApiUtilsTest(unittest.TestCase):
     self.assertEqual({'width': 123, 'height': 123},
                      _cloud_api_utils.convert_to_grid_dimensions(123))
     self.assertEqual({'width': 123, 'height': 123},
-                     _cloud_api_utils.convert_to_grid_dimensions((123)))
+                     _cloud_api_utils.convert_to_grid_dimensions(123))
     self.assertEqual({'width': 123, 'height': 234},
                      _cloud_api_utils.convert_to_grid_dimensions((123, 234)))
 
@@ -551,6 +550,22 @@ class CloudApiUtilsTest(unittest.TestCase):
         expected,
         _cloud_api_utils.convert_sources_to_one_platform_sources(old_sources))
 
+  def test_convert_to_operation_state(self):
+    def convert(state):
+      return _cloud_api_utils.convert_to_operation_state(state)
+
+    self.assertEqual('CANCELLED', convert('CANCELLED'))
+    self.assertEqual('CANCELLING', convert('CANCEL_REQUESTED'))
+    self.assertEqual('CANCELLING', convert('CANCELLING'))
+    self.assertEqual('FAILED', convert('FAILED'))
+    self.assertEqual('PENDING', convert('PENDING'))
+    self.assertEqual('PENDING', convert('READY'))
+    self.assertEqual('RUNNING', convert('RUNNING'))
+    self.assertEqual('SUCCEEDED', convert('COMPLETED'))
+    self.assertEqual('SUCCEEDED', convert('SUCCEEDED'))
+    self.assertEqual('UNKNOWN', convert('random_string'))
+    self.assertEqual('UNKNOWN', convert('UNKNOWN'))
+
 
 if __name__ == '__main__':
   unittest.main()

ee/tests/_helpers_test.py

@@ -7,14 +7,67 @@ name since that is the name we want to ensure works.
 """
 
 import io
+import json
 import unittest
+from unittest import mock
 
 import unittest
 import ee
+from ee import _helpers
 from ee import apifunction
 from ee import apitestcase
 from ee import computedobject
 from ee import ee_exception
+from ee import oauth
+
+
+class ServiceAccountCredentialsTest(unittest.TestCase):
+
+  def test_no_args(self):
+    with self.assertRaisesRegex(ValueError, 'At least one of'):
+      ee.ServiceAccountCredentials()
+
+  @mock.patch('google.oauth2.service_account.Credentials')
+  def test_json_file(self, mock_credentials):
+    ee.ServiceAccountCredentials(key_file='foo.json')
+    mock_credentials.from_service_account_file.assert_called_with(
+        'foo.json', scopes=oauth.SCOPES
+    )
+
+  @mock.patch('google.oauth2.service_account.Credentials')
+  def test_json_key_data(self, mock_credentials):
+    key_data = {'client_email': 'foo@bar.com'}
+    ee.ServiceAccountCredentials(key_data=json.dumps(key_data))
+    mock_credentials.from_service_account_info.assert_called_with(
+        key_data, scopes=oauth.SCOPES
+    )
+
+  @mock.patch('google.auth.crypt.RSASigner')
+  @mock.patch('google.oauth2.service_account.Credentials')
+  def test_pem_key_data(self, mock_credentials, mock_signer):
+    ee.ServiceAccountCredentials(email='foo@bar.com', key_data='pem_key_data')
+    mock_signer.from_string.assert_called_with('pem_key_data')
+    self.assertEqual(
+        mock_credentials.call_args[0][1], 'foo@bar.com'
+    )
+
+  @mock.patch('google.auth.crypt.RSASigner')
+  @mock.patch('google.oauth2.service_account.Credentials')
+  def test_pem_file(self, mock_credentials, mock_signer):
+    with mock.patch.object(
+        _helpers, 'open', mock.mock_open(read_data='pem_key_data')
+    ):
+      ee.ServiceAccountCredentials(email='foo@bar.com', key_file='foo.pem')
+    mock_signer.from_string.assert_called_with('pem_key_data')
+    self.assertEqual(
+        mock_credentials.call_args[0][1], 'foo@bar.com'
+    )
+
+  def test_bad_json_key_data(self):
+    # This causes a different failure based on where the test is run.
+    message = r'Could not deserialize key data|No key could be detected'
+    with self.assertRaisesRegex(ValueError, message):
+      ee.ServiceAccountCredentials(key_data='not json')
 
 
 class ProfilingTest(apitestcase.ApiTestCase):
@@ -25,23 +78,23 @@ class ProfilingTest(apitestcase.ApiTestCase):
     is_get_profiles = isinstance(
         value, computedobject.ComputedObject
     ) and value.func == apifunction.ApiFunction.lookup('Profile.getProfiles')
-    return 'hooked=%s getProfiles=%s' % (hooked, is_get_profiles)
+    return f'hooked={hooked} getProfiles={is_get_profiles}'
 
-  def testProfilePrinting(self):
+  def test_profile_printing(self):
     ee.data.computeValue = self.MockValue
     out = io.StringIO()
     with ee.profilePrinting(destination=out):
       self.assertEqual('hooked=True getProfiles=False', ee.Number(1).getInfo())
     self.assertEqual('hooked=False getProfiles=True', out.getvalue())
 
-  def testProfilePrintingDefaultSmoke(self):
+  def test_profile_printing_default_smoke(self):
     # This will print to sys.stderr, so we can't make any assertions about the
     # output. But we can check that it doesn't fail.
     ee.data.computeValue = self.MockValue
     with ee.profilePrinting():
       self.assertEqual('hooked=True getProfiles=False', ee.Number(1).getInfo())
 
-  def testProfilePrintingErrorGettingProfiles(self):
+  def test_profile_printing_error_getting_profiles(self):
     ee.data.computeValue = self.MockValue
     mock = unittest.mock.Mock()
     mock.call.side_effect = ee_exception.EEException('test')

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