mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-05 23:09:32 +01:00
Ruslan Kuprieiev
8455b7bece
* params: diff: introduce --all Makes `dvc params diff` also show unchanged params. * params: diff: show all present parameters
103 lines
2.7 KiB
Python
103 lines
2.7 KiB
Python
import logging
|
|
|
|
from dvc.cli import parse_args
|
|
from dvc.command.params import CmdParamsDiff, _show_diff
|
|
|
|
|
|
def test_params_diff(dvc, mocker):
|
|
cli_args = parse_args(
|
|
["params", "diff", "HEAD~10", "HEAD~1", "--all", "--show-json"]
|
|
)
|
|
assert cli_args.func == CmdParamsDiff
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch("dvc.repo.params.diff.diff", return_value={})
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
cmd.repo, a_rev="HEAD~10", b_rev="HEAD~1", all=True,
|
|
)
|
|
|
|
|
|
def test_params_diff_changed():
|
|
assert _show_diff({"params.yaml": {"a.b.c": {"old": 1, "new": 2}}}) == (
|
|
" Path Param Old New\n" "params.yaml a.b.c 1 2 "
|
|
)
|
|
|
|
|
|
def test_params_diff_list():
|
|
assert _show_diff(
|
|
{"params.yaml": {"a.b.c": {"old": 1, "new": [2, 3]}}}
|
|
) == (
|
|
" Path Param Old New \n"
|
|
"params.yaml a.b.c 1 [2, 3]"
|
|
)
|
|
|
|
|
|
def test_params_diff_unchanged():
|
|
assert _show_diff(
|
|
{"params.yaml": {"a.b.d": {"old": "old", "new": "new"}}}
|
|
) == (
|
|
" Path Param Old New\n" "params.yaml a.b.d old new"
|
|
)
|
|
|
|
|
|
def test_params_diff_no_changes():
|
|
assert _show_diff({}) == ""
|
|
|
|
|
|
def test_params_diff_new():
|
|
assert _show_diff(
|
|
{"params.yaml": {"a.b.d": {"old": None, "new": "new"}}}
|
|
) == (
|
|
" Path Param Old New\n" "params.yaml a.b.d None new"
|
|
)
|
|
|
|
|
|
def test_params_diff_deleted():
|
|
assert _show_diff(
|
|
{"params.yaml": {"a.b.d": {"old": "old", "new": None}}}
|
|
) == (
|
|
" Path Param Old New \n" "params.yaml a.b.d old None"
|
|
)
|
|
|
|
|
|
def test_params_diff_prec():
|
|
assert _show_diff(
|
|
{"params.yaml": {"train.lr": {"old": 0.0042, "new": 0.0043}}}
|
|
) == (
|
|
" Path Param Old New \n"
|
|
"params.yaml train.lr 0.0042 0.0043"
|
|
)
|
|
|
|
|
|
def test_params_diff_show_json(dvc, mocker, caplog):
|
|
cli_args = parse_args(
|
|
["params", "diff", "HEAD~10", "HEAD~1", "--show-json"]
|
|
)
|
|
cmd = cli_args.func(cli_args)
|
|
mocker.patch(
|
|
"dvc.repo.params.diff.diff", return_value={"params.yaml": {"a": "b"}}
|
|
)
|
|
with caplog.at_level(logging.INFO, logger="dvc"):
|
|
assert cmd.run() == 0
|
|
assert '{"params.yaml": {"a": "b"}}\n' in caplog.text
|
|
|
|
|
|
def test_params_diff_sorted():
|
|
assert _show_diff(
|
|
{
|
|
"params.yaml": {
|
|
"x.b": {"old": 5, "new": 6},
|
|
"a.d.e": {"old": 3, "new": 4},
|
|
"a.b.c": {"old": 1, "new": 2},
|
|
}
|
|
}
|
|
) == (
|
|
" Path Param Old New\n"
|
|
"params.yaml a.b.c 1 2 \n"
|
|
"params.yaml a.d.e 3 4 \n"
|
|
"params.yaml x.b 5 6 "
|
|
)
|