From 5a561b1813b31df06a043ffcbbd53104e6cb8d98 Mon Sep 17 00:00:00 2001 From: Lars Wirzenius Date: Wed, 23 Feb 2011 15:17:23 +0000 Subject: Use btree.open_forest in speed-test. --- speed-test | 9 ++++----- 1 file changed, 4 insertions(+), 5 deletions(-) (limited to 'speed-test') diff --git a/speed-test b/speed-test index 599ed18..7880c90 100755 --- a/speed-test +++ b/speed-test @@ -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. -- cgit v1.2.1