py39, py310 lint

py-polars/tests/unit/datatypes/test_enum.py

@@ -20,6 +20,13 @@
 )
 from polars.testing import assert_frame_equal, assert_series_equal
 
+try:
+    from enum import StrEnum
+
+    PyStrEnum: type[enum.Enum] | None = StrEnum
+except ImportError:
+    PyStrEnum = None
+
 
 def test_enum_creation() -> None:
     dtype = pl.Enum(["a", "b"])
@@ -617,14 +624,9 @@ def test_enum_19269() -> None:
 @pytest.mark.parametrize(
     "EnumBase",
     [
-        (enum.Enum,),
-        (enum.StrEnum,),
-        (str, enum.Enum),
-    ]
-    if sys.version_info >= (3, 11)
-    else [
         (enum.Enum,),
         (str, enum.Enum),
+        *([(PyStrEnum,)] if PyStrEnum is not None else []),
     ],
 )
 def test_init_frame_from_enums(EnumBase: tuple[type, ...]) -> None:

py-polars/tests/unit/functions/test_lit.py

@@ -1,7 +1,6 @@
 from __future__ import annotations
 
 import enum
-import sys
 from datetime import datetime, timedelta
 from decimal import Decimal
 from typing import TYPE_CHECKING, Any
@@ -19,6 +18,14 @@
     from polars._typing import PolarsDataType
 
 
+try:
+    from enum import StrEnum
+
+    PyStrEnum: type[enum.Enum] | None = StrEnum
+except ImportError:
+    PyStrEnum = None
+
+
 @pytest.mark.parametrize(
     "input",
     [
@@ -109,14 +116,9 @@ def test_lit_unsupported_type() -> None:
 @pytest.mark.parametrize(
     "EnumBase",
     [
-        (enum.Enum,),
-        (enum.StrEnum,),
-        (str, enum.Enum),
-    ]
-    if sys.version_info >= (3, 11)
-    else [
         (enum.Enum,),
         (str, enum.Enum),
+        *([(PyStrEnum,)] if PyStrEnum is not None else []),
     ],
 )
 def test_lit_enum_input_16668(EnumBase: tuple[type, ...]) -> None: