def pytest_funcarg__doctest(request):
from textwrap import dedent
from pycket.test.testhelper import check_equal, execute
from pycket.error import SchemeException
assert request.function.__doc__ is not None
code = dedent(request.function.__doc__)
lines = [lin for lin in code.splitlines() if lin]
setup = []
exprs = []
expect = []
errors = []
current_let = []
setup_done = False
for line in lines:
if line[0] == ";":
# skip comments
continue
if not line.strip():
continue
elif line[0] == "!":
if setup_done:
raise RuntimeError(
"Don't use ! in midst of other doctest cmds")
setup.append(line[2:])
elif line[0] == ">":
setup_done = True
current_let.append(line[2:])
elif line[0] == "E":
errors.append(line[1:])
elif line[0] in " \t":
current_let[-1] += "\n" + line[2:]
else:
exprs.append(current_let[0] if len(current_let) ==
1 else current_let)
current_let = []
expect.append(line)
pairs = []
for pair in zip(exprs, expect):
pairs.extend(pair)
check_equal(*pairs, extra="\n".join(setup))
for error in errors:
with pytest.raises(Exception):
execute(error, extra="\n".join(setup))
return True
def pytest_funcarg__doctest(request):
from textwrap import dedent
from pycket.test.testhelper import check_equal, execute
from pycket.error import SchemeException
assert request.function.__doc__ is not None
code = dedent(request.function.__doc__)
lines = [lin for lin in code.splitlines() if lin]
setup = []
exprs = []
expect = []
errors = []
current_let = []
setup_done = False
for line in lines:
if line[0] == ";":
# skip comments
continue
if not line.strip():
continue
elif line[0] == "!":
if setup_done:
raise RuntimeError("Don't use ! in midst of other doctest cmds")
setup.append(line[2:])
elif line[0] == ">":
setup_done = True
current_let.append(line[2:])
elif line[0] == "E":
errors.append(line[1:])
elif line[0] in " \t":
current_let[-1] += "\n" + line[2:]
else:
exprs.append(current_let[0] if len(current_let) == 1 else current_let)
current_let = []
expect.append(line)
pairs = []
for pair in zip(exprs,expect):
pairs.extend(pair)
check_equal(*pairs, extra="\n".join(setup))
for error in errors:
with pytest.raises(SchemeException):
execute(error, extra="\n".join(setup))
return True