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