def test_reset(self): code = Code(self.test_product_1) self.assertEqual(code.highlight_number, "1") code.get_next_number() self.assertEqual(code.highlight_number, "2") code.reset() code.get_next_number() self.assertEqual(code.highlight_number, "2")
def test_get_next_number(self): code = Code(self.test_product_2) self.assertEqual(code.highlight_number, "2") code.get_next_number() self.assertEqual(code.highlight_number, "3") code.get_next_number() self.assertEqual(code.highlight_number, "4") code.get_next_number() self.assertEqual(code.highlight_number, None)