TYP: handle mypy ignore in pandas/_testing

pandas/_testing.py

@@ -10,7 +10,17 @@
 from shutil import rmtree
 import string
 import tempfile
-from typing import Any, Callable, ContextManager, List, Optional, Type, Union, cast
+from typing import (
+    Any,
+    Callable,
+    ContextManager,
+    List,
+    Optional,
+    Tuple,
+    Type,
+    Union,
+    cast,
+)
 import warnings
 import zipfile
 
@@ -299,39 +309,35 @@ def write_to_compressed(compression, path, data, dest="test"):
     ------
     ValueError : An invalid compression value was passed in.
     """
-    if compression == "zip":
-        compress_method = zipfile.ZipFile
-    elif compression == "gzip":
-        # pandas\_testing.py:288: error: Incompatible types in assignment
-        # (expression has type "Type[GzipFile]", variable has type
-        # "Type[ZipFile]")
-        compress_method = gzip.GzipFile  # type: ignore[assignment]
-    elif compression == "bz2":
-        # pandas\_testing.py:290: error: Incompatible types in assignment
-        # (expression has type "Type[BZ2File]", variable has type
-        # "Type[ZipFile]")
-        compress_method = bz2.BZ2File  # type: ignore[assignment]
-    elif compression == "xz":
-        compress_method = get_lzma_file(lzma)
-    else:
-        raise ValueError(f"Unrecognized compression type: {compression}")
+    compress_method = _select_compress_method(compression)
 
+    args: Tuple[Any, ...]
     if compression == "zip":
         mode = "w"
         args = (dest, data)
         method = "writestr"
     else:
         mode = "wb"
-        # pandas\_testing.py:302: error: Incompatible types in assignment
-        # (expression has type "Tuple[Any]", variable has type "Tuple[Any,
-        # Any]")
-        args = (data,)  # type: ignore[assignment]
+        args = (data,)
         method = "write"
 
     with compress_method(path, mode=mode) as f:
         getattr(f, method)(*args)
 
 
+def _select_compress_method(compression: str) -> Callable:
+    if compression == "zip":
+        return zipfile.ZipFile
+    elif compression == "gzip":
+        return gzip.GzipFile
+    elif compression == "bz2":
+        return bz2.BZ2File
+    elif compression == "xz":
+        return get_lzma_file(lzma)
+    else:
+        raise ValueError(f"Unrecognized compression type: {compression}")
+
+
 def _get_tol_from_less_precise(check_less_precise: Union[bool, int]) -> float:
     """
     Return the tolerance equivalent to the deprecated `check_less_precise`