From 21388952e9cd7e9d4259d63843a7ff782aa09ca2 Mon Sep 17 00:00:00 2001 From: Aman Sharma Date: Fri, 10 May 2019 09:31:30 +0530 Subject: [PATCH] Write tests for `dvc version` --- tests/func/test_version.py | 20 ++++++++++++++++++++ 1 file changed, 20 insertions(+) create mode 100644 tests/func/test_version.py diff --git a/tests/func/test_version.py b/tests/func/test_version.py new file mode 100644 index 000000000..ab7ea4593 --- /dev/null +++ b/tests/func/test_version.py @@ -0,0 +1,20 @@ +from dvc.main import main +from dvc.command.version import CmdVersion +from dvc.cli import parse_args + +from tests.basic_env import TestDvc + + +class TestVersion(TestDvc): + def test_run(self): + ret = main(["version"]) + self.assertEqual(ret, 0) + + +def test_info(dvc, mocker, caplog): + cmd = CmdVersion(parse_args(["version"])) + cmd.run() + for record in caplog.records: + assert "DVC version" in record.msg + assert "Python version" in record.msg + assert "Platform" in record.msg