1 parent 9b00436 commit 61ed3d6Copy full SHA for 61ed3d6
1 file changed
test/test_indexing.py
@@ -27,3 +27,9 @@ def test_loc(stock):
27
assert isinstance(stock, StockDataFrame)
28
29
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