/
test_alsep.py
126 lines (109 loc) · 4.45 KB
/
test_alsep.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
# -*- coding: utf-8 -*-
import os
import unittest
from obspy.io.alsep.core import (_is_pse, _is_wtn, _is_wth,
_read_pse, _read_wtn, _read_wth)
class AlsepTestCase(unittest.TestCase):
def setUp(self):
# Directory where the test files are located
self.path = os.path.dirname(__file__)
def test_is_pse(self):
"""
Testing ALSEP PSE file format.
"""
testfile = os.path.join(self.path, 'data', 'pse.a15.1.2.mini')
self.assertEqual(_is_pse(testfile), True)
testfile = os.path.join(self.path, 'data', 'wtn.1.2.mini')
self.assertEqual(_is_pse(testfile), False)
testfile = os.path.join(self.path, 'data', 'wth.1.5.mini')
self.assertEqual(_is_pse(testfile), False)
def test_is_wtn(self):
"""
Testing ALSEP WTN file format.
"""
testfile = os.path.join(self.path, 'data', 'pse.a15.1.2.mini')
self.assertEqual(_is_wtn(testfile), False)
testfile = os.path.join(self.path, 'data', 'wtn.1.2.mini')
self.assertEqual(_is_wtn(testfile), True)
testfile = os.path.join(self.path, 'data', 'wth.1.5.mini')
self.assertEqual(_is_wtn(testfile), False)
def test_is_wth(self):
"""
Testing ALSEP WTH file format.
"""
testfile = os.path.join(self.path, 'data', 'pse.a15.1.2.mini')
self.assertEqual(_is_wth(testfile), False)
testfile = os.path.join(self.path, 'data', 'wtn.1.2.mini')
self.assertEqual(_is_wth(testfile), False)
testfile = os.path.join(self.path, 'data', 'wth.1.5.mini')
self.assertEqual(_is_wth(testfile), True)
def test_read_alsep_pse_file(self):
"""
Read ALSEP PSE file test via obspy.core.alsep._read.
"""
testfile = os.path.join(self.path, 'data', 'pse.a15.1.2.mini')
stream = _read_pse(testfile)
self.assertEqual(10, len(stream.traces))
def test_read_alsep_pse_file_with_ignore_error(self):
"""
Read ALSEP PSE file test via obspy.core.alsep._read.
"""
testfile = os.path.join(self.path, 'data', 'pse.a15.1.2.mini')
stream = _read_pse(testfile, ignore_error=True)
self.assertEqual(4654, len(stream.traces))
def test_read_alsep_wtn_file(self):
"""
Read ALSEP WTN file test via obspy.core.alsep._read.
"""
testfile = os.path.join(self.path, 'data', 'wtn.1.2.mini')
stream = _read_wtn(testfile)
self.assertEqual(27, len(stream.traces))
def test_read_alsep_wth_file(self):
"""
Read ALSEP WTH file test via obspy.core.alsep._read.
"""
testfile = os.path.join(self.path, 'data', 'wth.1.5.mini')
stream = _read_wth(testfile)
self.assertEqual(12, len(stream.traces))
def test_single_header_wtn(self):
"""
Read single header WTN file test
"""
testfile = os.path.join(self.path, 'data', 'wtn.6.30.mini')
stream = _read_wtn(testfile)
self.assertEqual(18, len(stream.traces))
def test_single_header_wth(self):
"""
Read single header WTH file test
"""
testfile = os.path.join(self.path, 'data', 'wth.5.6.mini')
stream = _read_wth(testfile)
st_geophone1 = stream.select(id='XA.S17..GP1')
self.assertEqual(3, len(st_geophone1))
def test_pse_new_format(self):
"""
Read PSE new format which does not have Apollo 12 SPZ
"""
testfile = os.path.join(self.path, 'data', 'pse.a12.6.117.mini')
stream = _read_pse(testfile)
st_spz = stream.select(id='XA.S12..SPZ')
self.assertEqual(0, len(st_spz))
def test_frame_loss(self):
"""
Check frame with many time skipping
"""
testfile = os.path.join(self.path, 'data', 'pse.a14.4.171.mini')
stream = _read_pse(testfile)
st_lpx = stream.select(id='XA.S14..LPX')
self.assertEqual(1, len(st_lpx))
def test_pse_read_year_option(self):
"""
Read pse data with year option to overwrite year
"""
testfile = os.path.join(self.path, 'data', 'pse.a12.10.91.mini')
stream = _read_pse(testfile)
st_lpx = stream.select(id='XA.S12..LPX')
self.assertEqual(1976, st_lpx[0].times("utcdatetime")[0].year)
stream = _read_pse(testfile, year=1975)
st_lpx = stream.select(id='XA.S12..LPX')
self.assertEqual(1975, st_lpx[0].times("utcdatetime")[0].year)