@@ -190,23 +190,23 @@ def test_get_indexer_methods():
190
190
mult_idx_2 = MultiIndex .from_product ([[0 ], [1 , 3 , 4 ]])
191
191
192
192
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 )
194
194
tm .assert_almost_equal (expected , indexer )
195
195
196
196
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 )
198
198
tm .assert_almost_equal (expected , backfill_indexer )
199
199
200
200
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 )
202
202
tm .assert_almost_equal (expected , backfill_indexer )
203
203
204
204
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 )
206
206
tm .assert_almost_equal (expected , pad_indexer )
207
207
208
208
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 )
210
210
tm .assert_almost_equal (expected , pad_indexer )
211
211
212
212
@@ -260,12 +260,12 @@ def test_get_indexer_three_or_more_levels():
260
260
261
261
# test with backfilling
262
262
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 )
264
264
tm .assert_almost_equal (expected , indexer_backfilled )
265
265
266
266
# now, the same thing, but forward-filled (aka "padded")
267
267
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 )
269
269
tm .assert_almost_equal (expected , indexer_padded )
270
270
271
271
# now, do the indexing in the other direction
@@ -288,11 +288,12 @@ def test_get_indexer_three_or_more_levels():
288
288
tm .assert_almost_equal (expected , indexer )
289
289
290
290
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 )
292
293
tm .assert_almost_equal (expected , backfill_indexer )
293
294
294
295
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" )
296
297
tm .assert_almost_equal (expected , pad_indexer )
297
298
298
299
@@ -331,15 +332,15 @@ def test_get_indexer_backfill_with_carrying():
331
332
assert mult_idx_1 [- 1 ] < mult_idx_2 [1 ]
332
333
333
334
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 )
335
336
tm .assert_almost_equal (expected , indexer )
336
337
337
338
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 )
339
340
tm .assert_almost_equal (expected , backfill_indexer )
340
341
341
342
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 )
343
344
tm .assert_almost_equal (expected , pad_indexer )
344
345
345
346
0 commit comments