mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-01 23:08:43 +01:00
46 lines
1.4 KiB
Python
46 lines
1.4 KiB
Python
import pytest
|
|
|
|
from dvc.repo.datasets import parse_url_and_type
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"url,expected",
|
|
[
|
|
("s3://bucket/path", {"type": "url", "url": "s3://bucket/path"}),
|
|
("gs://bucket/path", {"type": "url", "url": "gs://bucket/path"}),
|
|
("azure://container/path", {"type": "url", "url": "azure://container/path"}),
|
|
(
|
|
"remote://remote_name/path",
|
|
{"type": "url", "url": "remote://remote_name/path"},
|
|
),
|
|
("dvcx://dataset_name", {"type": "dvcx", "url": "dvcx://dataset_name"}),
|
|
(
|
|
"dvc+file:///home/user/repository",
|
|
{
|
|
"type": "dvc",
|
|
"url": "file:///home/user/repository",
|
|
},
|
|
),
|
|
(
|
|
"dvc://[email protected]:iterative/example-get-started.git",
|
|
{"type": "dvc", "url": "[email protected]:iterative/example-get-started.git"},
|
|
),
|
|
(
|
|
"dvc+https://github.com/iterative/example-get-started.git",
|
|
{
|
|
"type": "dvc",
|
|
"url": "https://github.com/iterative/example-get-started.git",
|
|
},
|
|
),
|
|
(
|
|
"dvc+ssh://github.com/iterative/example-get-started.git",
|
|
{
|
|
"type": "dvc",
|
|
"url": "ssh://github.com/iterative/example-get-started.git",
|
|
},
|
|
),
|
|
],
|
|
)
|
|
def test_url_parsing(url, expected):
|
|
assert parse_url_and_type(url) == expected
|