Skip to content

Commit e96054a

Browse files
TST: Add regression test for NaN sparse columns (#52772)
* Add test for GH 27781 * Modify test for GH 27781
1 parent fc4ff26 commit e96054a

File tree

1 file changed

+12
-0
lines changed

1 file changed

+12
-0
lines changed

pandas/tests/arrays/sparse/test_array.py

Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -478,3 +478,15 @@ def test_drop_duplicates_fill_value():
478478
result = df.drop_duplicates()
479479
expected = pd.DataFrame({i: SparseArray([0.0], fill_value=0) for i in range(5)})
480480
tm.assert_frame_equal(result, expected)
481+
482+
483+
def test_zero_sparse_column():
484+
# GH 27781
485+
df1 = pd.DataFrame({"A": SparseArray([0, 0, 0]), "B": [1, 2, 3]})
486+
df2 = pd.DataFrame({"A": SparseArray([0, 1, 0]), "B": [1, 2, 3]})
487+
result = df1.loc[df1["B"] != 2]
488+
expected = df2.loc[df2["B"] != 2]
489+
tm.assert_frame_equal(result, expected)
490+
491+
expected = pd.DataFrame({"A": SparseArray([0, 0]), "B": [1, 3]}, index=[0, 2])
492+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)