From 483b549020a7c8587e82b3e617b14990e55dd747 Mon Sep 17 00:00:00 2001
From: Nezar Abdennur <nabdennur@gmail.com>
Date: Tue, 13 Feb 2024 07:58:47 -0500
Subject: [PATCH] Add timing script for benchmarking

---
 bench/bench.py | 44 ++++++++++++++++++++++++++++++++++++++++++++
 1 file changed, 44 insertions(+)
 create mode 100644 bench/bench.py

diff --git a/bench/bench.py b/bench/bench.py
new file mode 100644
index 0000000..09ea3c7
--- /dev/null
+++ b/bench/bench.py
@@ -0,0 +1,44 @@
+import random
+import timeit
+
+import pqdict
+
+
+def get_dict(n):
+    a = pqdict.pqdict()
+    for _ in range(n):
+        a[random.random()] = random.random()
+    return a
+
+
+def update_dict(a):
+    for key, value in a.items():
+        a[key] = value + 0.001
+    return a
+
+
+def copy_dict(a):
+    return a.copy()
+
+
+def consume_dict(a):
+    while a:
+        assert a.popitem()[1]
+
+
+if __name__ == '__main__':
+    random.seed(1298472)
+
+    a = get_dict(10_000)
+
+    print("Time it takes to create the pqdict:")
+    print(timeit.timeit(lambda: get_dict(1000), number=100))
+
+    print("Time it takes to copy using pqdict.copy():")
+    print(timeit.timeit(lambda: copy_dict(a), number=100))
+
+    print("Time it takes to consume the pqdict after pqdict.copy():")
+    print(timeit.timeit(lambda: consume_dict(copy_dict(a)), number=100))
+
+    print("Time it takes to update all keys in the dict")
+    print(timeit.timeit(lambda: update_dict(a), number=100))