TST: Add some for the python version replacer

HaoZeke · Nov 24, 2024 · b844124 · b844124
1 parent 202e347
commit b844124
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/test/test_util.py b/test/test_util.py
@@ -551,3 +551,14 @@ def test_construct_pip_call(declaration, expected_result):
     parsed_declaration = util.ParsedPipDeclaration(declaration)
     result = util.construct_pip_call(pip_caller, parsed_declaration)
     assert result() == expected_result
+
+
+@pytest.mark.parametrize("arg, new_version, expected", [
+    ("python", "3.10", "python=3.10"),
+    ("python==2.7", "3.10", "python=3.10"),
+    ("python>=3.8", "3.10", "python=3.10"),
+    ("python==3.7", "3.10", "python=3.10"),
+    ("python_package", "3.10", "python_package"),
+])
+def test_replace_python_version(arg, new_version, expected):
+    assert util.replace_python_version(arg, new_version) == expected