added datetime as a replacement case

mliu08 · Nov 27, 2022 · d616d60 · d616d60
1 parent c243b44
commit d616d60
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pandas/tests/frame/indexing/test_where.py b/pandas/tests/frame/indexing/test_where.py
@@ -18,7 +18,7 @@
     isna,
 )
 import pandas._testing as tm
-from pandas._testing._hypothesis import OPTIONAL_ONE_OF_ALL
+from pandas._testing._hypothesis import OPTIONAL_ONE_OF_ALL, DATETIME_JAN_1_1900_OPTIONAL_TZ
 
 
 @pytest.fixture(params=["default", "float_string", "mixed_float", "mixed_int"])
@@ -1019,7 +1019,7 @@ def test_where_producing_ea_cond_for_np_dtype():
     tm.assert_frame_equal(result, expected)
 
 
-@pytest.mark.parametrize("replacement", [0.001, True, "snake"])
+@pytest.mark.parametrize("replacement", [0.001, True, "snake", DATETIME_JAN_1_1900_OPTIONAL_TZ])
 def test_where_int_overflow(replacement):
     # GH 31687
     df = DataFrame([[1.0, 2e19, "nine"],[np.nan, 0.1, None]])