def test_consume_existing(self): """Test the consume method when the token exists.""" parser = SimpleParser("Hello, World!") self.assertEqual(parser.consume("\w+"), "Hello") self.assertEqual(parser.consume(", "), ", ") self.assertEqual(parser.consume("\w+"), "World") self.assertEqual(parser.consume("!"), "!")
def test_consume_missing(self): """Test the consume method when the token does not exist.""" parser = SimpleParser("Hello, World!") with self.assertRaises(Exception): parser.consume("\d+")
def consume(self, regexp): token = SimpleParser.consume(self, regexp) self.buffer = self.buffer.lstrip() return token