|
@@ 441-452 (lines=12) @@
|
| 438 |
|
for i, oo in enumerate(self.out): |
| 439 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 440 |
|
|
| 441 |
|
def test_get_aacgm_coord_arr_arr_mix(self): |
| 442 |
|
"""Test array AACGMV2 calculation for an array and floats""" |
| 443 |
|
self.out = aacgmv2.get_aacgm_coord_arr(np.array(self.lat_in), |
| 444 |
|
self.lon_in[0], self.alt_in[0], |
| 445 |
|
self.dtime, self.method) |
| 446 |
|
|
| 447 |
|
assert len(self.out) == len(self.ref) |
| 448 |
|
assert [isinstance(oo, list) and len(oo) == len(self.lat_in) |
| 449 |
|
for oo in self.out] |
| 450 |
|
|
| 451 |
|
for i, oo in enumerate(self.out): |
| 452 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 453 |
|
|
| 454 |
|
def test_get_aacgm_coord_arr_mult_failure(self): |
| 455 |
|
"""Test aacgm_coord_arr failure with multi-dim array input""" |
|
@@ 428-439 (lines=12) @@
|
| 425 |
|
for i, oo in enumerate(self.out): |
| 426 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 427 |
|
|
| 428 |
|
def test_get_aacgm_coord_arr_list_mix(self): |
| 429 |
|
"""Test array AACGMV2 calculation for a list and floats""" |
| 430 |
|
self.out = aacgmv2.get_aacgm_coord_arr(self.lat_in, self.lon_in[0], |
| 431 |
|
self.alt_in[0], self.dtime, |
| 432 |
|
self.method) |
| 433 |
|
|
| 434 |
|
assert len(self.out) == len(self.ref) |
| 435 |
|
assert [isinstance(oo, list) and len(oo) == len(self.lat_in) |
| 436 |
|
for oo in self.out] |
| 437 |
|
|
| 438 |
|
for i, oo in enumerate(self.out): |
| 439 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 440 |
|
|
| 441 |
|
def test_get_aacgm_coord_arr_arr_mix(self): |
| 442 |
|
"""Test array AACGMV2 calculation for an array and floats""" |
|
@@ 180-191 (lines=12) @@
|
| 177 |
|
for i, oo in enumerate(self.out): |
| 178 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 179 |
|
|
| 180 |
|
def test_convert_latlon_arr_arr_mix(self): |
| 181 |
|
"""Test array latlon conversion for mixed type with an array""" |
| 182 |
|
self.out = aacgmv2.convert_latlon_arr(np.array(self.lat_in), |
| 183 |
|
self.lon_in[0], self.alt_in[0], |
| 184 |
|
self.dtime, self.method) |
| 185 |
|
|
| 186 |
|
assert len(self.out) == len(self.ref) |
| 187 |
|
assert [isinstance(oo, list) and len(oo) == len(self.ref[i]) |
| 188 |
|
for i, oo in enumerate(self.out)] |
| 189 |
|
|
| 190 |
|
for i, oo in enumerate(self.out): |
| 191 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 192 |
|
|
| 193 |
|
def test_convert_latlon_arr_mult_failure(self): |
| 194 |
|
"""Test array latlon conversion for mix type with multi-dim array""" |
|
@@ 167-178 (lines=12) @@
|
| 164 |
|
for i, oo in enumerate(self.out): |
| 165 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 166 |
|
|
| 167 |
|
def test_convert_latlon_arr_list_mix(self): |
| 168 |
|
"""Test array latlon conversion for mixed types with list""" |
| 169 |
|
self.out = aacgmv2.convert_latlon_arr(self.lat_in, self.lon_in[0], |
| 170 |
|
self.alt_in[0], self.dtime, |
| 171 |
|
self.method) |
| 172 |
|
|
| 173 |
|
assert len(self.out) == len(self.ref) |
| 174 |
|
assert [isinstance(oo, list) and len(oo) == len(self.ref[i]) |
| 175 |
|
for i, oo in enumerate(self.out)] |
| 176 |
|
|
| 177 |
|
for i, oo in enumerate(self.out): |
| 178 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 179 |
|
|
| 180 |
|
def test_convert_latlon_arr_arr_mix(self): |
| 181 |
|
"""Test array latlon conversion for mixed type with an array""" |
|
@@ 127-138 (lines=12) @@
|
| 124 |
|
for i, oo in enumerate(self.out): |
| 125 |
|
np.testing.assert_allclose(oo, [self.ref[i][0]], rtol=self.rtol) |
| 126 |
|
|
| 127 |
|
def test_convert_latlon_arr_list(self): |
| 128 |
|
"""Test array latlon conversion for list input""" |
| 129 |
|
self.out = aacgmv2.convert_latlon_arr(self.lat_in, self.lon_in, |
| 130 |
|
self.alt_in, self.dtime, |
| 131 |
|
self.method) |
| 132 |
|
|
| 133 |
|
assert len(self.out) == len(self.ref) |
| 134 |
|
assert [isinstance(oo, list) and len(oo) == len(self.ref[i]) |
| 135 |
|
for i, oo in enumerate(self.out)] |
| 136 |
|
|
| 137 |
|
for i, oo in enumerate(self.out): |
| 138 |
|
np.testing.assert_allclose(oo, self.ref[i], rtol=self.rtol) |
| 139 |
|
|
| 140 |
|
def test_convert_latlon_arr_arr_single(self): |
| 141 |
|
"""Test array latlon conversion for array input of shape (1,)""" |
aburrell /
aacgmv2
