From 6acd1b4d6f228e84aabee9e924236de90815cc9c Mon Sep 17 00:00:00 2001 From: ewezy Date: Thu, 7 Nov 2024 14:46:05 +0800 Subject: [PATCH] Fix incorrect function name called in unit test --- python/sdk/test/utils_unit_test.py | 8 ++++---- 1 file changed, 4 insertions(+), 4 deletions(-) diff --git a/python/sdk/test/utils_unit_test.py b/python/sdk/test/utils_unit_test.py index 9763b8c67..b63660ac9 100644 --- a/python/sdk/test/utils_unit_test.py +++ b/python/sdk/test/utils_unit_test.py @@ -12,7 +12,7 @@ # See the License for the specific language governing permissions and # limitations under the License. -from merlin.util import guess_mlp_ui_url, valid_name_check, get_bucket_name, get_artifact_path +from merlin.util import guess_mlp_ui_url, valid_name_check, get_blob_storage_scheme, get_bucket_name, get_artifact_path import pytest @@ -37,10 +37,10 @@ def test_name_check(): @pytest.mark.unit def test_get_blob_storage_scheme(): gcs_artifact_uri = 'gs://some-bucket/mlflow/81/ddd' - assert test_get_blob_storage_scheme(gcs_artifact_uri) == 'gs' + assert get_blob_storage_scheme(gcs_artifact_uri) == 'gs' s3_artifact_uri = 's3://some-bucket/mlflow/81/ddd' - assert test_get_blob_storage_scheme(s3_artifact_uri) == 's3' + assert get_blob_storage_scheme(s3_artifact_uri) == 's3' @pytest.mark.unit @@ -62,7 +62,7 @@ def test_get_artifact_path(): ("http://console.mydomain.com/merlin/api", "http://console.mydomain.com/merlin"), ("https://console.mydomain.com/merlin/api", "https://console.mydomain.com/merlin"), ("console.mydomain.com/merlin/api", "http://console.mydomain.com/merlin"), -]) + @pytest.mark.unit def test_guess_mlp_ui_url(mlp_api_url, expected): assert guess_mlp_ui_url(mlp_api_url) == expected