|
6 | 6 | import operator
|
7 | 7 | import warnings
|
8 | 8 |
|
| 9 | +from pandas import _np_version_under1p8 |
9 | 10 | from pandas.sparse.api import SparseArray
|
10 | 11 | from pandas.util.testing import assert_almost_equal, assertRaisesRegexp
|
11 | 12 | import pandas.util.testing as tm
|
@@ -37,11 +38,15 @@ def test_get_item(self):
|
37 | 38 | def test_take(self):
|
38 | 39 | self.assertTrue(np.isnan(self.arr.take(0)))
|
39 | 40 | self.assertTrue(np.isscalar(self.arr.take(2)))
|
40 |
| - self.assertEqual(self.arr.take(2), np.take(self.arr_data, 2)) |
41 |
| - self.assertEqual(self.arr.take(6), np.take(self.arr_data, 6)) |
| 41 | + |
| 42 | + # np.take in < 1.8 doesn't support scalar indexing |
| 43 | + if not _np_version_under1p8: |
| 44 | + self.assertEqual(self.arr.take(2), np.take(self.arr_data, 2)) |
| 45 | + self.assertEqual(self.arr.take(6), np.take(self.arr_data, 6)) |
42 | 46 |
|
43 | 47 | tm.assert_sp_array_equal(self.arr.take([2, 3]),
|
44 |
| - SparseArray(np.take(self.arr_data, [2, 3]))) |
| 48 | + SparseArray(np.take(self.arr_data, |
| 49 | + [2, 3]))) |
45 | 50 | tm.assert_sp_array_equal(self.arr.take([0, 1, 2]),
|
46 | 51 | SparseArray(np.take(self.arr_data,
|
47 | 52 | [0, 1, 2])))
|
|
0 commit comments