@@ -1705,25 +1705,31 @@ def test_where_with_bool_data_and_complex_other(self):
1705
1705
tm .assert_sp_frame_equal (result , sparse_expected )
1706
1706
1707
1707
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 ]]
1709
1710
q = 0.1
1710
1711
1711
1712
sparse_df = SparseDataFrame (data )
1712
1713
result = sparse_df .quantile (q )
1713
1714
1714
1715
dense_df = DataFrame (data )
1715
- expected = dense_df .quantile (q )
1716
+ dense_expected = dense_df .quantile (q )
1717
+ sparse_expected = SparseSeries (dense_expected )
1716
1718
1717
- tm .assert_series_equal (result , expected )
1719
+ tm .assert_series_equal (result , dense_expected )
1720
+ tm .assert_sp_series_equal (result , sparse_expected )
1718
1721
1719
1722
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 ]]
1721
1725
q = [0.1 , 0.5 ]
1722
1726
1723
1727
sparse_df = SparseDataFrame (data )
1724
1728
result = sparse_df .quantile (q )
1725
1729
1726
1730
dense_df = DataFrame (data )
1727
- expected = dense_df .quantile (q )
1731
+ dense_expected = dense_df .quantile (q )
1732
+ sparse_expected = SparseDataFrame (dense_expected )
1728
1733
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