def test_start_run_existing_run_from_environment_with_set_environment(empty_active_run_stack):
mock_run = mock.Mock()
mock_run.info.lifecycle_stage = LifecycleStage.ACTIVE
run_id = uuid.uuid4().hex
env_patch = mock.patch.dict("os.environ", {_RUN_ID_ENV_VAR: run_id})
with env_patch, mock.patch.object(MlflowClient, "get_run", return_value=mock_run):
with pytest.raises(MlflowException):
set_experiment("test-run")
active_run = start_run()
def test_start_run_existing_run_from_environment_with_set_environment(
empty_active_run_stack,
): # pylint: disable=unused-argument
mock_run = mock.Mock()
mock_run.info.lifecycle_stage = LifecycleStage.ACTIVE
run_id = uuid.uuid4().hex
env_patch = mock.patch.dict("os.environ", {_RUN_ID_ENV_VAR: run_id})
with env_patch, mock.patch.object(MlflowClient, "get_run", return_value=mock_run):
with pytest.raises(
MlflowException, match="active run ID does not match environment run ID"
):
set_experiment("test-run")
start_run()