Merge pull request #265 from curveresearch/version-test

Move version tests to proper test file
curveresearch · Sep 26, 2023 · f6b49bf · f6b49bf
2 parents ebbe91f + 6368840
commit f6b49bf
Show file tree

Hide file tree

Showing 2 changed files with 26 additions and 8 deletions.
diff --git a/curvesim/version.py b/curvesim/version.py
@@ -26,11 +26,3 @@ def parse_version(version):
 
 
 __version_info__ = parse_version(__version__)
-
-
-if __name__ == "__main__":
-    print(parse_version("0.3.5"))  # (0, 3, 5, 'release', 0)
-    print(parse_version("0.3.5.release"))  # (0, 3, 5, release, 0)
-    print(parse_version("0.3.5.rc3"))  # (0, 3, 5, rc, 3)
-    print(parse_version("0.3.5.rc0"))  # (0, 3, 5, rc, 0)
-    print(parse_version("0.3.5.a12"))  # (0, 3, 5, a, 12)
diff --git a/test/unit/test_version.py b/test/unit/test_version.py
@@ -0,0 +1,26 @@
+from curvesim.version import parse_version
+
+
+def test_parse_version():
+    """
+    Check the version string gets correctly parsed to 'info' tuple.
+    """
+    version_tuple = parse_version("0.3.5")
+    expected_result = (0, 3, 5, "release", 0)
+    assert version_tuple == expected_result
+
+    version_tuple = parse_version("0.3.5.release")
+    expected_result = (0, 3, 5, "release", 0)
+    assert version_tuple == expected_result
+
+    version_tuple = parse_version("0.3.5.rc3")
+    expected_result = (0, 3, 5, "rc", 3)
+    assert version_tuple == expected_result
+
+    version_tuple = parse_version("0.3.5.rc0")
+    expected_result = (0, 3, 5, "rc", 0)
+    assert version_tuple == expected_result
+
+    version_tuple = parse_version("0.3.5.a12")
+    expected_result = (0, 3, 5, "a", 12)
+    assert version_tuple == expected_result