diff --git a/tests/test_metrics.py b/tests/test_metrics.py index 68f08145e..2af0d578c 100644 --- a/tests/test_metrics.py +++ b/tests/test_metrics.py @@ -57,6 +57,25 @@ class TestMetrics(TestDvc): self.assertTrue(ret['baz']['metric_htsv'] == ['baz']) +class TestMetricsReproCLI(TestDvc): + def test(self): + + stage = self.dvc.run(outs_no_cache=['metrics'], + cmd='python {} {} {}'.format(self.CODE, self.FOO, 'metrics')) + + ret = main(['metrics', 'add', 'metrics']) + self.assertEqual(ret, 0) + + ret = main(['repro', '-m', stage.path]) + self.assertEqual(ret, 0) + + ret = main(['metrics', 'remove', 'metrics']) + self.assertEqual(ret, 0) + + ret = main(['repro', '-f', '-m', stage.path]) + self.assertEqual(ret, 0) + + class TestMetricsCLI(TestMetrics): def test(self): #FIXME check output diff --git a/tests/test_updater.py b/tests/test_updater.py new file mode 100644 index 000000000..36cbf4859 --- /dev/null +++ b/tests/test_updater.py @@ -0,0 +1,18 @@ +import os + +from dvc.updater import Updater + +from tests.basic_env import TestDvc + + +class TestDvcUpdater(TestDvc): + def test(self): + # NOTE: need to temporarily unset CI env to allow updater to work + env = os.environ.copy() + os.environ['CI'] = 'False' + + self.dvc.updater.check() + self.dvc.updater.check() + self.dvc.updater.check() + + os.environ = env