Add a `--version` option

1 files changed, 10 insertions, 2 deletions

diff --git a/tests/test_climat2.py b/tests/test_climat2.py
index 44703b7..67b56eb 100644
--- a/tests/test_climat2.py
+++ b/tests/test_climat2.py
@@ -8,12 +8,20 @@ class TestHelp(unittest.TestCase):
     def test_help(self):
         proc = subprocess.Popen(['./main.py', '--help'], stdout=subprocess.PIPE)
         stdout, _ = proc.communicate()
-        self.assertIn(b'usage: main.py [-h] [-c] [-l] [-s] [-L] [files [files ...]]', stdout)
+        self.assertIn(b'usage: main.py [-h] [-v] [-c] [-l] [-s] [-L] [files [files ...]]', stdout)
 
     def test_no_arg(self):
         proc = subprocess.Popen(['./main.py'], stdout=subprocess.PIPE)
         stdout, _ = proc.communicate()
-        self.assertIn(b'usage: main.py [-h] [-c] [-l] [-s] [-L] [files [files ...]]', stdout)
+        self.assertIn(b'usage: main.py [-h] [-v] [-c] [-l] [-s] [-L] [files [files ...]]', stdout)
+
+
+class TestVersion(unittest.TestCase):
+    def test_version(self):
+        proc = subprocess.Popen(['./main.py', '--version'], stdout=subprocess.PIPE)
+        stdout, _ = proc.communicate()
+        self.assertTrue(stdout.startswith(b'MAT2 '))
+
 
 
 class TestReturnValue(unittest.TestCase):