Skip to content

Commit 785bbcf

Browse files
committed
TST: SparseDataFrame quantile test
1 parent 9121f04 commit 785bbcf

File tree

1 file changed

+24
-0
lines changed

1 file changed

+24
-0
lines changed

pandas/tests/sparse/test_frame.py

+24
Original file line numberDiff line numberDiff line change
@@ -1420,3 +1420,27 @@ def test_where_with_float_other(self):
14201420
expected = dense_df.where(dense_df >= 2, other)
14211421

14221422
tm.assert_frame_equal(result, expected)
1423+
1424+
def test_quantile(self):
1425+
data = [[1, 1], [2, 10], [3, 100], [4, 100]]
1426+
q = 0.1
1427+
1428+
sparse_df = SparseDataFrame(data)
1429+
result = sparse_df.quantile(q)
1430+
1431+
dense_df = DataFrame(data)
1432+
expected = dense_df.quantile(q)
1433+
1434+
tm.assert_series_equal(result, expected)
1435+
1436+
def test_quantile_multi(self):
1437+
data = [[1, 1], [2, 10], [3, 100], [4, 100]]
1438+
q = [0.1, 0.5]
1439+
1440+
sparse_df = SparseDataFrame(data)
1441+
result = sparse_df.quantile(q)
1442+
1443+
dense_df = DataFrame(data)
1444+
expected = dense_df.quantile(q)
1445+
1446+
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)