tests for Spec.parse

Author: keewis

minimum_versions.py

@@ -73,7 +73,7 @@ def parse(cls, spec_text):
             version = Version(version_text)
             segments = version.segments()
 
-            if len(segments) != 2 or (len(segments) == 3 and segments[2] != 0):
+            if (len(segments) == 3 and segments[2] != [0]) or len(segments) > 3:
                 warnings.append(
                     f"package should be pinned to a minor version (got {version})"
                 )

test_script.py

@@ -0,0 +1,25 @@
+import pytest
+from rattler import Version
+
+from minimum_versions import Spec
+
+
+@pytest.mark.parametrize(
+    ["text", "expected_spec", "expected_name", "expected_warnings"],
+    (
+        ("numpy=1.23", Spec("numpy", Version("1.23")), "numpy", []),
+        ("xarray=2024.10.0", Spec("xarray", Version("2024.10.0")), "xarray", []),
+        (
+            "xarray=2024.10.1",
+            Spec("xarray", Version("2024.10.1")),
+            "xarray",
+            ["package should be pinned to a minor version (got 2024.10.1)"],
+        ),
+    ),
+)
+def test_spec_parse(text, expected_spec, expected_name, expected_warnings):
+    actual_spec, (actual_name, actual_warnings) = Spec.parse(text)
+
+    assert actual_spec == expected_spec
+    assert actual_name == expected_name
+    assert actual_warnings == expected_warnings