Skip tests if current Python version is too old

Author: pslacerda1

testing/tests/api/test_commanding.py

@@ -1,7 +1,9 @@
 from pytest import mark
 from pymol import cmd
+import sys
 from typing import List, Union, Any, Tuple
 from pathlib import Path
+from pymol import test_utils
 
 
 def test_docstring():
@@ -91,9 +93,12 @@ def func10(a: str="sele"):
     out, err = capsys.readouterr()
     assert out + err == ''
 
+@mark.skipif(
+    sys.version_info < (3, 11),
+    reason="Requires StrEnum of Python 3.11+"
+)
 def test_str_enum(capsys):
-    # TODO Fix imports requires
-    from strenum import StrEnum
+    from enum import StrEnum
     class E(StrEnum):
         A = "a"
     @cmd.new_command

testing/tests/helpers/test_utils.py

@@ -1,3 +1,4 @@
+import sys
 import tempfile
 from PIL import Image
 import numpy
@@ -131,3 +132,11 @@ def decorator(test_func):
             reason=reason or f"Requires PyMOL {version}"
         )(test_func)
     return decorator
+
+def requires_python(python_version, reason=None):
+    def decorator(test_func):
+        return pytest.mark.skipif(
+            sys.version_info < python_version,
+            reason=reason
+        )(test_func)
+    return decorator