Fix os.PathLike not compatible with pandas._typing.FilePathOrBuffer (#41)

* Fix os.PathLike not compatible with pandas._typing.FilePathOrBuffer

* Refactoring test case

* optimize imports

Co-authored-by: joannasendorek <joanna.sendorek@tesco.com>

Author: joannasendorek

setup.py

@@ -21,7 +21,7 @@ def list_packages(source_path: str = src_path) -> None:
 setup(
     name="pandas-stubs",
     package_dir={"": src_path},
-    version="1.1.0.5",
+    version="1.1.0.6",
     description="Type annotations for Pandas",
     long_description=(open("README.md").read()
                       if os.path.exists("README.md") else ""),

tests/snippets/test_frame.py

@@ -1,5 +1,4 @@
 # flake8: noqa: F841
-
 import tempfile
 from pathlib import Path
 
@@ -14,7 +13,7 @@ def test_types_init() -> None:
     pd.DataFrame(np.array([[1, 2, 3], [4, 5, 6], [7, 8, 9]]), columns=['a', 'b', 'c'], dtype=np.int8, copy=True)
 
 
-def test_types_csv() -> None:
+def test_types_to_csv() -> None:
     df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]})
     csv_df: str = df.to_csv()
 
@@ -32,6 +31,17 @@ def test_types_csv() -> None:
         df4: pd.DataFrame = pd.read_csv(file.name)
 
 
+def test_types_to_csv_when_path_passed() -> None:
+    df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4]})
+    path: Path = Path("./dummy_path.txt")
+    try:
+        assert not path.exists()
+        df.to_csv(path)
+        df5: pd.DataFrame = pd.read_csv(path)
+    finally:
+        path.unlink()
+
+
 def test_types_getitem() -> None:
     df = pd.DataFrame(data={'col1': [1, 2], 'col2': [3, 4], 5: [6, 7]})
     i = pd.Index(['col1', 'col2'])

third_party/3/pandas/_typing.pyi

@@ -4,8 +4,10 @@ import datetime
 import sys
 from decimal import Decimal
 from fractions import Fraction
+from io import RawIOBase, BufferedIOBase, TextIOBase, TextIOWrapper
+from mmap import mmap
+from os import PathLike
 from numbers import Number
-from pathlib import Path
 
 from pandas.core.arrays.base import ExtensionArray as ExtensionArray
 from pandas.core.indexes.base import Index as Index
@@ -65,7 +67,12 @@ UserCorrelationMethod = Callable[[np.ndarray, np.ndarray], Scalar]
 CorrelationMethod = Union[NamedCorrelationMethod, UserCorrelationMethod]
 
 Dtype: Any
-FilePathOrBuffer = Union[str, Path, IO[AnyStr]]
+T = TypeVar('T')
+
+Buffer = Union[IO[AnyStr], RawIOBase, BufferedIOBase, TextIOBase, TextIOWrapper, mmap]
+FileOrBuffer = Union[str, Buffer[T]]
+FilePathOrBuffer = Union["PathLike[str]", FileOrBuffer[T]]
+
 FrameOrSeriesUnion: Union[DataFrame, Series]
 FrameOrSeries = Union[DataFrame, Series]
 Axis = Union[str, int]
@@ -76,8 +83,6 @@ JSONSerializable = Union[PythonScalar, List, Dict]
 Axes = Collection
 Renamer = Union[Mapping[Label, Any], Callable[[Label], Label]]
 
-T = TypeVar('T')
-
 FillValue = Union[Scalar, Dict[Any, Any], FrameOrSeries]
 
 # Any plain Python or numpy function