Fixed pytest issues for tomllib

Author: laundry-96

config/__init__.py

@@ -11,16 +11,17 @@
 except ImportError:  # pragma: no cover
     yaml = None
 try:
-    import tomllib as toml
+    import tomllib as toml  # type: ignore [import-not-found,unused-ignore]
+
     toml_readtype = "rb"
 except ImportError:  # pragma: no cover
     try:
-        import toml
-        toml_readtype = "rt"
-    except ImportError:
-        toml = None
-        toml_readtype = None
+        import toml  # type: ignore [no-redef,unused-ignore]
 
+        toml_readtype = "rt"
+    except ImportError:  # pragma: no cover
+        toml = None  # type: ignore [assignment,unused-ignore]
+        toml_readtype = ""
 
 from .configuration import Configuration
 from .configuration_set import ConfigurationSet
@@ -870,7 +871,7 @@ def _reload(self, data: Union[str, TextIO], read_from_file: bool = False) -> Non
                 with open(data, toml_readtype) as f:
                     loaded = toml.load(f)
             else:
-                loaded = toml.load(data)
+                loaded = toml.load(data)  # type: ignore [arg-type,unused-ignore]
         else:
             data = cast(str, data)
             loaded = toml.loads(data)

tests/test_toml.py

@@ -1,9 +1,12 @@
+import tempfile
+
 import pytest
+
 from config import config_from_dict
-import tempfile
 
 try:
     import toml
+
     from config import config_from_toml
 except ImportError:
     toml = None
@@ -75,7 +78,16 @@ def test_load_toml_file():  # type: ignore
     with tempfile.NamedTemporaryFile() as f:
         f.file.write(TOML.encode())
         f.file.flush()
-        cfg = config_from_toml(open(f.name, "rt"), read_from_file=True)
+
+        # Two different toml libraries are in use
+        import sys
+
+        if sys.version_info[1] < 11:
+            read_type = "rt"
+        else:
+            read_type = "rb"
+
+        cfg = config_from_toml(open(f.name, read_type), read_from_file=True)
     assert cfg["a1.b1.c1"] == 1
     assert cfg["a1.b1"].as_dict() == {"c1": 1, "c2": 2, "c3": 3}
     assert cfg["a1.b2"].as_dict() == {"c1": "a", "c2": True, "c3": 1.1}
@@ -132,13 +144,15 @@ def test_reload_toml_with_section_prefix():  # type: ignore
         f.file.flush()
 
         cfg = config_from_toml(
-            f.name, section_prefix="tool.coverage.", read_from_file=True
+            f.name,
+            section_prefix="tool.coverage.",
+            read_from_file=True,
         )
         expected = config_from_dict(
             {
                 "run.branch": False,
                 "run.parallel": False,
-            }
+            },
         )
 
         assert cfg == expected
@@ -151,6 +165,6 @@ def test_reload_toml_with_section_prefix():  # type: ignore
         expected = config_from_dict(
             {
                 "report.ignore_errors": False,
-            }
+            },
         )
         assert cfg == expected