Skip to content

Commit 61ed3d6

Browse files
committed
test: adds more tests for indexing
1 parent 9b00436 commit 61ed3d6

1 file changed

Lines changed: 6 additions & 0 deletions

File tree

test/test_indexing.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -27,3 +27,9 @@ def test_loc(stock):
2727
assert isinstance(stock, StockDataFrame)
2828

2929
assert stock.iloc[0].name == 'b'
30+
31+
32+
def test_indexing_with_column_list(stock):
33+
stock = stock[['close', 'open']]
34+
assert isinstance(stock, StockDataFrame)
35+
assert list(stock.columns) == ['close', 'open']

0 commit comments

Comments
 (0)