|
15 | 15 | from pandas.core.api import (DataFrame, Index, Series, Panel, notnull, isnull,
|
16 | 16 | MultiIndex, DatetimeIndex, Float64Index, Timestamp)
|
17 | 17 | 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) |
19 | 20 | from pandas import compat, concat
|
20 | 21 |
|
21 | 22 | import pandas.util.testing as tm
|
@@ -1841,6 +1842,16 @@ def check_slicing_positional(index):
|
1841 | 1842 | #self.assertRaises(TypeError, lambda : s.iloc[2.0:5.0])
|
1842 | 1843 | #self.assertRaises(TypeError, lambda : s.iloc[2:5.0])
|
1843 | 1844 |
|
| 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 | + |
1844 | 1855 |
|
1845 | 1856 | if __name__ == '__main__':
|
1846 | 1857 | import nose
|
|
0 commit comments