mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +01:00
Casper da Costa-Luis
2bb7257a06
* test: clean flake8 config root clutter * test: pre-commit: add isort Compatible with black (see https://github.com/timothycrosley/isort/issues/694) * lint: reorder imports * lint: more complex imports * lint: fix deepsource * config: isort: add first party dvc * tests: restyled: add isort * temp commit to revert before merge * Revert "temp commit to revert before merge" This reverts commit 11db0fd8e95fe891597cbb9bf804480909b19e24.
174 lines
4.3 KiB
Python
174 lines
4.3 KiB
Python
import os
|
|
|
|
from dvc.cli import parse_args
|
|
from dvc.command.add import CmdAdd
|
|
from dvc.command.base import CmdBase
|
|
from dvc.command.checkout import CmdCheckout
|
|
from dvc.command.config import CmdConfig
|
|
from dvc.command.data_sync import CmdDataPull, CmdDataPush
|
|
from dvc.command.gc import CmdGC
|
|
from dvc.command.init import CmdInit
|
|
from dvc.command.remove import CmdRemove
|
|
from dvc.command.repro import CmdRepro
|
|
from dvc.command.run import CmdRun
|
|
from dvc.command.status import CmdDataStatus
|
|
from dvc.exceptions import DvcException
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
class TestArgParse(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["init"])
|
|
self.assertIsInstance(args.func(args), CmdInit)
|
|
|
|
|
|
class TestRun(TestDvc):
|
|
def test(self):
|
|
dep1 = "dep1"
|
|
dep2 = "dep2"
|
|
|
|
out1 = "out1"
|
|
out2 = "out2"
|
|
|
|
out_no_cache1 = "out_no_cache1"
|
|
out_no_cache2 = "out_no_cache2"
|
|
|
|
fname = "dvc.dvc"
|
|
cmd = "cmd"
|
|
arg1 = "arg1"
|
|
arg2 = "arg2"
|
|
|
|
args = parse_args(
|
|
[
|
|
"run",
|
|
"-d",
|
|
dep1,
|
|
"--deps",
|
|
dep2,
|
|
"-o",
|
|
out1,
|
|
"--outs",
|
|
out2,
|
|
"-O",
|
|
out_no_cache1,
|
|
"--outs-no-cache",
|
|
out_no_cache2,
|
|
"-f",
|
|
fname,
|
|
"--file",
|
|
fname,
|
|
cmd,
|
|
arg1,
|
|
arg2,
|
|
]
|
|
)
|
|
|
|
self.assertIsInstance(args.func(args), CmdRun)
|
|
self.assertEqual(args.deps, [dep1, dep2])
|
|
self.assertEqual(args.outs, [out1, out2])
|
|
self.assertEqual(args.outs_no_cache, [out_no_cache1, out_no_cache2])
|
|
self.assertEqual(args.file, fname)
|
|
self.assertEqual(args.command, [cmd, arg1, arg2])
|
|
|
|
|
|
class TestPull(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["pull"])
|
|
self.assertIsInstance(args.func(args), CmdDataPull)
|
|
|
|
|
|
class TestPush(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["push"])
|
|
self.assertIsInstance(args.func(args), CmdDataPush)
|
|
|
|
|
|
class TestStatus(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["status"])
|
|
self.assertIsInstance(args.func(args), CmdDataStatus)
|
|
|
|
|
|
class TestRepro(TestDvc):
|
|
def test(self):
|
|
target1 = "1"
|
|
target2 = "2"
|
|
|
|
args = parse_args(
|
|
["repro", target1, target2, "-f", "--force", "-s", "--single-item"]
|
|
)
|
|
|
|
self.assertIsInstance(args.func(args), CmdRepro)
|
|
self.assertEqual(args.targets, [target1, target2])
|
|
self.assertEqual(args.force, True)
|
|
self.assertEqual(args.single_item, True)
|
|
|
|
|
|
class TestRemove(TestDvc):
|
|
def test(self):
|
|
target1 = "1"
|
|
target2 = "2"
|
|
|
|
args = parse_args(["remove", target1, target2])
|
|
|
|
self.assertIsInstance(args.func(args), CmdRemove)
|
|
self.assertEqual(args.targets, [target1, target2])
|
|
|
|
|
|
class TestAdd(TestDvc):
|
|
def test(self):
|
|
target1 = "1"
|
|
target2 = "2"
|
|
|
|
args = parse_args(["add", target1, target2])
|
|
|
|
self.assertIsInstance(args.func(args), CmdAdd)
|
|
self.assertEqual(args.targets, [target1, target2])
|
|
|
|
|
|
class TestGC(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["gc"])
|
|
self.assertIsInstance(args.func(args), CmdGC)
|
|
|
|
|
|
class TestGCMultipleDvcRepos(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["gc", "-p", "/tmp/asdf", "/tmp/xyz"])
|
|
|
|
self.assertIsInstance(args.func(args), CmdGC)
|
|
|
|
self.assertEqual(args.repos, ["/tmp/asdf", "/tmp/xyz"])
|
|
|
|
|
|
class TestConfig(TestDvc):
|
|
def test(self):
|
|
name = "param"
|
|
value = "1"
|
|
|
|
args = parse_args(["config", "-u", "--unset", name, value])
|
|
|
|
self.assertIsInstance(args.func(args), CmdConfig)
|
|
self.assertEqual(args.unset, True)
|
|
self.assertEqual(args.name, name)
|
|
self.assertEqual(args.value, value)
|
|
|
|
|
|
class TestCheckout(TestDvc):
|
|
def test(self):
|
|
args = parse_args(["checkout"])
|
|
self.assertIsInstance(args.func(args), CmdCheckout)
|
|
|
|
|
|
class TestFindRoot(TestDvc):
|
|
def test(self):
|
|
os.chdir("..")
|
|
|
|
class A(object):
|
|
quiet = False
|
|
verbose = True
|
|
|
|
args = A()
|
|
with self.assertRaises(DvcException):
|
|
CmdBase(args)
|