@@ -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