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)
