from tempfile import mkdtemp
from .common import Benchmark
from pyrocko import gf, util, cake
from pyrocko.fomosto import qseis, psgrn_pscmp
random = num.random
logger = logging.getLogger('pyrocko.test.test_gf_static')
benchmark = Benchmark()
r2d = 180. / math.pi
d2r = 1.0 / r2d
km = 1000.
@unittest.skipUnless(qseis.have_backend(), 'backend qseis not available')
@unittest.skipUnless(psgrn_pscmp.have_backend(),
'backend psgrn_pscmp not available')
class GFStaticTest(unittest.TestCase):
tempdirs = []
def __init__(self, *args, **kwargs):
self._dummy_store = None
self.qseis_store_dir = None
self.pscmp_store_dir = None
unittest.TestCase.__init__(self, *args, **kwargs)
@classmethod
def tearDownClass(cls):
for d in cls.tempdirs:
shutil.rmtree(d)
north_shift=north,
east_shift=east,
depth=starget.depth)
values = store.statics(
dsource, receiver, starget.components,
interpolation=starget.interpolation)
for icomponent, value in enumerate(values):
out[i, icomponent] = value * sfactor
return out
@unittest.skipUnless(
psgrn_pscmp.have_backend(), 'backend psgrn_pscmp not available')
class GFPsgrnPscmpTestCase(unittest.TestCase):
tempdirs = []
def __init__(self, *args, **kwargs):
unittest.TestCase.__init__(self, *args, **kwargs)
@classmethod
def tearDownClass(cls):
for d in cls.tempdirs:
shutil.rmtree(d)
def test_fomosto_vs_psgrn_pscmp(self):
mod = cake.LayeredModel.from_scanlines(cake.read_nd_model_str('''
from pyrocko.fomosto import qseis, psgrn_pscmp
random = num.random
logger = logging.getLogger('pyrocko.test.test_gf_static')
benchmark = Benchmark()
r2d = 180. / math.pi
d2r = 1.0 / r2d
km = 1000.
@unittest.skipUnless(
qseis.have_backend(), 'backend qseis not available')
@unittest.skipUnless(
psgrn_pscmp.have_backend(), 'backend psgrn_pscmp not available')
class GFStaticTest(unittest.TestCase):
tempdirs = []
def __init__(self, *args, **kwargs):
self._dummy_store = None
self.qseis_store_dir = None
self.pscmp_store_dir = None
unittest.TestCase.__init__(self, *args, **kwargs)
@classmethod
def tearDownClass(cls):
for d in cls.tempdirs:
shutil.rmtree(d)
def get_qseis_store_dir(self):