Skip to content

Commit 0d3eaf8

Browse files
Clean-up timeseries/timedelta benchmarks
1 parent 38b472e commit 0d3eaf8

File tree

2 files changed

+240
-1027
lines changed

2 files changed

+240
-1027
lines changed

asv_bench/benchmarks/timedelta.py

+10-22
Original file line numberDiff line numberDiff line change
@@ -2,36 +2,24 @@
22
from pandas import to_timedelta
33

44

5-
class timedelta_convert_int(object):
5+
class Timedelta(object):
66
goal_time = 0.2
77

88
def setup(self):
99
self.arr = np.random.randint(0, 1000, size=10000)
10+
self.arr2 = ['{0} days'.format(i) for i in self.arr]
1011

11-
def time_timedelta_convert_int(self):
12-
to_timedelta(self.arr, unit='s')
13-
14-
15-
class timedelta_convert_string(object):
16-
goal_time = 0.2
12+
self.arr3 = np.random.randint(0, 60, size=10000)
13+
self.arr3 = ['00:00:{0:02d}'.format(i) for i in self.arr3]
1714

18-
def setup(self):
19-
self.arr = np.random.randint(0, 1000, size=10000)
20-
self.arr = ['{0} days'.format(i) for i in self.arr]
21-
22-
def time_timedelta_convert_string(self):
23-
to_timedelta(self.arr)
24-
25-
26-
class timedelta_convert_string_seconds(object):
27-
goal_time = 0.2
15+
def time_convert_int(self):
16+
to_timedelta(self.arr, unit='s')
2817

29-
def setup(self):
30-
self.arr = np.random.randint(0, 60, size=10000)
31-
self.arr = ['00:00:{0:02d}'.format(i) for i in self.arr]
18+
def time_convert_string(self):
19+
to_timedelta(self.arr2)
3220

33-
def time_timedelta_convert_string_seconds(self):
34-
to_timedelta(self.arr)
21+
def time_convert_string_seconds(self):
22+
to_timedelta(self.arr3)
3523

3624

3725
class timedelta_convert_bad_parse(object):

0 commit comments

Comments
 (0)