From f08ce13a69665b20c5576eb8843a1a118ef62cf6 Mon Sep 17 00:00:00 2001 From: Anna Schumaker Date: Fri, 23 Jul 2021 10:29:00 -0400 Subject: [PATCH] lib: Create a file for getting the current version Signed-off-by: Anna Schumaker --- debug | 1 + lib/test_version.py | 13 +++++++++++++ lib/version.py | 16 ++++++++++++++++ 3 files changed, 30 insertions(+) create mode 100644 debug create mode 100644 lib/test_version.py create mode 100644 lib/version.py diff --git a/debug b/debug new file mode 100644 index 0000000..0e5d74d --- /dev/null +++ b/debug @@ -0,0 +1 @@ +emmental diff --git a/lib/test_version.py b/lib/test_version.py new file mode 100644 index 0000000..8cd4571 --- /dev/null +++ b/lib/test_version.py @@ -0,0 +1,13 @@ +# Copyright 2021 (c) Anna Schumaker. +from . import version +import unittest + +class TestVersion(unittest.TestCase): + def test_version(self): + self.assertEqual(version.MAJOR, 2) + self.assertEqual(version.MINOR, 0) + + self.assertTrue(version.DEBUG) + self.assertTrue(version.TESTING) + + self.assertEqual(version.string(), "Emmental 2.0-debug") diff --git a/lib/version.py b/lib/version.py new file mode 100644 index 0000000..1b9ebf6 --- /dev/null +++ b/lib/version.py @@ -0,0 +1,16 @@ +# Copyright 2021 (c) Anna Schumaker. +import os + +MAJOR = 2 +MINOR = 0 + +DEBUG = False +with open("debug") as f: + if f.read().strip() == "emmental": + DEBUG = True + +TESTING = os.environ.get("EMMENTAL_TESTING") != None + + +def string(): + return f"Emmental {MAJOR}.{MINOR}{'-debug' if DEBUG else ''}"