Skip to content

Commit 5abb02e

Browse files
authored
TST: add ASV check for groupby with uint64 col (#41859)
1 parent 1e266f2 commit 5abb02e

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

asv_bench/benchmarks/groupby.py

+3-1
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)