def test_generations(self): group = GroupFactory() idols = [IdolFactory() for i in xrange(3)] [ MembershipFactory( group=group, idol=idols[i], generation=i + 1 ) for i in xrange(3) ] generations = group.generations() assert len(generations) == 3 for k, v in generations.iteritems(): assert isinstance(k, int) assert len(v) == 1
def test_generations_failure(self): group = GroupFactory() idols = [IdolFactory() for i in xrange(3)] [MembershipFactory(group=group, idol=idols[i]) for i in xrange(3)] assert not group.generations()