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()
