Add edge creation and bfs parallel tests
Reviewers: buda, mislav.bradac Reviewed By: mislav.bradac Subscribers: pullbot Differential Revision: https://phabricator.memgraph.io/D953
This commit is contained in:
20
tests/macro_benchmark/groups/bfs_parallel/setup.py
Normal file
20
tests/macro_benchmark/groups/bfs_parallel/setup.py
Normal file
@@ -0,0 +1,20 @@
|
||||
import random
|
||||
import common
|
||||
|
||||
for i in range(common.VERTEX_COUNT):
|
||||
print("CREATE (n: Node {id: %d});" % i)
|
||||
|
||||
print("CREATE INDEX ON :Node(id);")
|
||||
|
||||
# create a tree to be sure there is a path between each two nodes
|
||||
for i in range(1, common.VERTEX_COUNT):
|
||||
dad = int(random.random() * i)
|
||||
print("MATCH (a: Node {id: %d}), (b: Node {id: %d}) CREATE (a)-[:Friend]->(b);" % (dad, i))
|
||||
print("MATCH (a: Node {id: %d}), (b: Node {id: %d}) CREATE (a)-[:Friend]->(b);" % (i, dad))
|
||||
|
||||
# add random edges
|
||||
for i in range(common.VERTEX_COUNT * common.VERTEX_COUNT // common.SPARSE_FACTOR):
|
||||
a = int(random.random() * common.VERTEX_COUNT)
|
||||
b = int(random.random() * common.VERTEX_COUNT)
|
||||
print("MATCH (a: Node {id: %d}), (b: Node {id: %d}) CREATE (a)-[:Friend]->(b);" % (a, b))
|
||||
|
||||
Reference in New Issue
Block a user