1  from mlflow.tracking.context.system_environment_context import SystemEnvironmentContext
 2  
 3  
 4  def test_system_environment_context_in_context(monkeypatch):
 5      monkeypatch.setenv("MLFLOW_RUN_CONTEXT", '{"A": "B"}')
 6      assert SystemEnvironmentContext().in_context()
 7      monkeypatch.delenv("MLFLOW_RUN_CONTEXT", raising=True)
 8      assert not SystemEnvironmentContext().in_context()
 9  
10  
11  def test_system_environment_context_tags(monkeypatch):
12      monkeypatch.setenv("MLFLOW_RUN_CONTEXT", '{"A": "B", "C": "D"}')
13      assert SystemEnvironmentContext().tags() == {"A": "B", "C": "D"}