jc211
/
tractor
forked from goodboy/tractor
1
0
Fork 0
Code Pull Requests Activity

Search for guard and strip instead of hardcoding

init_sphinx_docs
Tyler Goodlet 2020-02-10 12:59:44 -05:00
parent cfc97c4204
commit d62610c44e
1 changed files with 8 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
tests/test_docs_examples.py
View File

@ -2,6 +2,7 @@
Let's make sure them docs work yah?
"""
from contextlib import contextmanager
import itertools
import os
import sys
import subprocess
@ -42,9 +43,13 @@ def run_example_in_subproc(loglevel, testdir, arb_addr):
os.path.join(str(testdir), '__main__.py')
)
# drop the ``if __name__ == '__main__'`` guard from the *NIX
# version of each script
script_code = '\n'.join(script_code.splitlines()[:-4])
# drop the ``if __name__ == '__main__'`` guard onwards from
# the *NIX version of each script
windows_script_lines = itertools.takewhile(
lambda line: "if __name__ ==" not in line,
script_code.splitlines()
)
script_code = '\n'.join(windows_script_lines)
script_file = testdir.makefile('.py', script_code)
# without this, tests hang on windows forever