Skip to content

Commit ddb3547

Browse files
committed
add test for issue pandas-dev#5006
1 parent e7b453b commit ddb3547

File tree

1 file changed

+12
-1
lines changed

1 file changed

+12
-1
lines changed

pandas/tests/test_indexing.py

+12-1
Original file line numberDiff line numberDiff line change
@@ -15,7 +15,8 @@
1515
from pandas.core.api import (DataFrame, Index, Series, Panel, notnull, isnull,
1616
MultiIndex, DatetimeIndex, Float64Index, Timestamp)
1717
from pandas.util.testing import (assert_almost_equal, assert_series_equal,
18-
assert_frame_equal, assert_panel_equal)
18+
assert_frame_equal, assert_panel_equal,
19+
assert_isinstance)
1920
from pandas import compat, concat
2021

2122
import pandas.util.testing as tm
@@ -1841,6 +1842,16 @@ def check_slicing_positional(index):
18411842
#self.assertRaises(TypeError, lambda : s.iloc[2.0:5.0])
18421843
#self.assertRaises(TypeError, lambda : s.iloc[2:5.0])
18431844

1845+
def test_array_indexing(self):
1846+
"""test that array indexing returns a sequence by calling len()"""
1847+
column = Series(np.arange(10))
1848+
indices = np.arange(5, 10)
1849+
assert_isinstance(column.iloc[indices], Series)
1850+
indices = np.array([5], dtype = int)
1851+
assert_isinstance(column.iloc[indices], Series)
1852+
indices = np.array([], dtype = int)
1853+
assert_isinstance(column.iloc[indices], Series)
1854+
18441855

18451856
if __name__ == '__main__':
18461857
import nose

0 commit comments

Comments
 (0)