def test_primary_membership_switching(self): idol = IdolFactory() membership1 = MembershipFactory(idol=idol, is_primary=True) assert idol.primary_membership == membership1 membership1.is_primary = False membership1.save() membership2 = MembershipFactory(idol=idol, is_primary=True) assert idol.primary_membership == membership2