Merge pull request #18 from Dr-Irv/issue17

Index.difference() accept None

Author: Dr-Irv

tests/pandas/test_indexes.py

@@ -48,14 +48,19 @@ def test_column_getitem() -> None:
 
     column = df.columns[0]
     a = df[column]
-    
+
+
 def test_column_contains() -> None:
     # https://github.com/microsoft/python-type-stubs/issues/199
-    df = pd.DataFrame({'A': [1, 2], 'B': ['c', 'd'], 'E': [3, 4]})
+    df = pd.DataFrame({"A": [1, 2], "B": ["c", "d"], "E": [3, 4]})
 
     collist = [column for column in df.columns]
- 
-    collist2 = [column for column in df.columns[df.columns.str.contains('A|B')]]
 
-    length = len(df.columns[df.columns.str.contains('A|B')])
-    
+    collist2 = [column for column in df.columns[df.columns.str.contains("A|B")]]
+
+    length = len(df.columns[df.columns.str.contains("A|B")])
+
+
+def test_difference_none() -> None:
+    ind = pd.Index([1, 2, 3])
+    id = ind.difference([1, None])

typings/pandas/core/indexes/base.pyi

@@ -149,7 +149,7 @@ class Index(IndexOpsMixin, PandasObject):
     def intersection(
         self, other: Union[List[T1], Index], sort: bool = ...
     ) -> Index: ...
-    def difference(self, other: Union[List[T1], Index]) -> Index: ...
+    def difference(self, other: Union[List[Union[T1, None]], Index]) -> Index: ...
     def symmetric_difference(
         self, other: Union[List[T1], Index], result_name=..., sort=...
     ) -> Index: ...
@@ -174,11 +174,11 @@ class Index(IndexOpsMixin, PandasObject):
     def __hash__(self) -> int: ...
     def __setitem__(self, key, value) -> None: ...
     @overload
-    def __getitem__(self, idx: Union[slice, np_ndarray_int64, Index, Series[bool]]) -> Index: ...
-    @overload
     def __getitem__(
-        self, idx: Union[int, Tuple[np_ndarray_int64, ...]]
-    ) -> Scalar: ...
+        self, idx: Union[slice, np_ndarray_int64, Index, Series[bool]]
+    ) -> Index: ...
+    @overload
+    def __getitem__(self, idx: Union[int, Tuple[np_ndarray_int64, ...]]) -> Scalar: ...
     def append(self, other): ...
     def putmask(self, mask, value): ...
     def equals(self, other) -> bool: ...