Skip to content

Commit 839fafb

Browse files
author
Mike Phung
committed
TST GH#40498 Add tests for fillna null value replacement.
1 parent de49e43 commit 839fafb

File tree

2 files changed

+31
-0
lines changed

2 files changed

+31
-0
lines changed

pandas/tests/frame/methods/test_fillna.py

Lines changed: 20 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -153,6 +153,26 @@ def test_fillna_tzaware_different_column(self):
153153
)
154154
tm.assert_frame_equal(result, expected)
155155

156+
def test_null_value_replacement(self):
157+
# GH$40498
158+
df = DataFrame(
159+
{
160+
"A": [1, 2, None, 4],
161+
"B": [2, None, 6, 8],
162+
}
163+
)
164+
value = np.nan
165+
result = df.fillna(value)
166+
167+
expected = DataFrame(
168+
{
169+
"A": [1, 2, np.nan, 4],
170+
"B": [2, np.nan, 6, 8],
171+
}
172+
)
173+
174+
tm.assert_frame_equal(result, expected)
175+
156176
def test_other_missing_values_not_modified(
157177
self, unique_nulls_fixture, unique_nulls_fixture2
158178
):

pandas/tests/series/methods/test_fillna.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -623,6 +623,17 @@ def test_fillna_numeric_inplace(self):
623623
expected = x.fillna(value=0)
624624
tm.assert_series_equal(y, expected)
625625

626+
def test_fillna_null_value_replacement(self):
627+
# GH#40498
628+
ser = Series([None])
629+
630+
value = np.nan
631+
result = ser.fillna(value)
632+
633+
expected = Series([np.nan])
634+
635+
tm.assert_series_equal(result, expected)
636+
626637
def test_fillna_other_missing_values_not_modified(
627638
self, unique_nulls_fixture, unique_nulls_fixture2
628639
):

0 commit comments

Comments
 (0)