def test_match(self):
inp = StringIO('mo')
cradle.init(inp)
self.assertEqual(cradle.Peek, 'm')
cradle.match('m')
self.assertEqual(cradle.Peek, 'o')
with self.assertRaises(SystemExit):
cradle.match('n')
self.assertEqual(self.stderr.getvalue(), "\n'n' expected.\n")
def test_get_number(self):
inp = StringIO('12z')
cradle.init(inp)
self.assertEqual(cradle.get_number(), '1')
self.assertEqual(cradle.get_number(), '2')
with self.assertRaises(SystemExit):
cradle.get_number()
self.assertEqual(self.stderr.getvalue(), "\n'Number' expected.\n")
def test_get_word(self):
inp = StringIO('ab1')
cradle.init(inp)
self.assertEqual(cradle.get_word(), 'a')
self.assertEqual(cradle.get_word(), 'b')
with self.assertRaises(SystemExit):
cradle.get_word()
self.assertEqual(self.stderr.getvalue(), "\n'Word' expected.\n")
def test_get_char(self):
inp = StringIO('abc')
cradle.init(inp)
self.assertEqual(cradle.get_char(), 'a')
self.assertEqual(cradle.get_char(), 'b')
self.assertEqual(cradle.get_char(), 'c')
