sapling/tests/heredoctest.py
Durham Goode 9440072474 checkcode: fix tests
Summary: Fix failures found by check-code.

Test Plan: Ran the tests

Reviewers: #sourcecontrol, ttung

Subscribers: ps

Differential Revision: https://phabricator.fb.com/D3221377
2016-04-26 13:00:31 -07:00

20 lines
502 B
Python

import sys
globalvars = {}
localvars = {}
lines = sys.stdin.readlines()
while lines:
l = lines.pop(0)
if l.startswith('SALT'):
print l[:-1]
elif l.startswith('>>> '):
snippet = l[4:]
while lines and lines[0].startswith('... '):
l = lines.pop(0)
snippet += "\n" + l[4:]
c = compile(snippet, '<heredoc>', 'single')
try:
exec c in globalvars, localvars
except Exception as inst:
print repr(inst)