def test_expressions_interactive_shows_prompt(self):
stream = StringIO.StringIO('\n')
outstream = StringIO.StringIO()
self.p = Parser(stream, outstream, '> ')
list(self.p.expressions())
outstream.seek(0)
self.assertEqual('> ', outstream.read())
def run_session(filename):
"""Run session test.
Output summary to stdout and return True if all test pass,
otherwise return False."""
env = Environment(namespace=Builtins.namespace())
num_tests, num_passed = 0, 0
with open(filename) as f:
input_expression = ""
# Iterate over non-blank lines
for line in ifilter(None, imap(str.strip, f)):
if line.startswith(';'):
continue
elif line.startswith('> '):
# Lines are stripped, so add a space to get whitespace
# between the lines.
input_expression += " " + line[2:]
elif input_expression:
# We got a line not starting with '> ', this is the
# expected result from the preceding input expression
num_tests += 1
parser = Parser(StringIO.StringIO(input_expression))
for expr in parser.expressions():
result = env.eval(expr)
result = repr(result) if result is not None else ''
expected = line.rstrip('\n')
if result != expected:
print 'Failure: "%s" != "%s"' % (result, expected)
else:
num_passed += 1
input_expression = ""
else:
# If we got a non-empty line without an input
# expression something is wrong
print 'Unexpected line: "%s"' % line
print "{:20} Ran {} tests, {} passed".format(os.path.basename(filename),
num_tests, num_passed)
return num_tests == num_passed
class test_parser(TestCase):
def list_expressions(self, inp):
stream = StringIO.StringIO(inp)
self.p = Parser(stream)
result = self.p.expressions()
return list(result)
def assert_expressions_results(self, inp, outp):
self.assertEqual(outp, self.list_expressions(inp)[0])
def assert_expressions_results_all(self, inp, outp):
self.assertEqual(outp, self.list_expressions(inp))
def assert_expressions_none(self, inp):
self.assertEqual(0, len(self.list_expressions(inp)))
def assert_expressions_throws(self, inp, error):
self.assertRaises(error, self.list_expressions, inp)
def test_expressions_empty(self):
self.assert_expressions_none('')
def test_expressions_symbol(self):
self.assert_expressions_results('symbol', E('symbol', E.NAME))
def test_expressions_symbol_with_whitespace(self):
self.assert_expressions_results(' symbol ', E('symbol', E.NAME))
def test_expressions_comment(self):
self.assert_expressions_results_all('42 ;43', [E(42, E.CONSTANT)])
def test_expressions_integer(self):
self.assert_expressions_results('42', E(42, E.CONSTANT))
def test_expressions_zero(self):
self.assert_expressions_results('0', E(0, E.CONSTANT))
def test_expressions_float(self):
self.assert_expressions_results('3.14', E(3.14, E.CONSTANT))
def test_expressions_string(self):
self.assert_expressions_results('"a b c"', E("a b c", E.CONSTANT))
def test_expressions_combination(self):
self.assert_expressions_results(
'(symbol 42 3.14)',
E([E('symbol', E.NAME),
E(42, E.CONSTANT),
E(3.14, E.CONSTANT)],
E.COMBINATION))
def test_expressions_nested_combination(self):
self.assert_expressions_results(
'(symbol (42 (3.14) 0))',
E([E('symbol', E.NAME),
E([E(42, E.CONSTANT),
E([E(3.14, E.CONSTANT)], E.COMBINATION),
E(0, E.CONSTANT)], E.COMBINATION)],
E.COMBINATION))
def test_expressions_complex(self):
self.assert_expressions_results(
'(a-b - + a+b .( a b c)(.9 o/ /))',
E([E('a-b', E.NAME),
E('-', E.NAME),
E('+', E.NAME),
E('a+b', E.NAME),
E('.', E.NAME),
E([E('a', E.NAME), E('b', E.NAME), E('c', E.NAME)],
E.COMBINATION),
E([E(.9, E.CONSTANT), E('o/', E.NAME), E('/', E.NAME)],
E.COMBINATION)],
E.COMBINATION))
def test_expressions_multiple(self):
self.assert_expressions_results_all(
'(a) b (c)',
[E([E('a', E.NAME)], E.COMBINATION),
E('b', E.NAME),
E([E('c', E.NAME)], E.COMBINATION)])
def test_expressions_unmatched_parantheses_throws_error(self):
self.assert_expressions_throws(')', ParseError)
def test_expressions_uncomplete_throws_error(self):
self.assert_expressions_throws('(+ 1', ParseError)
def test_expressions_interactive_shows_prompt(self):
stream = StringIO.StringIO('\n')
outstream = StringIO.StringIO()
self.p = Parser(stream, outstream, '> ')
list(self.p.expressions())
outstream.seek(0)
self.assertEqual('> ', outstream.read())
def test_expressions_interactive_shows_secondary_prompt(self):
stream = StringIO.StringIO('(\n)')
outstream = StringIO.StringIO()
self.p = Parser(stream, outstream, '> ', '. ')
list(self.p.expressions())
outstream.seek(0)
self.assertEqual('. ', outstream.read())
def test_from_string(self):
result = Parser.from_string('a 1')
self.assertEqual([E('a', E.NAME), E(1, E.CONSTANT)], result)
def list_expressions(self, inp):
stream = StringIO.StringIO(inp)
self.p = Parser(stream)
result = self.p.expressions()
return list(result)
def test_from_string(self):
result = Parser.from_string('a 1')
self.assertEqual([E('a', E.NAME), E(1, E.CONSTANT)], result)
