mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +01:00
GaoandSaugat Pachhai
96a97ec0e4
* fix #4838 `jobs` option for `dvc import` * option `jobs` effected * Pass the unit tests parse * Adding a functional test. * Pass `jobs` to `run` and remove some pylint * Update dvc/stage/__init__.py Co-authored-by: Saugat Pachhai <[email protected]> * remove unused param * Solve test fail on windows Co-authored-by: Saugat Pachhai <[email protected]>
75 lines
1.4 KiB
Python
75 lines
1.4 KiB
Python
from dvc.cli import parse_args
|
|
from dvc.command.imp import CmdImport
|
|
|
|
|
|
def test_import(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"import",
|
|
"repo_url",
|
|
"src",
|
|
"--out",
|
|
"out",
|
|
"--file",
|
|
"file",
|
|
"--rev",
|
|
"version",
|
|
"--desc",
|
|
"description",
|
|
"--jobs",
|
|
"3",
|
|
]
|
|
)
|
|
assert cli_args.func == CmdImport
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"repo_url",
|
|
path="src",
|
|
out="out",
|
|
fname="file",
|
|
rev="version",
|
|
no_exec=False,
|
|
desc="description",
|
|
jobs=3,
|
|
)
|
|
|
|
|
|
def test_import_no_exec(mocker):
|
|
cli_args = parse_args(
|
|
[
|
|
"import",
|
|
"repo_url",
|
|
"src",
|
|
"--out",
|
|
"out",
|
|
"--file",
|
|
"file",
|
|
"--rev",
|
|
"version",
|
|
"--no-exec",
|
|
"--desc",
|
|
"description",
|
|
]
|
|
)
|
|
|
|
cmd = cli_args.func(cli_args)
|
|
m = mocker.patch.object(cmd.repo, "imp", autospec=True)
|
|
|
|
assert cmd.run() == 0
|
|
|
|
m.assert_called_once_with(
|
|
"repo_url",
|
|
path="src",
|
|
out="out",
|
|
fname="file",
|
|
rev="version",
|
|
no_exec=True,
|
|
desc="description",
|
|
jobs=None,
|
|
)
|