From 95f3b49db8639bc03059f45d1d4d610c565c637a Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Sun, 1 Jul 2018 19:46:46 +0300 Subject: [PATCH 1/2] test: add basic test for metrics Signed-off-by: Ruslan Kuprieiev --- tests/test_metrics.py | 19 +++++++++++++++++++ 1 file changed, 19 insertions(+) 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 From 6219b28be6737844141e2817c86adb67d9fb75d7 Mon Sep 17 00:00:00 2001 From: Ruslan Kuprieiev Date: Sun, 1 Jul 2018 19:54:04 +0300 Subject: [PATCH 2/2] test: add test for Updater Signed-off-by: Ruslan Kuprieiev --- tests/test_updater.py | 18 ++++++++++++++++++ 1 file changed, 18 insertions(+) create mode 100644 tests/test_updater.py 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