Skip to content

Commit f212247

Browse files
smithto1JulianWgs
authored andcommitted
TST: add ASV check for groupby with uint64 col (pandas-dev#41859)
1 parent 4579146 commit f212247

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

asv_bench/benchmarks/groupby.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -393,7 +393,7 @@ class GroupByMethods:
393393

394394
param_names = ["dtype", "method", "application"]
395395
params = [
396-
["int", "float", "object", "datetime"],
396+
["int", "float", "object", "datetime", "uint"],
397397
[
398398
"all",
399399
"any",
@@ -442,6 +442,8 @@ def setup(self, dtype, method, application):
442442
values = rng.take(np.random.randint(0, ngroups, size=size))
443443
if dtype == "int":
444444
key = np.random.randint(0, size, size=size)
445+
elif dtype == "uint":
446+
key = np.random.randint(0, size, size=size, dtype=dtype)
445447
elif dtype == "float":
446448
key = np.concatenate(
447449
[np.random.random(ngroups) * 0.1, np.random.random(ngroups) * 10.0]

0 commit comments

Comments
 (0)