feat(pypi): improve resolving suitable python version (#1725)

1 files changed, 22 insertions, 0 deletions

diff --git a/tests/mason-core/pep440_spec.lua b/tests/mason-core/pep440_spec.lua
new file mode 100644
index 00000000..c8ff9880
--- /dev/null
+++ b/tests/mason-core/pep440_spec.lua
@@ -0,0 +1,22 @@
+local pep440 = require "mason-core.pep440"
+
+describe("pep440 version checking", function()
+    it("should check single version specifier", function()
+        assert.is_false(pep440.check_version("3.5.0", ">=3.6"))
+        assert.is_true(pep440.check_version("3.6.0", ">=3.6"))
+        assert.is_false(pep440.check_version("3.6.0", ">=3.6.1"))
+    end)
+
+    it("should check version specifier with lower and upper bound", function()
+        assert.is_true(pep440.check_version("3.8.0", ">=3.8,<3.12"))
+        assert.is_false(pep440.check_version("3.12.0", ">=3.8,<3.12"))
+        assert.is_true(pep440.check_version("3.12.0", ">=3.8,<4.0.0"))
+    end)
+
+    it("should check multiple specifiers with different constraints", function()
+        assert.is_false(pep440.check_version("3.5.0", "!=4.0,<=4.0,>=3.8"))
+        assert.is_false(pep440.check_version("4.0.0", "!=4.0,<=4.0,>=3.8"))
+        assert.is_true(pep440.check_version("3.8.1", "!=4.0,<=4.0,>=3.8"))
+        assert.is_true(pep440.check_version("3.12.0", "!=4.0,<=4.0,>=3.8"))
+    end)
+end)