/usr/lib/python2.7/dist-packages/PySPH-1.0a4.dev0-py2.7-linux-x86_64.egg/pysph/examples/tests/test_examples.py is in python-pysph 0~20160514.git91867dc-4build1.
This file is owned by root:root, with mode 0o644.
The actual contents of the file can be viewed below.
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 | import os
import tempfile
import shutil
import subprocess
import sys
from nose.plugins.attrib import attr
from pysph.examples import run
_orig_ets_toolkit = None
def setup_module():
# Set the ETS_TOOLKIT to null to avoid errors when importing TVTK.
global _orig_ets_toolkit
var = 'ETS_TOOLKIT'
_orig_ets_toolkit = os.environ.get(var)
os.environ[var] = 'null'
def teardown_module():
var = 'ETS_TOOLKIT'
if _orig_ets_toolkit is None:
del os.environ[var]
else:
os.environ[var] = _orig_ets_toolkit
def run_example(module):
"""This simply runs the example to make sure that the example executes
correctly. It wipes out the generated output directory.
"""
out_dir = tempfile.mkdtemp()
cmd = [sys.executable, "-m", module, "--max-steps", "1",
"--disable-output", "-q", "-d", out_dir]
env_vars = dict(os.environ)
env_vars['ETS_TOOLKIT'] = 'null'
try:
subprocess.check_output(cmd, env=env_vars)
finally:
shutil.rmtree(out_dir)
def _has_tvtk():
try:
from tvtk.api import tvtk
except ImportError:
return False
else:
return True
@attr(slow=True)
def test_example_should_run():
for module, doc in run.get_all_examples():
if module == 'pysph.examples.rigid_body.dam_break3D_sph' and \
not _has_tvtk():
continue
yield run_example, module
|