Ejemplos de get_registry_uri en Python

Ejemplo n.º 1

Archivo: test_utils.py Proyecto: anjosma/mlflow_experiments

def test_set_get_empty_registry_uri():
    with mock.patch("mlflow.tracking._model_registry.utils.get_tracking_uri") \
            as get_tracking_uri_mock:
        get_tracking_uri_mock.return_value = None
        set_registry_uri("")
        assert get_registry_uri() is None
        set_registry_uri(None)

Ejemplo n.º 2

Archivo: test_utils.py Proyecto: anjosma/mlflow_experiments

def test_default_get_registry_uri_with_tracking_uri_set():
    tracking_uri = "databricks://tracking_werohoz"
    with mock.patch("mlflow.tracking._model_registry.utils.get_tracking_uri") \
            as get_tracking_uri_mock:
        get_tracking_uri_mock.return_value = tracking_uri
        set_registry_uri(None)
        assert get_registry_uri() == tracking_uri

Ejemplo n.º 3

Archivo: test_utils.py Proyecto: anjosma/mlflow_experiments

def test_set_get_registry_uri():
    with mock.patch("mlflow.tracking._model_registry.utils.get_tracking_uri") \
            as get_tracking_uri_mock:
        get_tracking_uri_mock.return_value = "databricks://tracking_sldkfj"
        uri = "databricks://registry/path"
        set_registry_uri(uri)
        assert get_registry_uri() == uri
        set_registry_uri(None)