Skip to content

Commit e5a8d4f

Browse files
Terji PetersenTerji Petersen
authored andcommitted
DEPR: don't make Index instantiate Int64/Uint64/Flaot64Index
1 parent b32c503 commit e5a8d4f

File tree

1 file changed

+6
-8
lines changed

1 file changed

+6
-8
lines changed

pandas/tests/reshape/merge/test_multi.py

Lines changed: 6 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -441,14 +441,13 @@ def test_merge_datetime_index(self, klass):
441441
if klass is not None:
442442
on_vector = klass(on_vector)
443443

444-
expected = DataFrame({"a": [1, 2, 3], "key_1": [2016, 2017, 2018]})
444+
exp_years = np.array([2016, 2017, 2018], dtype=np.int32)
445+
expected = DataFrame({"a": [1, 2, 3], "key_1": exp_years})
445446

446447
result = df.merge(df, on=["a", on_vector], how="inner")
447448
tm.assert_frame_equal(result, expected)
448449

449-
expected = DataFrame(
450-
{"key_0": [2016, 2017, 2018], "a_x": [1, 2, 3], "a_y": [1, 2, 3]}
451-
)
450+
expected = DataFrame({"key_0": exp_years, "a_x": [1, 2, 3], "a_y": [1, 2, 3]})
452451

453452
result = df.merge(df, on=[df.index.year], how="inner")
454453
tm.assert_frame_equal(result, expected)
@@ -852,14 +851,13 @@ def test_merge_datetime_index(self, box):
852851
if box is not None:
853852
on_vector = box(on_vector)
854853

855-
expected = DataFrame({"a": [1, 2, 3], "key_1": [2016, 2017, 2018]})
854+
exp_years = np.array([2016, 2017, 2018], dtype=np.int32)
855+
expected = DataFrame({"a": [1, 2, 3], "key_1": exp_years})
856856

857857
result = df.merge(df, on=["a", on_vector], how="inner")
858858
tm.assert_frame_equal(result, expected)
859859

860-
expected = DataFrame(
861-
{"key_0": [2016, 2017, 2018], "a_x": [1, 2, 3], "a_y": [1, 2, 3]}
862-
)
860+
expected = DataFrame({"key_0": exp_years, "a_x": [1, 2, 3], "a_y": [1, 2, 3]})
863861

864862
result = df.merge(df, on=[df.index.year], how="inner")
865863
tm.assert_frame_equal(result, expected)

0 commit comments

Comments
 (0)