def setUpClass(self):
# Create 2 users, 2 datasets, 2 models and 20 documents
# user0 -> dataset0 (public) -> doc00 .. doc09 (id=1..10)
# user1 -> dataset1 (private) -> doc10 .. doc19 (id=11.20)
for i in range(2):
user = User()
user.username = "******" + str(i)
user.save()
dataset = Dataset()
dataset.owner = user
dataset.text_id = "dataset" + str(i)
dataset.is_public = (i == 0)
dataset.save()
for j in range(10):
document = Document()
document.dataset = dataset
document.index_id = j
document.title = "doc" + str(i) + str(j)
document.text = "text" + str(i) + str(j)
document.snippet = "snippet" + str(i) + str(j)
document.url = "url" + str(i) + str(j)
document.save()
