class PollTest(TestCase): def setUp(self): self.poll = Poll(title='test') self.poll.save() def test_add_question(self): self.poll.add_question('question', ['ans1', 'ans2']) self.assertEquals(1, self.poll.questions.all().count())