Fix tests

dfint · Jul 13, 2024 · 464f5e6 · 464f5e6
1 parent 72d8b60
commit 464f5e6
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 9 deletions.
diff --git a/alternative_encodings/common.py b/alternative_encodings/common.py
@@ -1,11 +1,11 @@
 import codecs
-from typing import TYPE_CHECKING
+from typing import TYPE_CHECKING, Union
 
 if TYPE_CHECKING:
-    from _codecs import _EncodingMap  # noqa: TCH004
+    from _codecs import _EncodingMap
 
 
-def get_codec(encoding_table: dict[int, int] | _EncodingMap, decoding_table: str) -> type[codecs.Codec]:
+def get_codec(encoding_table: Union[dict[int, int], "_EncodingMap"], decoding_table: str) -> type[codecs.Codec]:
     class Codec(codecs.Codec):
         def encode(self, input_string: str, errors: str = "strict") -> tuple[bytes, int]:
             return codecs.charmap_encode(input_string, errors, encoding_table)
@@ -16,7 +16,7 @@ def decode(self, input_bytes: bytes, errors: str = "strict") -> tuple[str, int]:
     return Codec
 
 
-def get_incremental_encoder(encoding_table: dict[int, int] | _EncodingMap) -> type[codecs.IncrementalEncoder]:
+def get_incremental_encoder(encoding_table: Union[dict[int, int], "_EncodingMap"]) -> type[codecs.IncrementalEncoder]:
     class IncrementalEncoder(codecs.IncrementalEncoder):
         def encode(self, input_string: str, _final: bool = False) -> bytes:  # noqa: FBT001, FBT002
             return codecs.charmap_encode(input_string, self.errors, encoding_table)[0]

diff --git a/tests/test_cp859.py b/tests/test_cp859.py
@@ -2,10 +2,11 @@
 from collections.abc import Generator
 
 import pytest
-from utils import register_codec
 
 from alternative_encodings import cp859
 
+from .utils import register_codec
+
 
 @pytest.fixture(scope="module", autouse=True)
 def _register_codec_fixture() -> Generator[None, None, None]:

diff --git a/tests/test_cp866i.py b/tests/test_cp866i.py
@@ -2,10 +2,11 @@
 from collections.abc import Generator
 
 import pytest
-from utils import register_codec
 
 from alternative_encodings import cp866i
 
+from .utils import register_codec
+
 
 @pytest.fixture(scope="module", autouse=True)
 def _register_codec_fixture() -> Generator[None, None, None]:

diff --git a/tests/test_general.py b/tests/test_general.py
@@ -1,10 +1,11 @@
 import codecs
 
 import pytest
-from utils import register_codec
 
 from alternative_encodings import cp859, cp866i, viscii
 
+from .utils import register_codec
+
 codecs_pairs = [
     ("cp859", cp859),
     ("cp866i", cp866i),

diff --git a/tests/test_romaji.py b/tests/test_romaji.py
@@ -2,10 +2,11 @@
 from collections.abc import Generator
 
 import pytest
-from utils import register_codec
 
 from alternative_encodings import romaji
 
+from .utils import register_codec
+
 
 @pytest.fixture(scope="module", autouse=True)
 def _register_codec_fixture() -> Generator[None, None, None]:

diff --git a/tests/test_viscii.py b/tests/test_viscii.py
@@ -2,10 +2,11 @@
 from collections.abc import Generator
 
 import pytest
-from utils import register_codec
 
 from alternative_encodings import viscii
 
+from .utils import register_codec
+
 
 @pytest.fixture(scope="module", autouse=True)
 def _register_codec_fixture() -> Generator[None, None, None]: