TST: differentiate 0 and np.nan (pandas-dev#35498)

alexhlim · Aug 21, 2020 · 9338d3b · 9338d3b
1 parent 4f9d9df
commit 9338d3b
Showing 1 changed file with 15 additions and 0 deletions.
diff --git a/pandas/tests/indexes/test_indexing.py b/pandas/tests/indexes/test_indexing.py
@@ -120,6 +120,21 @@ def test_getitem_deprecated_float(idx):
             [NaT],
             np.array([1, 2], dtype=np.int64),
         ),
+        (
+            np.array([np.nan, 0, np.nan], dtype=object),
+            [np.nan],
+            np.array([0, 2], dtype=np.int64),
+        ),
+        (
+            np.array([np.nan, 0, np.nan], dtype=object),
+            [0],
+            np.array([1], dtype=np.int64),
+        ),
+        (
+            np.array([np.nan, 0, np.nan], dtype=object),
+            [np.nan, 0],
+            np.array([0, 2, 1], dtype=np.int64),
+        ),
     ],
 )
 def test_get_indexer_non_unique_multiple_nans(idx, target, expected):