Skip to content

Commit eac3fd8

Browse files
committed
PERF: Add new vbench tests to asv
1 parent 7dbae09 commit eac3fd8

File tree

3 files changed

+38
-4
lines changed

3 files changed

+38
-4
lines changed

asv_bench/asv.conf.json

+3-3
Original file line numberDiff line numberDiff line change
@@ -42,9 +42,9 @@
4242
"scipy": [],
4343
"numexpr": [],
4444
"tables": [],
45-
"openpyxl": [],
46-
"xlrd": [],
47-
"xlwt": []
45+
"openpyxl": [],
46+
"xlrd": [],
47+
"xlwt": []
4848
},
4949

5050
// The directory (relative to the current directory) that benchmarks are

asv_bench/benchmarks/groupby.py

+10
Original file line numberDiff line numberDiff line change
@@ -1660,6 +1660,16 @@ def time_groupby_transform_ufunc(self):
16601660
self.data.groupby(level='date').transform(np.max)
16611661

16621662

1663+
class series_value_counts_float64(object):
1664+
goal_time = 0.2
1665+
1666+
def setup(self):
1667+
self.s = Series(np.random.randint(0, 1000, size=100000)).astype(float)
1668+
1669+
def time_series_value_counts_float64(self):
1670+
self.s.value_counts()
1671+
1672+
16631673
class series_value_counts_int64(object):
16641674
goal_time = 0.2
16651675

asv_bench/benchmarks/indexing.py

+25-1
Original file line numberDiff line numberDiff line change
@@ -428,6 +428,30 @@ def time_series_loc_slice(self):
428428
self.s.loc[:800000]
429429

430430

431+
class series_take_dtindex(object):
432+
goal_time = 0.2
433+
434+
def setup(self):
435+
self.s = Series(np.random.rand(100000))
436+
self.ts = Series(np.random.rand(100000), index=date_range('2011-01-01', freq='S', periods=100000))
437+
self.indexer = ([True, False, True, True, False] * 20000)
438+
439+
def time_series_take_dtindex(self):
440+
self.ts.take(self.indexer)
441+
442+
443+
class series_take_intindex(object):
444+
goal_time = 0.2
445+
446+
def setup(self):
447+
self.s = Series(np.random.rand(100000))
448+
self.ts = Series(np.random.rand(100000), index=date_range('2011-01-01', freq='S', periods=100000))
449+
self.indexer = ([True, False, True, True, False] * 20000)
450+
451+
def time_series_take_intindex(self):
452+
self.s.take(self.indexer)
453+
454+
431455
class series_xs_mi_ix(object):
432456
goal_time = 0.2
433457

@@ -462,4 +486,4 @@ def setup(self):
462486
self.midx = self.midx.take(np.random.permutation(np.arange(100000)))
463487

464488
def time_sort_level_zero(self):
465-
self.midx.sortlevel(0)
489+
self.midx.sortlevel(0)

0 commit comments

Comments
 (0)