Skip to content

Commit 7842218

Browse files
committed
use indexer dtypes in tests for compatibility with windows
1 parent 36c6d4e commit 7842218

File tree

1 file changed

+13
-12
lines changed

1 file changed

+13
-12
lines changed

pandas/tests/indexes/multi/test_indexing.py

Lines changed: 13 additions & 12 deletions
Original file line numberDiff line numberDiff line change
@@ -190,23 +190,23 @@ def test_get_indexer_methods():
190190
mult_idx_2 = MultiIndex.from_product([[0], [1, 3, 4]])
191191

192192
indexer = mult_idx_1.get_indexer(mult_idx_2)
193-
expected = np.array([-1, 2, 3], dtype="int64")
193+
expected = np.array([-1, 2, 3], dtype=indexer.dtype)
194194
tm.assert_almost_equal(expected, indexer)
195195

196196
backfill_indexer = mult_idx_1.get_indexer(mult_idx_2, method="backfill")
197-
expected = np.array([1, 2, 3], dtype="int64")
197+
expected = np.array([1, 2, 3], dtype=backfill_indexer.dtype)
198198
tm.assert_almost_equal(expected, backfill_indexer)
199199

200200
backfill_indexer = mult_idx_1.get_indexer(mult_idx_2, method="bfill")
201-
expected = np.array([1, 2, 3], dtype="int64")
201+
expected = np.array([1, 2, 3], dtype=backfill_indexer.dtype)
202202
tm.assert_almost_equal(expected, backfill_indexer)
203203

204204
pad_indexer = mult_idx_1.get_indexer(mult_idx_2, method="pad")
205-
expected = np.array([0, 2, 3], dtype="int64")
205+
expected = np.array([0, 2, 3], dtype=pad_indexer.dtype)
206206
tm.assert_almost_equal(expected, pad_indexer)
207207

208208
pad_indexer = mult_idx_1.get_indexer(mult_idx_2, method="ffill")
209-
expected = np.array([0, 2, 3], dtype="int64")
209+
expected = np.array([0, 2, 3], dtype=pad_indexer.dtype)
210210
tm.assert_almost_equal(expected, pad_indexer)
211211

212212

@@ -260,12 +260,12 @@ def test_get_indexer_three_or_more_levels():
260260

261261
# test with backfilling
262262
indexer_backfilled = mult_idx_1.get_indexer(mult_idx_2, method="backfill")
263-
expected = np.array([0, 4, 5, 6, 6, 6, -1], dtype="int64")
263+
expected = np.array([0, 4, 5, 6, 6, 6, -1], dtype=indexer_backfilled.dtype)
264264
tm.assert_almost_equal(expected, indexer_backfilled)
265265

266266
# now, the same thing, but forward-filled (aka "padded")
267267
indexer_padded = mult_idx_1.get_indexer(mult_idx_2, method="pad")
268-
expected = np.array([-1, 3, 5, 5, 5, 5, 11], dtype="int64")
268+
expected = np.array([-1, 3, 5, 5, 5, 5, 11], dtype=indexer_padded.dtype)
269269
tm.assert_almost_equal(expected, indexer_padded)
270270

271271
# now, do the indexing in the other direction
@@ -288,11 +288,12 @@ def test_get_indexer_three_or_more_levels():
288288
tm.assert_almost_equal(expected, indexer)
289289

290290
backfill_indexer = mult_idx_2.get_indexer(mult_idx_1, method="bfill")
291-
expected = np.array([1, 1, 1, 1, 2, 2, 6, 6, 6, 6, 6, 6], dtype="int64")
291+
expected = np.array([1, 1, 1, 1, 2, 2, 6, 6, 6, 6, 6, 6],
292+
dtype=backfill_indexer.dtype)
292293
tm.assert_almost_equal(expected, backfill_indexer)
293294

294295
pad_indexer = mult_idx_2.get_indexer(mult_idx_1, method="pad")
295-
expected = np.array([0, 0, 0, 0, 1, 2, 5, 5, 5, 5, 5, 5], dtype="int64")
296+
expected = np.array([0, 0, 0, 0, 1, 2, 5, 5, 5, 5, 5, 5],dtype="int64")
296297
tm.assert_almost_equal(expected, pad_indexer)
297298

298299

@@ -331,15 +332,15 @@ def test_get_indexer_backfill_with_carrying():
331332
assert mult_idx_1[-1] < mult_idx_2[1]
332333

333334
indexer = mult_idx_1.get_indexer(mult_idx_2)
334-
expected = np.array([-1, -1], dtype="int64")
335+
expected = np.array([-1, -1], dtype=indexer.dtype)
335336
tm.assert_almost_equal(expected, indexer)
336337

337338
backfill_indexer = mult_idx_1.get_indexer(mult_idx_2, method="bfill")
338-
expected = np.array([8, -1], dtype="int64")
339+
expected = np.array([8, -1], dtype=backfill_indexer.dtype)
339340
tm.assert_almost_equal(expected, backfill_indexer)
340341

341342
pad_indexer = mult_idx_1.get_indexer(mult_idx_2, method="ffill")
342-
expected = np.array([7, 15], dtype="int64")
343+
expected = np.array([7, 15], dtype=pad_indexer.dtype)
343344
tm.assert_almost_equal(expected, pad_indexer)
344345

345346

0 commit comments

Comments
 (0)