From 5ef08f841ae8a0d6a299020e37f4da2220057c4a Mon Sep 17 00:00:00 2001
From: jeremy <jeremy.bailleux@memgraph.io>
Date: Mon, 31 Oct 2022 11:56:20 +0100
Subject: [PATCH] Update Dataset creation script
---
tests/mgbench/dataset_creator.py | 4 ++--
tests/mgbench/datasets.py | 26 +++++++++++++-------------
2 files changed, 15 insertions(+), 15 deletions(-)
diff --git a/tests/mgbench/dataset_creator.py b/tests/mgbench/dataset_creator.py
index 1b48838b6..c63ae9731 100644
--- a/tests/mgbench/dataset_creator.py
+++ b/tests/mgbench/dataset_creator.py
@@ -100,8 +100,8 @@ if __name__ == "__main__":
for outer_index in range(0, number_of_files):
for inner_index in range(0, number_of_identities):
- file_uuid = outer_index
- identity_uuid = number_of_files + inner_index
+ file_uuid = outer_index + 1
+ identity_uuid = number_of_files + inner_index + 1
if random.random() <= percentage_of_permissions:
f.write(f'CREATE (:Permission {{uuid: {uuid}, name: "name_permission_{uuid}"}});\n')
diff --git a/tests/mgbench/datasets.py b/tests/mgbench/datasets.py
index c2e8499bd..953111807 100644
--- a/tests/mgbench/datasets.py
+++ b/tests/mgbench/datasets.py
@@ -314,26 +314,26 @@ class AccessControl(Dataset):
"edges": -1, # not used
"uuid_ranges": {
"File": {"first_uuid": 1, "last_uuid": 10},
- "Permission": {"first_uuid": 11, "last_uuid": 20},
- "Identity": {"first_uuid": 21, "last_uuid": 30},
+ "Identity": {"first_uuid": 11, "last_uuid": 20},
+ "Permission": {"first_uuid": 21, "last_uuid": 120}, # 120=10*10+20
},
},
"medium": {
+ "vertices": 3000,
+ "edges": -1, # not used
+ "uuid_ranges": {
+ "File": {"first_uuid": 1, "last_uuid": 1000},
+ "Identity": {"first_uuid": 1001, "last_uuid": 2000},
+ "Permission": {"first_uuid": 2001, "last_uuid": 1002000}, # 1002000=1000*1000+2000
+ },
+ },
+ "large": {
"vertices": 30000,
"edges": -1, # not used
"uuid_ranges": {
"File": {"first_uuid": 1, "last_uuid": 10000},
- "Permission": {"first_uuid": 10001, "last_uuid": 20000},
- "Identity": {"first_uuid": 10001, "last_uuid": 30000},
- },
- },
- "large": {
- "vertices": 3000000,
- "edges": -1, # not used
- "uuid_ranges": {
- "File": {"first_uuid": 1, "last_uuid": 1000000},
- "Permission": {"first_uuid": 100001, "last_uuid": 2000000},
- "Identity": {"first_uuid": 1000001, "last_uuid": 3000000},
+ "Identity": {"first_uuid": 10001, "last_uuid": 20000},
+ "Permission": {"first_uuid": 20001, "last_uuid": 100020000}, # 100020000=10000*10000+20000
},
},
}