def test_remaining_none(self): reviewer = Reviewer(FakeCollection(None, 0)) reviewer.init() self.assertEqual({'new': 1, 'now': None, 'learning': 2, 'to_review': 3}, reviewer.remaining())
def test_remaining_to_review(self): reviewer = Reviewer(FakeCollection(2, 2)) reviewer.init() self.assertEqual({'new': 1, 'now': 'to_review', 'learning': 2, 'to_review': 3}, reviewer.remaining())