diff options
author | Lars Wirzenius <liw@liw.fi> | 2011-02-23 15:17:23 +0000 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2011-02-23 15:17:23 +0000 |
commit | 5a561b1813b31df06a043ffcbbd53104e6cb8d98 (patch) | |
tree | 18b795ba2aa204b407c3ff67c280f4f578a8034d /speed-test | |
parent | a369301852ab60d9d71bd2180569a93993b30244 (diff) | |
download | larch-5a561b1813b31df06a043ffcbbd53104e6cb8d98.tar.gz |
Use btree.open_forest in speed-test.
Diffstat (limited to 'speed-test')
-rwxr-xr-x | speed-test | 9 |
1 files changed, 4 insertions, 5 deletions
@@ -60,8 +60,6 @@ class SpeedTest(cliapp.Application): value_size = 128 node_size = 64*1024 - codec = btree.NodeCodec(key_size) - n = self['keys'] location = self['location'] @@ -69,14 +67,15 @@ class SpeedTest(cliapp.Application): raise Exception('You must set number of keys with --keys') if not location: - ns = btree.NodeStoreMemory(node_size, codec) + forest = btree.open_forest(key_size=key_size, node_size=node_size, + node_store=btree.NodeStoreMemory) else: if os.path.exists(location): raise Exception('%s exists already' % location) os.mkdir(location) - ns = btree.NodeStoreDisk(node_size, codec, dirname=location) + forest = btree.open_forest(key_size=key_size, node_size=node_size, + dirname=location) - forest = btree.Forest(ns) tree = forest.new_tree() # Create list of keys. |