code: def test_env_context(self):
      default_env = env.get_env()
      self.assertIsInstance(default_env, env._DefaultEnv)
      test_env = _TestEnv()
      with test_env:
          self.assertIs(env.get_env(), test_env)
      self.assertIs(env.get_env(), default_env)
  
file_path: tfx/tfx/orchestration/experimental/core/env_test.py
imports: import tensorflow
  from tfx.orchestration.experimental.core import env
  from tfx.orchestration.experimental.core import test_utils
  from tfx.utils import status
label_desc: Use image node_id 1 for calling the EnvTest obj's underlying member method code with example usage: obj.test_env_context() without return types