earthengine-api/python/ee/tests/_state_test.py
Nate Schmitz f22d01a6db state: Consolidate EE global state in an EEState object.
PiperOrigin-RevId: 812887334
2025-09-29 12:03:38 -07:00

39 lines
851 B
Python

#!/usr/bin/env python3
"""Tests for ee._state."""
import unittest
from ee import _state
class StateTest(unittest.TestCase):
def setUp(self):
super().setUp()
_state.reset_state()
def test_get_state(self):
state = _state.get_state()
self.assertIsInstance(state, _state.EEState)
self.assertEqual(state, _state.EEState())
def test_update_state(self):
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')
def test_reset_state(self):
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()