Skip to content

Commit ee4c2c7

Browse files
committed
TST: skip np.take comparisons on numpy < 1.8
closes #12784
1 parent 9f68a96 commit ee4c2c7

File tree

1 file changed

+8
-3
lines changed

1 file changed

+8
-3
lines changed

pandas/sparse/tests/test_array.py

+8-3
Original file line numberDiff line numberDiff line change
@@ -6,6 +6,7 @@
66
import operator
77
import warnings
88

9+
from pandas import _np_version_under1p8
910
from pandas.sparse.api import SparseArray
1011
from pandas.util.testing import assert_almost_equal, assertRaisesRegexp
1112
import pandas.util.testing as tm
@@ -37,11 +38,15 @@ def test_get_item(self):
3738
def test_take(self):
3839
self.assertTrue(np.isnan(self.arr.take(0)))
3940
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))
4246

4347
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])))
4550
tm.assert_sp_array_equal(self.arr.take([0, 1, 2]),
4651
SparseArray(np.take(self.arr_data,
4752
[0, 1, 2])))

0 commit comments

Comments
 (0)