Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-02 07:18:25 +01:00
Code Issues Packages Projects Releases Wiki Activity

exp init: minor refactoring

Browse Source
This commit is contained in:
Saugat Pachhai (सौगात)
2021-12-07 08:40:46 +05:45
committed by Saugat Pachhai
parent 4385720fab
commit cd37ebca37
3 changed files with 19 additions and 16 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/command/test_experiments.py
+8 -7
View File
@@ -613,7 +613,7 @@ def test_experiments_init(dvc, scm, mocker, capsys, extra_args):
assert cmd.run() == 0
m.assert_called_once_with(
ANY(Repo),
name=None,
name="train",
type="default",
defaults={
"code": "src",
@@ -653,7 +653,7 @@ def test_experiments_init_config(dvc, scm, mocker):
assert cmd.run() == 0
m.assert_called_once_with(
ANY(Repo),
name=None,
name="train",
type="default",
defaults={
"code": "new_src",
@@ -678,7 +678,7 @@ def test_experiments_init_explicit(dvc, mocker):
assert cmd.run() == 0
m.assert_called_once_with(
ANY(Repo),
name=None,
name="train",
type="default",
defaults={},
overrides={"cmd": "cmd"},
@@ -710,9 +710,9 @@ def test_experiments_init_cmd_not_required_for_interactive_mode(dvc, mocker):
@pytest.mark.parametrize(
"extra_args, expected_kw",
[
(["--type", "default"], {"type": "default"}),
(["--type", "dl"], {"type": "dl"}),
(["--force"], {"force": True}),
(["--type", "default"], {"type": "default", "name": "train"}),
(["--type", "dl"], {"type": "dl", "name": "train"}),
(["--force"], {"force": True, "name": "train"}),
(
["--name", "name", "--type", "dl"],
{"name": "name", "type": "dl"},
@@ -735,6 +735,7 @@ def test_experiments_init_cmd_not_required_for_interactive_mode(dvc, mocker):
"live",
],
{
"name": "train",
"overrides": {
"plots": "p",
"models": "m",
@@ -744,7 +745,7 @@ def test_experiments_init_cmd_not_required_for_interactive_mode(dvc, mocker):
"data": "d",
"live": "live",
"cmd": "cmd",
}
},
},
),
],