Skip to content

Commit dcb680e

Browse files
committed
TST: Fix sparse quantile test
1 parent 51b26d0 commit dcb680e

File tree

1 file changed

+12
-6
lines changed

1 file changed

+12
-6
lines changed

pandas/tests/sparse/test_frame.py

Lines changed: 12 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1705,25 +1705,31 @@ def test_where_with_bool_data_and_complex_other(self):
17051705
tm.assert_sp_frame_equal(result, sparse_expected)
17061706

17071707
def test_quantile(self):
1708-
data = [[1, 1], [2, 10], [3, 100], [4, 100]]
1708+
# GH 17386
1709+
data = [[1, 1], [2, 10], [3, 100], [nan, nan]]
17091710
q = 0.1
17101711

17111712
sparse_df = SparseDataFrame(data)
17121713
result = sparse_df.quantile(q)
17131714

17141715
dense_df = DataFrame(data)
1715-
expected = dense_df.quantile(q)
1716+
dense_expected = dense_df.quantile(q)
1717+
sparse_expected = SparseSeries(dense_expected)
17161718

1717-
tm.assert_series_equal(result, expected)
1719+
tm.assert_series_equal(result, dense_expected)
1720+
tm.assert_sp_series_equal(result, sparse_expected)
17181721

17191722
def test_quantile_multi(self):
1720-
data = [[1, 1], [2, 10], [3, 100], [4, 100]]
1723+
# GH 17386
1724+
data = [[1, 1], [2, 10], [3, 100], [nan, nan]]
17211725
q = [0.1, 0.5]
17221726

17231727
sparse_df = SparseDataFrame(data)
17241728
result = sparse_df.quantile(q)
17251729

17261730
dense_df = DataFrame(data)
1727-
expected = dense_df.quantile(q)
1731+
dense_expected = dense_df.quantile(q)
1732+
sparse_expected = SparseDataFrame(dense_expected)
17281733

1729-
tm.assert_frame_equal(result, expected)
1734+
tm.assert_frame_equal(result, dense_expected)
1735+
tm.assert_sp_frame_equal(result, sparse_expected)

0 commit comments

Comments
 (0)