Skip to content

Commit 23dae93

Browse files
anhqlesweb
authored andcommitted
CLN PEP8 compliance
1 parent 6b7b030 commit 23dae93

File tree

2 files changed

+6
-3
lines changed

2 files changed

+6
-3
lines changed

pandas/core/reshape/reshape.py

+4-2
Original file line numberDiff line numberDiff line change
@@ -127,10 +127,12 @@ def __init__(self, values, index, level=-1, value_columns=None,
127127
self.removed_level = self.new_index_levels.pop(self.level)
128128
self.removed_level_full = index.levels[self.level]
129129

130-
num_rows = np.max([index_level.size for index_level in self.new_index_levels])
130+
num_rows = np.max([index_level.size for index_level
131+
in self.new_index_levels])
131132
num_columns = self.removed_level.size
132133
if num_rows * num_columns > (2 ** 31 - 1):
133-
raise ValueError('Unstacked data frame is too big, causing int32 overflow')
134+
raise ValueError('Unstacked DataFrame is too big, '
135+
'causing int32 overflow')
134136

135137
self._make_sorted_values_labels()
136138
self._make_selectors()

pandas/tests/test_multilevel.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -1215,7 +1215,8 @@ def test_unstack_unobserved_keys(self):
12151215
@pytest.mark.slow
12161216
def test_unstack_number_of_levels_larger_than_int32(self):
12171217
# GH 20601
1218-
df = DataFrame(np.random.randn(2 ** 16, 2), index=[np.arange(2 ** 16), np.arange(2 ** 16)])
1218+
df = DataFrame(np.random.randn(2 ** 16, 2),
1219+
index=[np.arange(2 ** 16), np.arange(2 ** 16)])
12191220
with tm.assert_raises_regex(ValueError, 'int32 overflow'):
12201221
df.unstack()
12211222

0 commit comments

Comments
 (0)