def test_creation(self):
""" Create party, save, retrieve """
party = Party(name="Plaid Cymru", abbreviation="PC")
self.assertIsNone(party.id)
party.save()
self.assertIsNotNone(party.id)
self.assertIs(type(party.id), int)
party_again = Party.objects.get(id=party.id)
self.assertEqual(party.name, party_again.name)
self.assertEqual(party.abbreviation, party_again.abbreviation)
def test_creation(self):
""" Create answerset """
constituency = Constituency(name="Merthyr Tydfil and Rhymney")
constituency.save()
party = Party(name="Plaid Cymru", abbreviation="PC")
party.save()
answers = AnswerSet(
constituency=constituency,
going_to_vote="yes",
voting_for=party
)
answers.save()
