def initialize_diary(): diary = Diary(STUDENTS, CLASSES, DATES) for i in xrange(NUMBER_OF_SCORES): diary.add_score(random.choice(CLASSES), random.choice(STUDENTS), random.choice(SCORES)) for clazz in CLASSES: for date in DATES: for student in STUDENTS: if random.choice([True, False]): diary.add_attendance(clazz, student, date) return diary