-
Notifications
You must be signed in to change notification settings - Fork 2
/
Copy pathtest_track_check.py
271 lines (228 loc) · 10.7 KB
/
test_track_check.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
import unittest
import qc
import math
import Extended_IMMA as ex
from IMMA1 import IMMA
import numpy as np
import spherical_geometry as sph
import track_check as tc
from netCDF4 import Dataset
class TestTrackQCMethodsIncrementPosition(unittest.TestCase):
# increment_position(alat1,alat2,avs,ads,timdif)
def test_ship_heading_north_at_60knots_goes1degree_in_1hour(self):
"""
A ship travelling north at 60 knots will go 1 degree in 1 hours
"""
km_to_nm = 0.539957
for lat in range(-90, 90):
alat1 = lat
alon1 = lat
avs = 60.0 / km_to_nm
ads = 0.0
timdif = 2.0
alat2, alon2 = tc.increment_position(alat1, alon1, avs, ads, timdif)
self.assertAlmostEqual(alon2, 0, delta=0.001)
self.assertAlmostEqual(alat2, 1, delta=0.001)
def test_ship_heading_east_at_60knots_goes1degree_in_1hour(self):
"""
A ship travelling east at 60 knots will go 1 degree in 1 hour
"""
km_to_nm = 0.539957
alat1 = 0.0
alat2 = 0.0
avs = 60.0 / km_to_nm
ads = 90.0
timdif = 2.0
aud1, avd1 = tc.increment_position(alat1, alat2, avs, ads, timdif)
self.assertAlmostEqual(avd1, 1, delta=0.001)
self.assertAlmostEqual(aud1, 0, delta=0.001)
def test_ship_heading_east_at_60knots_at_latitude60_goes2degrees_in_1hour(self):
"""
A ship travelling east at 60 knots will go 2 degrees in 1 hour at 60N
"""
km_to_nm = 0.539957
alat = 60.0
alon = 0.0
avs = 60.0 / km_to_nm
ads = 90.0
timdif = 2.0
dlat, dlon = tc.increment_position(alat, alon, avs, ads, timdif)
distance = sph.sphere_distance(alat, alon, alat + dlat, alon + dlon) * km_to_nm
self.assertAlmostEqual(distance, 60, delta=0.0001)
def test_ship_goes_southwest(self):
km_to_nm = 0.539957
alat1 = 0.0
alat2 = 0.0
avs = 60.0 / km_to_nm
ads = 225.0
timdif = 2.0
aud1, avd1 = tc.increment_position(alat1, alat2, avs, ads, timdif)
self.assertAlmostEqual(avd1, -1. / np.sqrt(2), delta=0.001)
self.assertAlmostEqual(aud1, -1. / np.sqrt(2), delta=0.001)
class TestTrackQCMethodsModesp(unittest.TestCase):
def test_noinput(self):
m = tc.modesp([])
self.assertEqual(None, m)
def test_one_input(self):
m = tc.modesp([17.0])
self.assertEqual(None, m)
def test_single_speed_input_over8point5(self):
km_to_nm = 0.539957
speeds = [20.0 / km_to_nm, 20.0 / km_to_nm, 20.0 / km_to_nm, 20.0 / km_to_nm, 20.0 / km_to_nm, 20.0 / km_to_nm]
m = tc.modesp(speeds)
self.assertEqual(19.5 / km_to_nm, m)
def test_single_speed_input_under8point5(self):
km_to_nm = 0.539957
speeds = [2.0 / km_to_nm, 2.0 / km_to_nm, 2.0 / km_to_nm, 2.0 / km_to_nm, 2.0 / km_to_nm, 2.0 / km_to_nm]
m = tc.modesp(speeds)
self.assertEqual(8.5 / km_to_nm, m)
def test_single_speed_input_overmaximum(self):
km_to_nm = 0.539957
speeds = [200.0 / km_to_nm, 200.0 / km_to_nm, 200.0 / km_to_nm, 200.0 / km_to_nm, 200.0 / km_to_nm,
200.0 / km_to_nm]
m = tc.modesp(speeds)
self.assertEqual(34.5 / km_to_nm, m)
def test_one_of_each_speed_input_min_under8point5(self):
km_to_nm = 0.539957
speeds = []
for i in range(1, 20):
speeds.append(i / km_to_nm)
m = tc.modesp(speeds)
self.assertEqual(8.5 / km_to_nm, m)
class TestTrackQCMethodsDistr1(unittest.TestCase):
# rec = IMMA()
# v = {'ID':'SHIP1 ', 'YR':2001, 'MO':1, 'DY':1, 'HR':hour, 'LAT': hour*speed1/60., 'LON':0.0, 'DS':8, 'VS':4 }
# for key in v: rec.data[key] = v[key]
# self.trip1.add_report(ex.MarineReport(rec))
# shipid, lat, lon, sst, mat, year, month, day, hour, icoads_ds, icoads_vs, uid
def setUp(self):
self.km_to_nm = 0.539957
self.trip1 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 24):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip1.add_report(ex.MarineReport(rec))
self.trip2 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 3):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip2.add_report(ex.MarineReport(rec))
self.trip2.reps[1].setvar('LON', 1.0)
def test_first_entry_missing(self):
difference_from_estimated_location = tc.distr1(self.trip1)
self.assertEqual(difference_from_estimated_location[0], None)
def test_ship_is_at_computed_location(self):
difference_from_estimated_location = tc.distr1(self.trip1)
for i, diff in enumerate(difference_from_estimated_location):
if 0 < i < len(difference_from_estimated_location) - 1:
self.assertAlmostEqual(diff, 0, delta=0.00001)
def test_misplaced_ob_out_by_1degree_times_coslat(self):
difference_from_estimated_location = tc.distr1(self.trip2)
self.assertAlmostEqual(difference_from_estimated_location[1],
(2 * np.pi * 6371.0088) * np.cos(self.trip2.reps[1].lat() * np.pi / 180.) / 360.,
delta=0.00001)
class TestTrackQCMethodsDistr2(unittest.TestCase):
def setUp(self):
self.km_to_nm = 0.539957
self.trip1 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 24):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip1.add_report(ex.MarineReport(rec))
self.trip2 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 3):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip2.add_report(ex.MarineReport(rec))
self.trip2.reps[1].setvar('LON', 1.0)
def test_last_entry_missing(self):
difference_from_estimated_location = tc.distr2(self.trip1)
self.assertEqual(difference_from_estimated_location[-1], None)
def test_ship_is_at_computed_location(self):
difference_from_estimated_location = tc.distr2(self.trip1)
for i, diff in enumerate(difference_from_estimated_location):
if 0 < i < len(difference_from_estimated_location) - 1:
self.assertAlmostEqual(diff, 0, delta=0.00001)
def test_misplaced_ob_out_by_1degree_times_coslat(self):
difference_from_estimated_location = tc.distr2(self.trip2)
self.assertAlmostEqual(difference_from_estimated_location[1],
(2 * np.pi * 6371.0088) * np.cos(self.trip2.reps[1].lat() * np.pi / 180.) / 360.,
delta=0.00001)
class TestTrackQCMethodsMidpt(unittest.TestCase):
# midpoint_discrepancies = midpt(inreps,time_differences)
def setUp(self):
self.km_to_nm = 0.539957
self.trip1 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 24):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip1.add_report(ex.MarineReport(rec))
self.trip2 = ex.Voyage()
self.speed1 = 18. / self.km_to_nm
for hour in range(0, 3):
rec = IMMA()
v = {'ID': 'SHIP1 ', 'YR': 2001, 'MO': 1, 'DY': 1, 'HR': hour,
'LAT': hour * self.speed1 * 360. / (2 * np.pi * 6371.0088), 'LON': 0.0, 'DS': 8, 'VS': 4}
for key in v:
rec.data[key] = v[key]
self.trip2.add_report(ex.MarineReport(rec))
self.trip2.reps[1].setvar('LON', 1.0)
def test_first_and_last_are_missing(self):
midpoint_discrepancies = tc.midpt(self.trip1)
self.assertEqual(midpoint_discrepancies[0], None)
self.assertEqual(midpoint_discrepancies[-1], None)
def test_midpt_1_deg_error_out_by_60coslat(self):
midpoint_discrepancies = tc.midpt(self.trip2)
self.assertAlmostEqual(midpoint_discrepancies[1],
(2 * np.pi * 6371.0088) * math.cos(self.trip2.reps[1].lat() * np.pi / 180) / 360.,
delta=0.00001)
def test_midpt_at_computed_location(self):
midpoint_discrepancies = tc.midpt(self.trip1)
for i, pt in enumerate(midpoint_discrepancies):
if 0 < i < len(midpoint_discrepancies) - 1:
self.assertNotEqual(pt, None, 'Failed at ' + str(i) + ' out of ' + str(len(midpoint_discrepancies)))
self.assertAlmostEqual(pt, 0, msg='Failed at ' + str(i) + ' with mid point = ' + str(pt), delta=0.00001)
class TestTrackQCMethodsDirectionContinuity(unittest.TestCase):
# result = direction_continuity(dsi,dsi_previous,ship_directions)
def test_just_pass_and_just_fail(self):
for angle in [0, 45, 90, 135, 180, 225, 270, 315, 360]:
self.assertEqual(10, tc.direction_continuity(angle, angle, angle + 60.1))
self.assertEqual(0, tc.direction_continuity(angle, angle, angle + 59.9))
class TestTrackQCMethodsSpeedContinuity(unittest.TestCase):
def test_1(self):
km_to_nm = 0.539957
result = tc.speed_continuity(12, 12, 12 / km_to_nm)
self.assertEqual(0, result)
def test_just_fails(self):
km_to_nm = 0.539957
result = tc.speed_continuity(12, 12, (12 + 10.01) / km_to_nm)
self.assertEqual(10, result)
def test_just_passes(self):
km_to_nm = 0.539957
result = tc.speed_continuity(12, 12, (12 + 9.99) / km_to_nm)
self.assertEqual(0, result)
def test_input_speed_is_None(self):
result = tc.speed_continuity(12, 12, None)
self.assertEqual(0, result)
if __name__ == '__main__':
unittest.main()