some testing

test/test_parse.py

@@ -0,0 +1,181 @@
+import re
+from pycalver import parse
+
+
+def test_readme_pycalver1():
+    version_str = "v201712.0001-alpha"
+    version_info = parse.PYCALVER_RE.match(version_str).groupdict()
+
+    assert version_info == {
+        "version" : "v201712.0001-alpha",
+        "calver"  : "v201712",
+        "year"    : "2017",
+        "month"   : "12",
+        "build"   : ".0001",
+        "release" : "-alpha",
+    }
+
+
+def test_readme_pycalver2():
+    version_str = "v201712.0033"
+    version_info = parse.PYCALVER_RE.match(version_str).groupdict()
+
+    assert version_info == {
+        "version" : "v201712.0033",
+        "calver"  : "v201712",
+        "year"    : "2017",
+        "month"   : "12",
+        "build"   : ".0033",
+        "release" : None,
+    }
+
+
+def test_re_pattern_parts():
+    part_re_by_name = {
+        part_name: re.compile(part_re_str)
+        for part_name, part_re_str in parse.RE_PATTERN_PARTS.items()
+    }
+
+    cases = [
+        ("pep440_version", "201712.31", "201712.31"),
+        ("pep440_version", "v201712.0032", None),
+        ("pep440_version", "201712.0033-alpha", None),
+        ("version", "v201712.0034", "v201712.0034"),
+        ("version", "v201712.0035-alpha", "v201712.0035-alpha"),
+        ("version", "v201712.0036-alpha0", "v201712.0036-alpha"),
+        ("version", "v201712.0037-pre", "v201712.0037"),
+        ("version", "201712.38a0", None),
+        ("version", "201712.0039", None),
+        ("calver", "v201712", "v201712"),
+        ("calver", "v201799", "v201799"),   # maybe date validation should be a thing
+        ("calver", "201712", None),
+        ("calver", "v20171", None),
+        ("build", ".0012", ".0012"),
+        ("build", ".11012", ".11012"),
+        ("build", ".012", None),
+        ("build", "11012", None),
+        ("release", "-alpha", "-alpha"),
+        ("release", "-beta", "-beta"),
+        ("release", "-dev", "-dev"),
+        ("release", "-rc", "-rc"),
+        ("release", "-post", "-post"),
+        ("release", "-pre", ""),
+        ("release", "alpha", ""),
+    ]
+
+    for part_name, line, expected in cases:
+        part_re = part_re_by_name[part_name]
+        result = part_re.search(line)
+        if result is None:
+            assert expected is None, (part_name, line)
+        else:
+            result_val = result.group(0)
+            assert result_val == expected, (part_name, line)
+
+
+def test_parse_version_info():
+    version_str = "v201712.0001-alpha"
+    version_nfo = parse.parse_version_info(version_str)
+
+    assert version_nfo.pep440_version == "201712.1a0"
+    assert version_nfo.version == "v201712.0001-alpha"
+    assert version_nfo.calver == "v201712"
+    assert version_nfo.year == "2017"
+    assert version_nfo.month == "12"
+    assert version_nfo.build == ".0001"
+    assert version_nfo.release == "-alpha"
+
+    version_str = "v201712.0001"
+    version_nfo = parse.parse_version_info(version_str)
+
+    assert version_nfo.pep440_version == "201712.1"
+    assert version_nfo.version == "v201712.0001"
+    assert version_nfo.calver == "v201712"
+    assert version_nfo.year == "2017"
+    assert version_nfo.month == "12"
+    assert version_nfo.build == ".0001"
+    assert version_nfo.release is None
+
+
+def test_default_parse_patterns():
+    lines = [
+        "# setup.py",
+        "import setuptools",
+        "__version__ = 'v201712.0002-alpha'",
+        "setuptools.setup(",
+        "...",
+        "   version='201712.2a0',",
+    ]
+
+    patterns = [
+        "{version}",
+        "{pep440_version}",
+    ]
+
+    matches = parse.parse_patterns(lines, patterns)
+    assert len(matches) == 2
+
+    assert matches[0].lineno == 2
+    assert matches[1].lineno == 5
+
+    assert matches[0].pattern == patterns[0]
+    assert matches[1].pattern == patterns[1]
+
+    assert matches[0].match == "v201712.0002-alpha"
+    assert matches[1].match == "201712.2a0"
+
+
+def test_explicit_parse_patterns():
+    lines = [
+        "# setup.py",
+        "import setuptools",
+        "__version__ = 'v201712.0002-alpha'",
+        "setuptools.setup(",
+        "...",
+        "   version='201712.2a0',",
+    ]
+
+    patterns = [
+        "__version__ = '{version}'",
+        "version='{pep440_version}'",
+    ]
+
+    matches = parse.parse_patterns(lines, patterns)
+    assert len(matches) == 2
+
+    assert matches[0].lineno == 2
+    assert matches[1].lineno == 5
+
+    assert matches[0].pattern == patterns[0]
+    assert matches[1].pattern == patterns[1]
+
+    assert matches[0].match == "__version__ = 'v201712.0002-alpha'"
+    assert matches[1].match == "version='201712.2a0'"
+
+
+def test_badge_parse_patterns():
+    lines = [
+        ":alt: PyPI version",
+        "",
+        ".. |version| image:: https://img.shields.io/badge/CalVer-v201809.0002--beta-blue.svg",
+        ":target: https://calver.org/",
+        ":alt: CalVer v201809.0002-beta",
+        "",
+    ]
+
+    patterns = [
+        "badge/CalVer-{calver}{build}-{release}-blue.svg",
+        ":alt: CalVer {version}",
+    ]
+
+    matches = parse.parse_patterns(lines, patterns)
+    assert len(matches) == 2
+
+    assert matches[0].lineno == 2
+    assert matches[1].lineno == 4
+
+    assert matches[0].pattern == patterns[0]
+    assert matches[1].pattern == patterns[1]
+
+    assert matches[0].match == "badge/CalVer-v201809.0002--beta-blue.svg"
+    assert matches[1].match == ":alt: CalVer v201809.0002-beta"