adding variable for numbers

node/test_nodes.py

 # coding: utf-8
 
-
+from django.db import transaction
 from node.models import Node, NodeType
 from django.contrib.auth.models import User
 
@@ -20,6 +20,9 @@ nodeTypeProject.save()
 nodeTypeDocument = NodeType(name="Document")
 nodeTypeDocument.save()
 
+number_userIndex = 100
+number_projectIndex = 50
+number_documentIndex = 10000
 
 # In[ ]:
 
@@ -27,19 +30,19 @@ Node.objects.all().delete()
 User.objects.filter(username__startswith = "User #").delete()
 
 t0 = time()
-for userIndex in range(100):
+for userIndex in range(number_userIndex):
     userName = 'User #%d' % (userIndex, )
     user = User(username=userName)
     user.save()
     rootNode = Node.add_root(name=userName, type_id=nodeTypeRoot.pk, user_id=user.id)
     print(userName)
-    for projectIndex in range(50):
+    for projectIndex in range(number_projectIndex):
         projectName = 'Project #%d-%d' % (userIndex, projectIndex, )
         projectNode = Node(name=projectName, type_id=nodeTypeProject.pk, user_id=user.id)
         rootNode.add_child(instance=projectNode)
         documents = [
             {"data": {"name":'Document #%d-%d-%d' % (userIndex, projectIndex, documentIndex, ), "type_id":nodeTypeDocument.pk, "user_id":user.id}}
-            for documentIndex in range(10000)
+            for documentIndex in range(number_documentIndex)
         ]
         Node.load_bulk(documents, projectNode)
 print(time() - t0)