You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
137 lines
3.2 KiB
137 lines
3.2 KiB
# python 2/3
|
|
from __future__ import division, print_function, absolute_import
|
|
import sys
|
|
import unittest
|
|
import os
|
|
import glob
|
|
try:
|
|
from urllib2 import HTTPError
|
|
except ImportError:
|
|
from urllib.error import HTTPError
|
|
|
|
from .. import common
|
|
|
|
from pyrocko import ExternalProgramMissing
|
|
|
|
from pyrocko import util
|
|
from pyrocko import example
|
|
from pyrocko import pile
|
|
from pyrocko.plot import gmtpy
|
|
|
|
from pyrocko.gui import snuffler
|
|
from pyrocko.dataset import topo
|
|
|
|
|
|
op = os.path
|
|
|
|
test_dir = op.dirname(op.abspath(__file__))
|
|
|
|
|
|
skip_examples = [
|
|
'trace_restitution_dseed.py',
|
|
'gf_forward_viscoelastic.py'
|
|
]
|
|
|
|
|
|
def tutorial_run_dir():
|
|
return op.join(test_dir, '..', 'example_run_dir')
|
|
|
|
|
|
def noop(*args, **kwargs):
|
|
pass
|
|
|
|
|
|
class ExamplesTestCase(unittest.TestCase):
|
|
|
|
@classmethod
|
|
def setUpClass(cls):
|
|
from matplotlib import pyplot as plt
|
|
|
|
cls.cwd = os.getcwd()
|
|
cls.run_dir = tutorial_run_dir()
|
|
util.ensuredir(cls.run_dir)
|
|
cls.dn = open(os.devnull, 'w')
|
|
sys.stdout = cls.dn
|
|
os.chdir(cls.run_dir)
|
|
|
|
plt.show_orig_testex = plt.show
|
|
plt.show = noop
|
|
|
|
snuffler.snuffle_orig_testex = snuffler.snuffle
|
|
snuffler.snuffle = noop
|
|
|
|
cls._show_progress_force_off_orig = pile.show_progress_force_off
|
|
pile.show_progress_force_off = True
|
|
|
|
@classmethod
|
|
def tearDownClass(cls):
|
|
from matplotlib import pyplot as plt
|
|
|
|
cls.dn.close()
|
|
sys.stdout = sys.__stdout__
|
|
os.chdir(cls.cwd)
|
|
|
|
snuffler.snuffle = snuffler.snuffle_orig_testex
|
|
plt.show = plt.show_orig_testex
|
|
pile.show_progress_force_off = cls._show_progress_force_off_orig
|
|
|
|
|
|
example_files = [fn for fn in glob.glob(op.join(test_dir, 'examples', '*.py'))
|
|
if os.path.basename(fn) not in skip_examples]
|
|
|
|
|
|
def _make_function(test_name, fn):
|
|
def f(self):
|
|
imp = imp2 = None
|
|
try:
|
|
import imp
|
|
|
|
except ImportError:
|
|
import importlib.machinery as imp2
|
|
|
|
try:
|
|
if imp:
|
|
imp.load_source(test_name, fn)
|
|
else:
|
|
imp2.SourceFileLoader(test_dir, fn)
|
|
|
|
except example.util.DownloadError:
|
|
raise unittest.SkipTest('could not download required data file')
|
|
|
|
except HTTPError as e:
|
|
raise unittest.SkipTest('skipped due to %s: "%s"' % (
|
|
e.__class__.__name__, str(e)))
|
|
|
|
except ExternalProgramMissing as e:
|
|
raise unittest.SkipTest(str(e))
|
|
|
|
except ImportError as e:
|
|
raise unittest.SkipTest(str(e))
|
|
|
|
except topo.AuthenticationRequired:
|
|
raise unittest.SkipTest(
|
|
'cannot download topo data (no auth credentials)')
|
|
|
|
except gmtpy.GMTInstallationProblem:
|
|
raise unittest.SkipTest('GMT not installed or not usable')
|
|
|
|
except Exception as e:
|
|
raise e
|
|
|
|
f.__name__ = 'test_example_' + test_name
|
|
|
|
return f
|
|
|
|
|
|
for fn in sorted(example_files):
|
|
test_name = op.splitext(op.split(fn)[-1])[0]
|
|
setattr(
|
|
ExamplesTestCase,
|
|
'test_example_' + test_name,
|
|
_make_function(test_name, fn))
|
|
|
|
|
|
if __name__ == '__main__':
|
|
util.setup_logging('test_examples', 'warning')
|
|
common.matplotlib_use_agg()
|
|
unittest.main()
|