def test_get_question_none(self): reviewer = Reviewer(FakeCollection(None, 1)) reviewer.init() self.assertEqual(None, reviewer.get_question())
def test_get_question_some(self): reviewer = Reviewer(FakeCollection(3, 1)) reviewer.init() self.assertEqual('fake question', reviewer.get_question())