1  import pytest
 2  
 3  import mlflow.deployments.utils
 4  from mlflow.deployments.interface import get_deploy_client
 5  
 6  mlflow.deployments.utils._deployments_target = None
 7  
 8  
 9  def test_get_deploy_client_no_args():
10      mlflow.deployments.utils._deployments_target = None
11      assert get_deploy_client() is None
12  
13  
14  def test_get_deploy_client_none():
15      mlflow.deployments.utils._deployments_target = None
16      assert get_deploy_client(None) is None
17  
18  
19  def test_get_deploy_client_from_set_deployments_target():
20      from mlflow.deployments import set_deployments_target
21  
22      set_deployments_target("databricks")
23      assert get_deploy_client(None) is not None
24  
25  
26  @pytest.fixture
27  def set_deployment_envs(monkeypatch):
28      monkeypatch.setenv("MLFLOW_DEPLOYMENTS_TARGET", "databricks")
29  
30  
31  def test_get_deploy_client_from_env(set_deployment_envs):
32      assert get_deploy_client(None) is not None