From 25d07df155fc398a74410adda17c75e24a22e593 Mon Sep 17 00:00:00 2001 From: nohehf Date: Mon, 13 May 2024 17:11:33 +0200 Subject: [PATCH] chore: bench file --- tests/bench.py | 38 ++++++++++++++++++++++++++++++++++++++ 1 file changed, 38 insertions(+) create mode 100644 tests/bench.py diff --git a/tests/bench.py b/tests/bench.py new file mode 100644 index 0000000..82f9d20 --- /dev/null +++ b/tests/bench.py @@ -0,0 +1,38 @@ +import os +import time + +from stack_graphs_python import index, Querier, Position, Language + +# index ./js_sample directory + +# convert ./js_sample directory to absolute path +dir = os.path.abspath("./tests/js_sample") +db_path = os.path.abspath("./db.sqlite") + +print("Indexing directory: ", dir) +print("Database path: ", db_path) + +index([dir], db_path, language=Language.TypeScript) + +source_reference = Position(path=dir + "/index.js", line=2, column=12) + +print("Querying definition for: ", source_reference.path) + +querier = Querier(db_path) + + +def run_n_queries(n): + start = time.time() + for i in range(n): + querier.definitions(source_reference) + end = time.time() + # delta in ms + delta = (end - start) * 1000 + print( + f"{n} queries took: {delta}ms", + ) + + +run_n_queries(100) +run_n_queries(1000) +run_n_queries(10000)