def test_buttons_case_3(self):
reviewer = Reviewer(FakeCollection(4, 1))
reviewer.init()
self.assertEqual(['again', 'hard', 'good', 'easy'], reviewer.buttons())
def test_buttons_case_4(self):
reviewer = Reviewer(FakeCollection(None, 1))
reviewer.init()
self.assertEqual([], reviewer.buttons())
def test_buttons(self):
reviewer = Reviewer(FakeCollection(2, 1))
reviewer.init()
self.assertEqual(['again', 'good'], reviewer.buttons())