def test_get_answer_none(self): reviewer = Reviewer(FakeCollection(None, 1)) reviewer.init() self.assertEqual(None, reviewer.get_answer())
def test_get_answer_some(self): reviewer = Reviewer(FakeCollection(3, 1)) reviewer.init() self.assertEqual('fake answer', reviewer.get_answer())