wehub-resource-sync
14 lines
570 B
Python
14 lines
570 B
Python
from mlflow.tracking.context.system_environment_context import SystemEnvironmentContext
|
|
|
|
|
|
def test_system_environment_context_in_context(monkeypatch):
|
|
monkeypatch.setenv("MLFLOW_RUN_CONTEXT", '{"A": "B"}')
|
|
assert SystemEnvironmentContext().in_context()
|
|
monkeypatch.delenv("MLFLOW_RUN_CONTEXT", raising=True)
|
|
assert not SystemEnvironmentContext().in_context()
|
|
|
|
|
|
def test_system_environment_context_tags(monkeypatch):
|
|
monkeypatch.setenv("MLFLOW_RUN_CONTEXT", '{"A": "B", "C": "D"}')
|
|
assert SystemEnvironmentContext().tags() == {"A": "B", "C": "D"}
|