Skip to content

Commit 48f81cd

Browse files
committed
updated to better match original issue
1 parent af2dfbf commit 48f81cd

File tree

1 file changed

+4
-7
lines changed

1 file changed

+4
-7
lines changed

pandas/tests/frame/indexing/test_where.py

+4-7
Original file line numberDiff line numberDiff line change
@@ -18,10 +18,7 @@
1818
isna,
1919
)
2020
import pandas._testing as tm
21-
from pandas._testing._hypothesis import (
22-
DATETIME_JAN_1_1900_OPTIONAL_TZ,
23-
OPTIONAL_ONE_OF_ALL,
24-
)
21+
from pandas._testing._hypothesis import OPTIONAL_ONE_OF_ALL
2522

2623

2724
@pytest.fixture(params=["default", "float_string", "mixed_float", "mixed_int"])
@@ -1023,12 +1020,12 @@ def test_where_producing_ea_cond_for_np_dtype():
10231020

10241021

10251022
@pytest.mark.parametrize(
1026-
"replacement", [0.001, True, "snake", DATETIME_JAN_1_1900_OPTIONAL_TZ]
1023+
"replacement", [0.001, True, "snake", None, datetime(2022, 5, 4)]
10271024
)
10281025
def test_where_int_overflow(replacement):
10291026
# GH 31687
1030-
df = DataFrame([[1.0, 2e19, "nine"], [np.nan, 0.1, None]])
1027+
df = DataFrame([[1.0, 2e25, "nine"], [np.nan, 0.1, None]])
10311028
result = df.where(pd.notnull(df), replacement)
1032-
expected = DataFrame([[1.0, 2e19, "nine"], [replacement, 0.1, replacement]])
1029+
expected = DataFrame([[1.0, 2e25, "nine"], [replacement, 0.1, replacement]])
10331030

10341031
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)