diff options
author | Lars Wirzenius <liw@liw.fi> | 2013-01-20 10:50:37 +0000 |
---|---|---|
committer | Lars Wirzenius <liw@liw.fi> | 2013-01-20 10:50:37 +0000 |
commit | 669695c1fb3bdc8f6a169ddd2c1eb2bf10fb6833 (patch) | |
tree | c927ec8c68067e0b8603b2953da27c2ab5b25aa4 /larch | |
parent | 206fb1529bf3bef9c51e9c47dd5b3fa053c485ba (diff) | |
parent | 42d9d6c7d048580e06cab7003db5fdd61a0a3f11 (diff) | |
download | larch-669695c1fb3bdc8f6a169ddd2c1eb2bf10fb6833.tar.gz |
Fix speed-test
Diffstat (limited to 'larch')
-rw-r--r-- | larch/forest_tests.py | 3 | ||||
-rw-r--r-- | larch/nodestore.py | 3 | ||||
-rw-r--r-- | larch/nodestore_disk.py | 3 | ||||
-rw-r--r-- | larch/nodestore_memory.py | 5 | ||||
-rw-r--r-- | larch/nodestore_memory_tests.py | 3 | ||||
-rw-r--r-- | larch/tree_tests.py | 9 |
6 files changed, 17 insertions, 9 deletions
diff --git a/larch/forest_tests.py b/larch/forest_tests.py index 9a786bd..b390844 100644 --- a/larch/forest_tests.py +++ b/larch/forest_tests.py @@ -26,7 +26,8 @@ class ForestTests(unittest.TestCase): def setUp(self): self.codec = larch.NodeCodec(3) - self.ns = larch.NodeStoreMemory(64, self.codec) + self.ns = larch.NodeStoreMemory( + allow_writes=True, node_size=64, codec=self.codec) self.forest = larch.Forest(self.ns) def test_new_node_ids_grow(self): diff --git a/larch/nodestore.py b/larch/nodestore.py index 929450a..dd9b536 100644 --- a/larch/nodestore.py +++ b/larch/nodestore.py @@ -98,7 +98,8 @@ class NodeStore(object): # pragma: no cover ''' - def __init__(self, node_size, codec): + def __init__(self, allow_writes, node_size, codec): + self.allow_writes = allow_writes self.node_size = node_size self.codec = codec self.max_value_size = (node_size / 2) - codec.leaf_header.size diff --git a/larch/nodestore_disk.py b/larch/nodestore_disk.py index 197a411..6e8cd2f 100644 --- a/larch/nodestore_disk.py +++ b/larch/nodestore_disk.py @@ -117,7 +117,8 @@ class NodeStoreDisk(larch.NodeStore): if format is not None: tracing.trace('forcing format_base: %s', format) self.format_base = format - larch.NodeStore.__init__(self, node_size, codec) + larch.NodeStore.__init__( + self, allow_writes=allow_writes, node_size=node_size, codec=codec) self.dirname = dirname self.metadata_name = os.path.join(dirname, 'metadata') self.metadata = None diff --git a/larch/nodestore_memory.py b/larch/nodestore_memory.py index 510ce5e..9ba2b02 100644 --- a/larch/nodestore_memory.py +++ b/larch/nodestore_memory.py @@ -26,8 +26,9 @@ class NodeStoreMemory(larch.NodeStore): ''' - def __init__(self, node_size, codec): - larch.NodeStore.__init__(self, node_size, codec) + def __init__(self,allow_writes, node_size, codec): + larch.NodeStore.__init__( + self, allow_writes=allow_writes, node_size=node_size, codec=codec) self.nodes = dict() self.refcounts = dict() self.metadata = dict() diff --git a/larch/nodestore_memory_tests.py b/larch/nodestore_memory_tests.py index eecb905..56e3806 100644 --- a/larch/nodestore_memory_tests.py +++ b/larch/nodestore_memory_tests.py @@ -25,5 +25,6 @@ class NodeStoreMemoryTests(unittest.TestCase, larch.NodeStoreTests): def setUp(self): self.node_size = 4096 self.codec = larch.NodeCodec(self.key_bytes) - self.ns = nodestore_memory.NodeStoreMemory(self.node_size, self.codec) + self.ns = nodestore_memory.NodeStoreMemory( + allow_writes=True, node_size=self.node_size, codec=self.codec) diff --git a/larch/tree_tests.py b/larch/tree_tests.py index 86895f0..2fee2d5 100644 --- a/larch/tree_tests.py +++ b/larch/tree_tests.py @@ -67,7 +67,8 @@ class BTreeTests(unittest.TestCase): # We use a small node size so that all code paths are traversed # during testing. Use coverage.py to make sure they do. self.codec = larch.NodeCodec(3) - self.ns = DummyNodeStore(64, self.codec) + self.ns = DummyNodeStore( + allow_writes=True, node_size=64, codec=self.codec) self.forest = DummyForest() self.tree = larch.BTree(self.forest, self.ns, None) self.dump = False @@ -549,7 +550,8 @@ class BTreeDecrementTests(unittest.TestCase): # We use a small node size so that all code paths are traversed # during testing. Use coverage.py to make sure they do. self.codec = larch.NodeCodec(3) - self.ns = DummyNodeStore(64, self.codec) + self.ns = DummyNodeStore( + allow_writes=True, node_size=64, codec=self.codec) self.forest = DummyForest() self.tree = larch.BTree(self.forest, self.ns, None) self.tree.insert('foo', 'bar') @@ -574,7 +576,8 @@ class BTreeDecrementTests(unittest.TestCase): class BTreeBalanceTests(unittest.TestCase): def setUp(self): - ns = DummyNodeStore(4096, larch.NodeCodec(2)) + ns = DummyNodeStore( + allow_writes=True, node_size=4096, codec=larch.NodeCodec(2)) forest = DummyForest() self.tree = larch.BTree(forest, ns, None) self.keys = ['%02d' % i for i in range(10)] |