mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
Saugat Pachhai (सौगात)
c73764156e
This adds support for custom metadata in outputs under `meta` key. Also adds `--meta` flag to `add`/`import`/`import-url` that supports `key=value` values.
48 lines
1.2 KiB
Python
48 lines
1.2 KiB
Python
import pytest
|
|
|
|
from dvc.cli import parse_args
|
|
from tests.utils.asserts import called_once_with_subset
|
|
|
|
|
|
@pytest.mark.parametrize(
|
|
"func, args",
|
|
[
|
|
("add", ("add", "model.pkl")),
|
|
("imp", ("import", "https://github.com/org/repo", "model.pkl")),
|
|
("imp_url", ("import-url", "https://dvc.org/data.xml", "data.xml")),
|
|
],
|
|
ids=["add", "import", "import-url"],
|
|
)
|
|
def test_add_annotations(mocker, func, args):
|
|
mocker.patch("dvc.repo.Repo")
|
|
cli_args = parse_args(
|
|
[
|
|
*args,
|
|
"--label",
|
|
"example-get-started",
|
|
"--label",
|
|
"model-registry",
|
|
"--type",
|
|
"model",
|
|
"--desc",
|
|
"description",
|
|
"--meta",
|
|
"key1=value1",
|
|
"--meta",
|
|
"key2=value2",
|
|
]
|
|
)
|
|
|
|
_, *posargs = args
|
|
cmd = cli_args.func(cli_args)
|
|
|
|
assert cmd.run() == 0
|
|
assert called_once_with_subset(
|
|
getattr(cmd.repo, func),
|
|
posargs,
|
|
desc="description",
|
|
type="model",
|
|
labels=["example-get-started", "model-registry"],
|
|
meta={"key1": "value1", "key2": "value2"},
|
|
)
|