-
Notifications
You must be signed in to change notification settings - Fork 529
/
test_beachball.py
328 lines (299 loc) · 12.1 KB
/
test_beachball.py
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
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
144
145
146
147
148
149
150
151
152
153
154
155
156
157
158
159
160
161
162
163
164
165
166
167
168
169
170
171
172
173
174
175
176
177
178
179
180
181
182
183
184
185
186
187
188
189
190
191
192
193
194
195
196
197
198
199
200
201
202
203
204
205
206
207
208
209
210
211
212
213
214
215
216
217
218
219
220
221
222
223
224
225
226
227
228
229
230
231
232
233
234
235
236
237
238
239
240
241
242
243
244
245
246
247
248
249
250
251
252
253
254
255
256
257
258
259
260
261
262
263
264
265
266
267
268
269
270
271
272
273
274
275
276
277
278
279
280
281
282
283
284
285
286
287
288
289
290
291
292
293
294
295
296
297
298
299
300
301
302
303
304
305
306
307
308
309
310
311
312
313
314
315
316
317
318
319
320
321
322
323
324
325
326
327
328
# -*- coding: utf-8 -*-
"""
The obspy.imaging.beachball test suite.
"""
import warnings
import matplotlib.pyplot as plt
import pytest
from obspy.core.util.base import NamedTemporaryFile
from obspy.core.util.testing import WarningsCapture
from obspy.imaging.beachball import (tdl, aux_plane, beach, beachball,
MomentTensor, mt2axes, mt2plane,
strike_dip)
class TestBeachballPlot:
"""
Test cases for beachball generation.
"""
def test_beachball(self, image_path):
"""
Create beachball examples in tests/output directory.
"""
# https://en.wikipedia.org/wiki/File:USGS_sumatra_mts.gif
data = [[0.91, -0.89, -0.02, 1.78, -1.55, 0.47],
[274, 13, 55],
[130, 79, 98],
[264.98, 45.00, -159.99],
[160.55, 76.00, -46.78],
[1.45, -6.60, 5.14, -2.67, -3.16, 1.36],
[235, 80, 35],
[138, 56, 168],
# Explosion
[1, 1, 1, 0, 0, 0],
# Implosion
[-1, -1, -1, 0, 0, 0],
# CLVD - Compensate Linear Vector Dipole
[1, -2, 1, 0, 0, 0],
# Double Couple
[1, -1, 0, 0, 0, 0],
# Lars
[1, -1, 0, 0, 0, -1],
# https://wwweic.eri.u-tokyo.ac.jp/yuji/Aki-nada/
[179, 55, -78],
[10, 42.5, 90],
[10, 42.5, 92],
# https://wwweic.eri.u-tokyo.ac.jp/yuji/tottori/
[150, 87, 1],
# http://iisee.kenken.go.jp/staff/thara/2004/09/20040905_1/
# 2nd.html
[0.99, -2.00, 1.01, 0.92, 0.48, 0.15],
# http://iisee.kenken.go.jp/staff/thara/2004/09/20040905_0/
# 1st.html
[5.24, -6.77, 1.53, 0.81, 1.49, -0.05],
# http://iisee.kenken.go.jp/staff/thara/miyagi.htm
[16.578, -7.987, -8.592, -5.515, -29.732, 7.517],
# http://iisee.kenken.go.jp/staff/thara/20050613/chile.html
[-2.39, 1.04, 1.35, 0.57, -2.94, -0.94],
]
filenames = ['bb_sumatra_mt.png', 'bb_sumatra_np1.png',
'bb_sumatra_np2.png', 'bb_19950128_np1.png',
'bb_19950128_np2.png', 'bb_20090102_mt.png',
'bb_20090102_np1.png', 'bb-20090102-np2.png',
'bb_explosion.png', 'bb_implosion.png', 'bb_clvd.png',
'bb_double_couple.png', 'bb_lars.png', 'bb_geiyo_np1.png',
'bb_honshu_np1.png', 'bb_honshu_np2.png',
'bb_tottori_np1.png', 'bb_20040905_1_mt.png',
'bb_20040905_0_mt.png', 'bb_miyagi_mt.png',
'bb_chile_mt.png',
]
for data_, filename in zip(data, filenames):
file_name = image_path.parent / filename
beachball(data_, outfile=file_name)
plt.close()
def test_beachball_output_format(self):
"""
Tests various output formats.
"""
fm = [115, 35, 50]
# PDF - Some matplotlib versions internally raise some warnings here
# which we don't want to see in the tests.
with warnings.catch_warnings():
warnings.simplefilter("ignore")
data = beachball(fm, format='pdf')
assert data[0:4] == b"%PDF"
# as file
# create and compare image
with NamedTemporaryFile(suffix='.pdf') as tf:
beachball(fm, format='pdf', outfile=tf.name)
# PS
data = beachball(fm, format='ps')
assert data[0:4] == b"%!PS"
# as file
with NamedTemporaryFile(suffix='.ps') as tf:
beachball(fm, format='ps', outfile=tf.name)
# PNG
data = beachball(fm, format='png')
assert data[1:4] == b"PNG"
# as file
with NamedTemporaryFile(suffix='.png') as tf:
beachball(fm, format='png', outfile=tf.name)
# SVG
data = beachball(fm, format='svg')
assert data[0:5] == b"<?xml"
# as file
with NamedTemporaryFile(suffix='.svg') as tf:
beachball(fm, format='svg', outfile=tf.name)
def test_strike_dip(self):
"""
Test strike_dip function - all values are taken from MatLab.
"""
sl1 = -0.048901208623019
sl2 = 0.178067035725425
sl3 = 0.982802524713469
(strike, dip) = strike_dip(sl2, sl1, sl3)
assert round(abs(strike-254.64386091007400), 7) == 0
assert round(abs(dip-10.641291652406172), 7) == 0
def test_aux_plane(self):
"""
Test aux_plane function - all values are taken from MatLab.
"""
# https://en.wikipedia.org/wiki/File:USGS_sumatra_mts.gif
s1 = 132.18005257215460
d1 = 84.240987194376590
r1 = 98.963372641038790
(s2, d2, r2) = aux_plane(s1, d1, r1)
assert round(abs(s2-254.64386091007400), 7) == 0
assert round(abs(d2-10.641291652406172), 7) == 0
assert round(abs(r2-32.915578422454380), 7) == 0
#
s1 = 160.55
d1 = 76.00
r1 = -46.78
(s2, d2, r2) = aux_plane(s1, d1, r1)
assert round(abs(s2-264.98676854650216), 7) == 0
assert round(abs(d2-45.001906942415623), 7) == 0
assert round(abs(r2--159.99404307049076), 7) == 0
def test_aux_plane_735(self):
"""
Test aux_plane precision issue #735
"""
s, d, r = aux_plane(164, 90, -32)
assert round(abs(s-254.), 7) == 0
assert round(abs(d-58.), 7) == 0
assert round(abs(r--180.), 7) == 0
def test_tdl(self):
"""
Test tdl function - all values are taken from MatLab.
"""
an = [0.737298200871146, -0.668073596186761, -0.100344571703004]
bn = [-0.178067035261159, -0.048901208638715, -0.982802524796805]
(ft, fd, fl) = tdl(an, bn)
assert round(abs(ft-227.81994742784540), 7) == 0
assert round(abs(fd-84.240987194376590), 7) == 0
assert round(abs(fl-81.036627358961210), 7) == 0
def test_mt2plane(self):
"""
Tests mt2plane.
"""
mt = MomentTensor((0.91, -0.89, -0.02, 1.78, -1.55, 0.47), 0)
np = mt2plane(mt)
assert round(abs(np.strike-129.86262672080011), 7) == 0
assert round(abs(np.dip-79.022700906654734), 7) == 0
assert round(abs(np.rake-97.769255185515192), 7) == 0
def test_mt2axes(self):
"""
Tests mt2axes.
"""
# https://en.wikipedia.org/wiki/File:USGS_sumatra_mts.gif
mt = MomentTensor((0.91, -0.89, -0.02, 1.78, -1.55, 0.47), 0)
(t, n, p) = mt2axes(mt)
assert round(abs(t.val-2.52461359), 7) == 0
assert round(abs(t.dip-55.33018576), 7) == 0
assert round(abs(t.strike-49.53656116), 7) == 0
assert round(abs(n.val-0.08745048), 7) == 0
assert round(abs(n.dip-7.62624529), 7) == 0
assert round(abs(n.strike-308.37440488), 7) == 0
assert round(abs(p.val--2.61206406), 7) == 0
assert round(abs(p.dip-33.5833323), 7) == 0
assert round(abs(p.strike-213.273886), 7) == 0
def test_collection(self, image_path):
"""
Tests to plot beachballs as collection into an existing axis
object. The moment tensor values are taken form the
test_Beachball unit test. See that test for more information about
the parameters.
"""
mt = [[0.91, -0.89, -0.02, 1.78, -1.55, 0.47],
[274, 13, 55],
[130, 79, 98],
[264.98, 45.00, -159.99],
[160.55, 76.00, -46.78],
[1.45, -6.60, 5.14, -2.67, -3.16, 1.36],
[235, 80, 35],
[138, 56, 168],
[1, 1, 1, 0, 0, 0],
[-1, -1, -1, 0, 0, 0],
[1, -2, 1, 0, 0, 0],
[1, -1, 0, 0, 0, 0],
[1, -1, 0, 0, 0, -1],
[179, 55, -78],
[10, 42.5, 90],
[10, 42.5, 92],
[150, 87, 1],
[0.99, -2.00, 1.01, 0.92, 0.48, 0.15],
[5.24, -6.77, 1.53, 0.81, 1.49, -0.05],
[16.578, -7.987, -8.592, -5.515, -29.732, 7.517],
[-2.39, 1.04, 1.35, 0.57, -2.94, -0.94],
[150, 87, 1]]
# Initialize figure
fig = plt.figure(figsize=(6, 6), dpi=300)
ax = fig.add_subplot(111, aspect='equal')
# Plot the stations or borders
ax.plot([-100, -100, 100, 100], [-100, 100, -100, 100], 'rv')
x = -100
y = -100
for i, t in enumerate(mt):
# add the beachball (a collection of two patches) to the axis
ax.add_collection(beach(t, width=30, xy=(x, y), linewidth=.6))
x += 50
if (i + 1) % 5 == 0:
x = -100
y += 50
# set the x and y limits
ax.axis([-120, 120, -120, 120])
# save the output
fig.savefig(image_path)
def collection_aspect(self, axis, image_path):
"""
Common part of the test_collection_aspect_[xy] tests.
"""
mt = [0.91, -0.89, -0.02, 1.78, -1.55, 0.47]
filename_width = image_path.parent / 'width.png'
# Test passing only a width
# Initialize figure
fig = plt.figure()
ax = fig.add_subplot(111)
# add the beachball (a collection of two patches) to the axis
# give it an axes to keep make the beachballs circular
# even though axes are not scaled
ax.add_collection(beach(mt, width=400, xy=(0, 0), linewidth=.6,
axes=ax))
# set the x and y limits
ax.axis(axis)
fig.savefig(filename_width)
# Test passing a width and a height
filename_height = image_path.parent / 'height.png'
# Initialize figure
fig = plt.figure()
ax = fig.add_subplot(111)
# add the beachball (a collection of two patches) to the axis
# give it an axes to keep make the beachballs circular
# even though axes are not scaled
ax.add_collection(beach(mt, width=(400, 200), xy=(0, 0),
linewidth=.6, axes=ax))
# set the x and y limits
ax.axis(axis)
# save the output
fig.savefig(filename_height)
def test_collection_aspect_x(self, image_path):
"""
Tests to plot beachball into a non-scaled axes with an x-axis larger
than y-axis. Use the 'axes' kwarg to make beachballs circular.
"""
self.collection_aspect(axis=[-10000, 10000, -100, 100],
image_path=image_path)
def test_collection_aspect_y(self, image_path):
"""
Tests to plot beachball into a non-scaled axes with a y-axis larger
than x-axis. Use the 'axes' kwarg to make beachballs circular.
"""
self.collection_aspect(axis=[-100, 100, -10000, 10000],
image_path=image_path)
def test_mopad_fallback(self, image_path):
"""
Test the fallback to mopad.
"""
mt = [0.000, -1.232e25, 1.233e25, 0.141e25, -0.421e25, 2.531e25]
with WarningsCapture() as w:
# Always raise warning.
beachball(mt, outfile=image_path)
# Make sure the appropriate warnings has been raised.
assert w
# Filter
w = [_i.message.args[0] for _i in w]
w = [_i for _i in w
if "falling back to the mopad wrapper" in _i.lower()]
assert w
@pytest.mark.image_path_suffix('pdf')
def test_beach_with_axes_to_pdf(self, image_path):
"""
Regression test for #2887
Tests using `beach` with passing in an `Axes` object and saving to pdf
"""
fig = plt.figure(figsize=(6, 4))
ax = fig.add_subplot(111, aspect='equal')
bb = beach((0, 45, 90), facecolor='k', width=120, axes=ax, xy=(0, 0))
ax.add_collection(bb)
ax.set_xlim(-2, 4)
ax.set_ylim(-1, 2)
ax.grid()
fig.savefig(image_path)