From 342efad5db8e7a8eb73be5025246b013f6b87e12 Mon Sep 17 00:00:00 2001 From: jschendel Date: Fri, 9 Nov 2018 16:46:09 -0700 Subject: [PATCH] TST: Use intp as expected dtype in IntervalIndex indexing tests --- pandas/tests/indexes/interval/test_interval.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/pandas/tests/indexes/interval/test_interval.py b/pandas/tests/indexes/interval/test_interval.py index 258f2dc41fb79..49d093d312cf1 100644 --- a/pandas/tests/indexes/interval/test_interval.py +++ b/pandas/tests/indexes/interval/test_interval.py @@ -412,9 +412,9 @@ def test_get_loc_value(self): assert idx.get_loc(0.5) == 0 assert idx.get_loc(1) == 0 tm.assert_numpy_array_equal(idx.get_loc(1.5), - np.array([0, 1], dtype='int64')) + np.array([0, 1], dtype='intp')) tm.assert_numpy_array_equal(np.sort(idx.get_loc(2)), - np.array([0, 1], dtype='int64')) + np.array([0, 1], dtype='intp')) assert idx.get_loc(3) == 1 pytest.raises(KeyError, idx.get_loc, 3.5) @@ -537,12 +537,12 @@ def test_get_loc_datetimelike_overlapping(self, arrays): value = index[0].mid + Timedelta('12 hours') result = np.sort(index.get_loc(value)) - expected = np.array([0, 1], dtype='int64') + expected = np.array([0, 1], dtype='intp') assert tm.assert_numpy_array_equal(result, expected) interval = Interval(index[0].left, index[1].right) result = np.sort(index.get_loc(interval)) - expected = np.array([0, 1, 2], dtype='int64') + expected = np.array([0, 1, 2], dtype='intp') assert tm.assert_numpy_array_equal(result, expected) # To be removed, replaced by test_interval_new.py (see #16316, #16386) @@ -617,7 +617,7 @@ def test_get_reindexer_datetimelike(self, arrays): target = IntervalIndex.from_tuples(tuples) result = index._get_reindexer(target) - expected = np.array([0, 3], dtype='int64') + expected = np.array([0, 3], dtype='intp') tm.assert_numpy_array_equal(result, expected) @pytest.mark.parametrize('breaks', [