def test_subclass_sparse_slice(self): rows = [[1.0, 2.0, 3.0], [4.0, 5.0, 6.0], [7.0, 8.0, 9.0]] ssdf = tm.SubclassedSparseDataFrame(rows) ssdf.testattr = "testattr" tm.assert_sp_frame_equal(ssdf.loc[:2], tm.SubclassedSparseDataFrame(rows[:3])) tm.assert_sp_frame_equal(ssdf.iloc[:2], tm.SubclassedSparseDataFrame(rows[:2])) tm.assert_sp_frame_equal(ssdf[:2], tm.SubclassedSparseDataFrame(rows[:2])) assert ssdf.loc[:2].testattr == "testattr" assert ssdf.iloc[:2].testattr == "testattr" assert ssdf[:2].testattr == "testattr" tm.assert_sp_series_equal( ssdf.loc[1], tm.SubclassedSparseSeries(rows[1]), check_names=False, check_kind=False, ) tm.assert_sp_series_equal( ssdf.iloc[1], tm.SubclassedSparseSeries(rows[1]), check_names=False, check_kind=False, )
def test_subclass_sparse_to_frame(self): s = tm.SubclassedSparseSeries([1, 2], index=list('abcd'), name='xxx') res = s.to_frame() exp_arr = pd.SparseArray([1, 2], dtype=np.int64, kind='block', fill_value=0) exp = tm.SubclassedSparseDataFrame({'xxx': exp_arr}, index=list('abcd'), default_fill_value=0) tm.assert_sp_frame_equal(res, exp) # create from int dict res = tm.SubclassedSparseDataFrame({'xxx': [1, 2]}, index=list('abcd'), default_fill_value=0) tm.assert_sp_frame_equal(res, exp) s = tm.SubclassedSparseSeries([1.1, 2.1], index=list('abcd'), name='xxx') res = s.to_frame() exp = tm.SubclassedSparseDataFrame({'xxx': [1.1, 2.1]}, index=list('abcd')) tm.assert_sp_frame_equal(res, exp)
def test_subclass_sparse_transpose(self): ossdf = tm.SubclassedSparseDataFrame([[1, 2, 3], [4, 5, 6]]) essdf = tm.SubclassedSparseDataFrame([[1, 4], [2, 5], [3, 6]]) tm.assert_sp_frame_equal(ossdf.T, essdf)