diff --git a/age_adapter/benchmark.py b/age_adapter/benchmark.py index 4262dac14..11b22107d 100644 --- a/age_adapter/benchmark.py +++ b/age_adapter/benchmark.py @@ -43,7 +43,7 @@ async def main(): await age_adapter.connect() await neo4j_adapter.initialize() - batch_size = 3000 + batch_size = 500 node_ids = [f"node_{i}" for i in range(batch_size)] nodes = [(nid, ["TestNode"], {"name": f"Node {i}", "value": i}) for i, nid in enumerate(node_ids)] @@ -339,36 +339,6 @@ async def main(): print(f"Count Edges: AGE={age_time_count_edges:.4f}s, Neo4j={neo4j_time_count_edges:.4f}s") - start = time.perf_counter() - query = f""" - MATCH (a {{id: '{query_node_ids[0]}'}})-[*1..3]-(b {{id: '{query_node_ids[5]}'}}) - RETURN a, b - LIMIT 1 - """ - try: - await age_adapter.execute_cypher(query) - age_time_path_query = time.perf_counter() - start - except Exception as e: - age_time_path_query = time.perf_counter() - start - - start = time.perf_counter() - try: - src_uuid = UUID(query_node_ids[0]) if '-' in query_node_ids[0] else UUID(int=hash(query_node_ids[0]) & ((1 << 128) - 1)) - except: - src_uuid = UUID(int=hash(query_node_ids[0]) & ((1 << 128) - 1)) - try: - tgt_uuid = UUID(query_node_ids[5]) if '-' in query_node_ids[5] else UUID(int=hash(query_node_ids[5]) & ((1 << 128) - 1)) - except: - tgt_uuid = UUID(int=hash(query_node_ids[5]) & ((1 << 128) - 1)) - query = f""" - MATCH (a:`__Node__`{{id: $src_id}})-[*1..3]-(b:`__Node__`{{id: $tgt_id}}) - RETURN a, b - LIMIT 1 - """ - await neo4j_adapter.query(query, {"src_id": str(src_uuid), "tgt_id": str(tgt_uuid)}) - neo4j_time_path_query = time.perf_counter() - start - - print(f"Path Query (1-3 hops): AGE={age_time_path_query:.4f}s, Neo4j={neo4j_time_path_query:.4f}s") await age_adapter.close() await neo4j_adapter.driver.close()