|
@@ 1189-1230 (lines=42) @@
|
| 1186 |
|
|
| 1187 |
|
# test 1D array return values |
| 1188 |
|
|
| 1189 |
|
def test_basevectors_apex_array(): |
| 1190 |
|
apex_out = Apex(date=2000, refh=300) |
| 1191 |
|
(f1, f2, f3, g1, g2, g3, d1, d2, d3, e1, e2, |
| 1192 |
|
e3) = apex_out.basevectors_apex([0, 30], 15, 100, coords='geo') |
| 1193 |
|
(_, _, _, _, f1_1, f2_1, _, d1_1, d2_1, d3_1, _, e1_1, e2_1, |
| 1194 |
|
e3_1) = apex_out._geo2apexall(0, 15, 100) |
| 1195 |
|
(_, _, _, _, f1_2, f2_2, _, d1_2, d2_2, d3_2, _, e1_2, e2_2, |
| 1196 |
|
e3_2) = apex_out._geo2apexall(30, 15, 100) |
| 1197 |
|
|
| 1198 |
|
assert_allclose(f1[:, 0], f1_1) |
| 1199 |
|
assert_allclose(f2[:, 0], f2_1) |
| 1200 |
|
assert_allclose(d1[:, 0], d1_1) |
| 1201 |
|
assert_allclose(d2[:, 0], d2_1) |
| 1202 |
|
assert_allclose(d3[:, 0], d3_1) |
| 1203 |
|
assert_allclose(e1[:, 0], e1_1) |
| 1204 |
|
assert_allclose(e2[:, 0], e2_1) |
| 1205 |
|
assert_allclose(e3[:, 0], e3_1) |
| 1206 |
|
|
| 1207 |
|
assert_allclose(f3[:, 0], np.array([0.092637, -0.245951, 0.938848]), |
| 1208 |
|
rtol=1e-4) |
| 1209 |
|
assert_allclose(g1[:, 0], np.array([0.939012, 0.073416, -0.07342]), |
| 1210 |
|
rtol=1e-4) |
| 1211 |
|
assert_allclose(g2[:, 0], np.array([0.055389, 1.004155, 0.257594]), |
| 1212 |
|
rtol=1e-4) |
| 1213 |
|
assert_allclose(g3[:, 0], np.array([0, 0, 1.065135]), rtol=1e-4) |
| 1214 |
|
|
| 1215 |
|
assert_allclose(f1[:, 1], f1_2) |
| 1216 |
|
assert_allclose(f2[:, 1], f2_2) |
| 1217 |
|
assert_allclose(d1[:, 1], d1_2) |
| 1218 |
|
assert_allclose(d2[:, 1], d2_2) |
| 1219 |
|
assert_allclose(d3[:, 1], d3_2) |
| 1220 |
|
assert_allclose(e1[:, 1], e1_2) |
| 1221 |
|
assert_allclose(e2[:, 1], e2_2) |
| 1222 |
|
assert_allclose(e3[:, 1], e3_2) |
| 1223 |
|
|
| 1224 |
|
assert_allclose(f3[:, 1], np.array([-0.036618, -0.071019, 0.861604]), |
| 1225 |
|
rtol=1e-4) |
| 1226 |
|
assert_allclose(g1[:, 1], np.array([0.844391, 0.015353, 0.037152]), |
| 1227 |
|
rtol=1e-4) |
| 1228 |
|
assert_allclose(g2[:, 1], np.array([0.050808, 1.02131, 0.086342]), |
| 1229 |
|
rtol=1e-4) |
| 1230 |
|
assert_allclose(g3[:, 1], np.array([0, 0, 1.160625]), rtol=1e-4) |
| 1231 |
|
|
| 1232 |
|
|
| 1233 |
|
# test that vectors are calculated correctly |
|
@@ 1118-1159 (lines=42) @@
|
| 1115 |
|
|
| 1116 |
|
|
| 1117 |
|
# test correct vectorization of height |
| 1118 |
|
def test_basevectors_apex_vectorization_height(): |
| 1119 |
|
apex_out = Apex(date=2000, refh=0) |
| 1120 |
|
(f1, f2, f3, g1, g2, g3, d1, d2, d3, e1, e2, |
| 1121 |
|
e3) = apex_out.basevectors_apex(60, 15, [200, 400], coords='geo') |
| 1122 |
|
(_, _, _, _, f1_1, f2_1, _, d1_1, d2_1, d3_1, _, e1_1, e2_1, |
| 1123 |
|
e3_1) = apex_out._geo2apexall(60, 15, 200) |
| 1124 |
|
(_, _, _, _, f1_2, f2_2, _, d1_2, d2_2, d3_2, _, e1_2, e2_2, |
| 1125 |
|
e3_2) = apex_out._geo2apexall(60, 15, 400) |
| 1126 |
|
|
| 1127 |
|
assert_allclose(f1[:, 0], f1_1) |
| 1128 |
|
assert_allclose(f2[:, 0], f2_1) |
| 1129 |
|
assert_allclose(d1[:, 0], d1_1) |
| 1130 |
|
assert_allclose(d2[:, 0], d2_1) |
| 1131 |
|
assert_allclose(d3[:, 0], d3_1) |
| 1132 |
|
assert_allclose(e1[:, 0], e1_1) |
| 1133 |
|
assert_allclose(e2[:, 0], e2_1) |
| 1134 |
|
assert_allclose(e3[:, 0], e3_1) |
| 1135 |
|
|
| 1136 |
|
assert_allclose(f3[:, 0], np.array([-0.088671, -0.018272, 0.993576]), |
| 1137 |
|
rtol=1e-4) |
| 1138 |
|
assert_allclose(g1[:, 0], np.array([0.903098, 0.245273, 0.085107]), |
| 1139 |
|
rtol=1e-4) |
| 1140 |
|
assert_allclose(g2[:, 0], np.array([-0.103495, 1.072078, 0.01048]), |
| 1141 |
|
rtol=1e-4) |
| 1142 |
|
assert_allclose(g3[:, 0], np.array([0, 0, 1.006465]), rtol=1e-4) |
| 1143 |
|
|
| 1144 |
|
assert_allclose(f1[:, 1], f1_2) |
| 1145 |
|
assert_allclose(f2[:, 1], f2_2) |
| 1146 |
|
assert_allclose(d1[:, 1], d1_2) |
| 1147 |
|
assert_allclose(d2[:, 1], d2_2) |
| 1148 |
|
assert_allclose(d3[:, 1], d3_2) |
| 1149 |
|
assert_allclose(e1[:, 1], e1_2) |
| 1150 |
|
assert_allclose(e2[:, 1], e2_2) |
| 1151 |
|
assert_allclose(e3[:, 1], e3_2) |
| 1152 |
|
|
| 1153 |
|
assert_allclose(f3[:, 1], np.array([-0.085415, -0.021176, 0.989645]), |
| 1154 |
|
rtol=1e-4) |
| 1155 |
|
assert_allclose(g1[:, 1], np.array([0.902695, 0.246919, 0.083194]), |
| 1156 |
|
rtol=1e-4) |
| 1157 |
|
assert_allclose(g2[:, 1], np.array([-0.11051, 1.066094, 0.013274]), |
| 1158 |
|
rtol=1e-4) |
| 1159 |
|
assert_allclose(g3[:, 1], np.array([0, 0, 1.010463]), rtol=1e-4) |
| 1160 |
|
|
| 1161 |
|
|
| 1162 |
|
# test scalar return values |
aburrell /
apexpy
