Skip to content

Commit 7e4b9dc

Browse files
committed
revised hdfstore_bench to use new query syntax
rename table variables
1 parent f880728 commit 7e4b9dc

File tree

1 file changed

+23
-25
lines changed

1 file changed

+23
-25
lines changed

asv_bench/benchmarks/hdfstore_bench.py

Lines changed: 23 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -31,16 +31,12 @@ def setup(self):
3131
self.remove(self.f)
3232

3333
self.store = HDFStore(self.f)
34-
self.store.put('df1', self.df)
35-
self.store.put('df_mixed', self.df_mixed)
36-
37-
self.store.append('df5', self.df_mixed)
38-
self.store.append('df7', self.df)
39-
40-
self.store.append('df9', self.df_wide)
41-
42-
self.store.append('df11', self.df_wide2)
43-
self.store.append('df12', self.df2)
34+
self.store.put('fixed', self.df)
35+
self.store.put('fixed_mixed', self.df_mixed)
36+
self.store.append('table', self.df2)
37+
self.store.append('table_mixed', self.df_mixed)
38+
self.store.append('table_wide', self.df_wide)
39+
self.store.append('table_wide2', self.df_wide2)
4440

4541
def teardown(self):
4642
self.store.close()
@@ -52,45 +48,47 @@ def remove(self, f):
5248
pass
5349

5450
def time_read_store(self):
55-
self.store.get('df1')
51+
self.store.get('fixed')
5652

5753
def time_read_store_mixed(self):
58-
self.store.get('df_mixed')
54+
self.store.get('fixed_mixed')
5955

6056
def time_write_store(self):
61-
self.store.put('df2', self.df)
57+
self.store.put('fixed_write', self.df)
6258

6359
def time_write_store_mixed(self):
64-
self.store.put('df_mixed2', self.df_mixed)
60+
self.store.put('fixed_mixed_write', self.df_mixed)
6561

6662
def time_read_store_table_mixed(self):
67-
self.store.select('df5')
63+
self.store.select('table_mixed')
6864

6965
def time_write_store_table_mixed(self):
70-
self.store.append('df6', self.df_mixed)
66+
self.store.append('table_mixed_write', self.df_mixed)
7167

7268
def time_read_store_table(self):
73-
self.store.select('df7')
69+
self.store.select('table')
7470

7571
def time_write_store_table(self):
76-
self.store.append('df8', self.df)
72+
self.store.append('table_write', self.df)
7773

7874
def time_read_store_table_wide(self):
79-
self.store.select('df9')
75+
self.store.select('table_wide')
8076

8177
def time_write_store_table_wide(self):
82-
self.store.append('df10', self.df_wide)
78+
self.store.append('table_wide_write', self.df_wide)
8379

8480
def time_write_store_table_dc(self):
85-
self.store.append('df15', self.df, data_columns=True)
81+
self.store.append('table_dc_write', self.df_dc, data_columns=True)
8682

8783
def time_query_store_table_wide(self):
88-
self.store.select('df11', [('index', '>', self.df_wide2.index[10000]),
89-
('index', '<', self.df_wide2.index[15000])])
84+
start = self.df_wide2.index[10000]
85+
stop = self.df_wide2.index[15000]
86+
self.store.select('table_wide', where="index > start and index < stop")
9087

9188
def time_query_store_table(self):
92-
self.store.select('df12', [('index', '>', self.df2.index[10000]),
93-
('index', '<', self.df2.index[15000])])
89+
start = self.df2.index[10000]
90+
stop = self.df2.index[15000]
91+
self.store.select('table', where="index > start and index < stop")
9492

9593

9694
class HDF5Panel(object):

0 commit comments

Comments
 (0)