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())
