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