Skip to content

Commit 273e6c8

Browse files
committed
TST: Update tests in multi for new return
Previously these relied worked around the return type by wrapping list-likes in `np.array` and relying on that to cast to float. These workarounds are no longer nescessary.
1 parent 6016720 commit 273e6c8

File tree

1 file changed

+4
-5
lines changed

1 file changed

+4
-5
lines changed

pandas/tests/indexes/test_multi.py

Lines changed: 4 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -776,7 +776,7 @@ def test_from_arrays_empty(self):
776776
arrays = [[]] * N
777777
names = list('ABC')[:N]
778778
result = MultiIndex.from_arrays(arrays=arrays, names=names)
779-
expected = MultiIndex(levels=[np.array([])] * N, labels=[[]] * N,
779+
expected = MultiIndex(levels=[[]] * N, labels=[[]] * N,
780780
names=names)
781781
tm.assert_index_equal(result, expected)
782782

@@ -829,7 +829,7 @@ def test_from_product_empty(self):
829829

830830
# 1 level
831831
result = MultiIndex.from_product([[]], names=['A'])
832-
expected = pd.Float64Index([], name='A')
832+
expected = pd.Index([], name='A')
833833
tm.assert_index_equal(result, expected)
834834

835835
# 2 levels
@@ -838,7 +838,7 @@ def test_from_product_empty(self):
838838
names = ['A', 'B']
839839
for first, second in zip(l1, l2):
840840
result = MultiIndex.from_product([first, second], names=names)
841-
expected = MultiIndex(levels=[np.array(first), np.array(second)],
841+
expected = MultiIndex(levels=[first, second],
842842
labels=[[], []], names=names)
843843
tm.assert_index_equal(result, expected)
844844

@@ -847,8 +847,7 @@ def test_from_product_empty(self):
847847
for N in range(4):
848848
lvl2 = lrange(N)
849849
result = MultiIndex.from_product([[], lvl2, []], names=names)
850-
expected = MultiIndex(levels=[np.array(A)
851-
for A in [[], lvl2, []]],
850+
expected = MultiIndex(levels=[[], lvl2, []],
852851
labels=[[], [], []], names=names)
853852
tm.assert_index_equal(result, expected)
854853

0 commit comments

Comments
 (0)