Skip to content

Commit 6de9afc

Browse files
committed
Add short str test
1 parent 17d0b23 commit 6de9afc

File tree

1 file changed

+10
-6
lines changed

1 file changed

+10
-6
lines changed

vb_suite/factorize.py

+10-6
Original file line numberDiff line numberDiff line change
@@ -16,21 +16,25 @@
1616
indices = np.random.randint(100, size=SIZE)
1717
int_values_dup = int_values_uniq.take(indices)
1818
str_values_dup = str_values_uniq.take(indices)
19+
shortstr_values_dup = Index(np.take(['AA', 'BB', 'CC', 'DD'],
20+
np.random.randint(4, size=SIZE)))
1921
float_values_dup = float_values_uniq.take(indices)
2022
"""
2123

2224

23-
factorize_int_dup = Benchmark("factorize(int_values_dup)", setup,
24-
start_date=START_DATE)
2525
factorize_int_uniq = Benchmark("factorize(int_values_uniq)", setup,
2626
start_date=START_DATE)
27-
28-
factorize_str_dup = Benchmark("factorize(str_values_dup)", setup,
27+
factorize_int_dup = Benchmark("factorize(int_values_dup)", setup,
2928
start_date=START_DATE)
29+
3030
factorize_str_uniq = Benchmark("factorize(str_values_uniq)", setup,
3131
start_date=START_DATE)
32+
factorize_str_dup = Benchmark("factorize(str_values_dup)", setup,
33+
start_date=START_DATE)
34+
factorize_shortstr_dup = Benchmark("factorize(shortstr_values_dup)", setup,
35+
start_date=START_DATE)
3236

33-
factorize_float_dup = Benchmark("factorize(float_values_dup)", setup,
34-
start_date=START_DATE)
3537
factorize_float_uniq = Benchmark("factorize(float_values_uniq)", setup,
3638
start_date=START_DATE)
39+
factorize_float_dup = Benchmark("factorize(float_values_dup)", setup,
40+
start_date=START_DATE)

0 commit comments

Comments
 (0)