Add remove to speed-test.

1 files changed, 10 insertions, 0 deletions

diff --git a/speed-test b/speed-test
index 73a5d58..3a23eb9 100755
--- a/speed-test
+++ b/speed-test
@@ -125,6 +125,15 @@ def main():
         insert2_time = measure(keys, lambda key: tree.insert(key, value),
                                lambda: forest.commit())
 
+    # Measure removes from tree.
+    random.shuffle(keys)
+    if do_profile:
+        profile(keys, lambda key: tree.remove(key), 
+                lambda: forest.commit(), 'remove')
+    else:
+        remove_time = measure(keys, lambda key: tree.remove(key),
+                              lambda: forest.commit())
+
     # Report
     if do_profile:
         print 'View *.prof with ./viewprof for profiling results.'
@@ -135,6 +144,7 @@ def main():
         print 'lookup_range: %5.3f s (%.1f/s)' % (lookup_range_time, 
                                                  n/lookup_range_time)
         print 'insert2     : %5.3f s (%.1f/s)' % (insert2_time, n/insert2_time)
+        print 'remove      : %5.3f s (%.1f/s)' % (remove_time, n/remove_time)
 
     # Clean up
     if location: