Skip to content

Commit 7fd2601

Browse files
committed
ENH: Explicitly set dtype to int32 due to OS dependant behavior
1 parent 5f03406 commit 7fd2601

File tree

2 files changed

+2
-1
lines changed

2 files changed

+2
-1
lines changed

pandas/core/reshape/reshape.py

+1-1
Original file line numberDiff line numberDiff line change
@@ -160,7 +160,7 @@ def _make_selectors(self):
160160
comp_index = ensure_platform_int(comp_index)
161161
stride = self.index.levshape[self.level] + self.lift
162162
self.full_shape = ngroups, stride
163-
num_entries = np.prod(self.full_shape)
163+
num_entries = np.prod(self.full_shape, dtype=np.int32)
164164

165165
if num_entries < 0:
166166
raise ValueError('Unstack leads to too many (max int32) entries, '

pandas/tests/reshape/test_reshape.py

+1
Original file line numberDiff line numberDiff line change
@@ -584,6 +584,7 @@ def test_get_dummies_duplicate_columns(self, df):
584584
tm.assert_frame_equal(result, expected)
585585

586586

587+
@pytest.mark.slow
587588
def test_too_many_items_to_unstack_raises_value_error():
588589
index = pd.MultiIndex.from_arrays([np.arange(979758),
589590
np.arange(979758)],

0 commit comments

Comments
 (0)