adding pd.NaT to tested null-objects

pandas-dev · Nov 9, 2021 · 792cb7b · 792cb7b
1 parent 088b494
commit 792cb7b
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pandas/tests/libs/test_hashtable.py b/pandas/tests/libs/test_hashtable.py
@@ -454,9 +454,10 @@ def test_mode_stable(self, dtype, writable):
 
 def test_modes_with_nans():
     # GH42688, nans aren't mangled
-    values = np.array([True, pd.NA, np.nan, pd.NA, np.nan], dtype=np.object_)
+    nulls = [pd.NA, np.nan, pd.NaT, None]
+    values = np.array([True] + nulls * 2, dtype=np.object_)
     modes = ht.mode(values, False)
-    assert modes.size == 2
+    assert modes.size == len(nulls)
 
 
 def test_unique_label_indices_intp(writable):