Skip to content

Commit 1d85ad8

Browse files
committed
re-fix case_when test
1 parent 09d27aa commit 1d85ad8

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

pandas/tests/series/methods/test_case_when.py

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -2,7 +2,6 @@
22
import pytest
33

44
from pandas import (
5-
NA,
65
DataFrame,
76
Series,
87
array as pd_array,
@@ -100,7 +99,7 @@ def test_case_when_multiple_conditions_replacement_extension_dtype(df):
10099
(df["a"].gt(1) & df["b"].eq(5), pd_array([1, 2, 3], dtype="Int64")),
101100
],
102101
)
103-
expected = Series([1, 2, NA], dtype="Float64")
102+
expected = Series([1, 2, np.nan], dtype="Float64")
104103
tm.assert_series_equal(result, expected)
105104

106105

0 commit comments

Comments
 (0)