From d616d60b257a609bfcdbc4d9d1b66959b6a25511 Mon Sep 17 00:00:00 2001 From: mliu08 Date: Mon, 14 Nov 2022 21:01:04 -0800 Subject: [PATCH] added datetime as a replacement case --- pandas/tests/frame/indexing/test_where.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/pandas/tests/frame/indexing/test_where.py b/pandas/tests/frame/indexing/test_where.py index 8c0efbd9872131..b3c087e8c99fc9 100644 --- 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]])