Skip to content

Commit f26d3d5

Browse files
TST: GH28813 test .diff() on Sparse dtype (#34268)
1 parent 8a4572b commit f26d3d5

File tree

1 file changed

+11
-0
lines changed

1 file changed

+11
-0
lines changed

pandas/tests/frame/methods/test_diff.py

Lines changed: 11 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -158,3 +158,14 @@ def test_diff_axis1_mixed_dtypes_negative_periods(self):
158158

159159
result = df.diff(axis=1, periods=-1)
160160
tm.assert_frame_equal(result, expected)
161+
162+
def test_diff_sparse(self):
163+
# GH#28813 .diff() should work for sparse dataframes as well
164+
sparse_df = pd.DataFrame([[0, 1], [1, 0]], dtype="Sparse[int]")
165+
166+
result = sparse_df.diff()
167+
expected = pd.DataFrame(
168+
[[np.nan, np.nan], [1.0, -1.0]], dtype=pd.SparseDtype("float", 0.0)
169+
)
170+
171+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)