diff --git a/sparsity/test/test_sparse_frame.py b/sparsity/test/test_sparse_frame.py index ac377b3..81a2050 100644 --- a/sparsity/test/test_sparse_frame.py +++ b/sparsity/test/test_sparse_frame.py @@ -604,6 +604,16 @@ def test_getitem(): pdt.assert_index_equal(idx, sf[idx.values].columns) +def test_getitem_empty(): + df = pd.DataFrame([], columns=list('abcdefghij'), dtype=float) + sf = SparseFrame(df) + + assert sf['a'].empty + assert sf['a'].columns.tolist() == ['a'] + assert sf[['a', 'b']].empty + assert sf[['a', 'b']].columns.tolist() == ['a', 'b'] + + def test_vstack(): frames = [] data = []