Skip to content

Commit edbaf44

Browse files
committed
test larger sides to avoid a segfault
1 parent a864b1d commit edbaf44

File tree

2 files changed

+8
-2
lines changed

2 files changed

+8
-2
lines changed

pandas/_libs/reshape.pyx

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -114,11 +114,11 @@ def explode(ndarray[object] values):
114114
Py_ssize_t i, j, count, n
115115
object v
116116
ndarray[object] result
117-
ndarray[uint8_t] counts
117+
ndarray[int64_t] counts
118118

119119
# find the resulting len
120120
n = len(values)
121-
counts = numpy.zeros(n, dtype='uint8')
121+
counts = numpy.zeros(n, dtype='int64')
122122
for i in range(n):
123123
v = values[i]
124124
if is_list_like(v):

pandas/tests/series/test_explode.py

Lines changed: 6 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -52,3 +52,9 @@ def test_multi_index():
5252
[0, 1, 2, np.nan, np.nan, 3, 4], index=index, dtype=object, name="foo"
5353
)
5454
tm.assert_series_equal(result, expected)
55+
56+
57+
def test_large():
58+
s = pd.Series([range(256)]).explode()
59+
result = s.explode()
60+
tm.assert_series_equal(result, s)

0 commit comments

Comments
 (0)