diff --git a/seminar/testutils.py b/seminar/testutils.py index 017c621a..e39a2d01 100644 --- a/seminar/testutils.py +++ b/seminar/testutils.py @@ -61,7 +61,7 @@ def gen_osoby(size): -def gen_skoly(): +def gen_skoly(): #TODO někdy to přepsat, aby jich bylo více skoly = [] prvnizs = Skola.objects.create(mesto='Praha', stat='CZ', psc='101 00', ulice='Krátká 5', nazev='První ZŠ', je_zs=True, je_ss=False) @@ -97,7 +97,7 @@ def gen_prijemci(osoby, kolik=10): prijemci = [] for i in range(kolik): rand_os = rnd.choice(osoby) - prijemci.add(Prijemce.objects.create(osoba=rand_os)) + prijemci.append(Prijemce.objects.create(osoba=rand_os)) return prijemci def gen_organizatori(osoby, last_rocnik): @@ -106,7 +106,7 @@ def gen_organizatori(osoby, last_rocnik): rand = rnd.randint(0, 8) if (rand % 8 == 0): pusobnost = rnd.randint(1, last_rocnik) - od = 1993 + last_rocnik - pusobnost + od = 1993 + pusobnost do = od + rnd.randint(1, 6) # aktualni organizatori jeste nemaji vyplnene organizuje_do #if do > datetime.datetime.now().year: