|
@@ 1143-1184 (lines=42) @@
|
| 1140 |
|
|
| 1141 |
|
# test 1D array return values |
| 1142 |
|
|
| 1143 |
|
def test_basevectors_apex_array(): |
| 1144 |
|
apex_out = Apex(date=2000, refh=300) |
| 1145 |
|
(f1, f2, f3, g1, g2, g3, d1, d2, d3, e1, e2, |
| 1146 |
|
e3) = apex_out.basevectors_apex([0, 30], 15, 100, coords='geo') |
| 1147 |
|
(_, _, _, _, f1_1, f2_1, _, d1_1, d2_1, d3_1, _, e1_1, e2_1, |
| 1148 |
|
e3_1) = apex_out._geo2apexall(0, 15, 100) |
| 1149 |
|
(_, _, _, _, f1_2, f2_2, _, d1_2, d2_2, d3_2, _, e1_2, e2_2, |
| 1150 |
|
e3_2) = apex_out._geo2apexall(30, 15, 100) |
| 1151 |
|
|
| 1152 |
|
assert_allclose(f1[:, 0], f1_1) |
| 1153 |
|
assert_allclose(f2[:, 0], f2_1) |
| 1154 |
|
assert_allclose(d1[:, 0], d1_1) |
| 1155 |
|
assert_allclose(d2[:, 0], d2_1) |
| 1156 |
|
assert_allclose(d3[:, 0], d3_1) |
| 1157 |
|
assert_allclose(e1[:, 0], e1_1) |
| 1158 |
|
assert_allclose(e2[:, 0], e2_1) |
| 1159 |
|
assert_allclose(e3[:, 0], e3_1) |
| 1160 |
|
|
| 1161 |
|
assert_allclose(f3[:, 0], np.array([0.092637, -0.245951, 0.938848]), |
| 1162 |
|
rtol=1e-4) |
| 1163 |
|
assert_allclose(g1[:, 0], np.array([0.939012, 0.073416, -0.07342]), |
| 1164 |
|
rtol=1e-4) |
| 1165 |
|
assert_allclose(g2[:, 0], np.array([0.055389, 1.004155, 0.257594]), |
| 1166 |
|
rtol=1e-4) |
| 1167 |
|
assert_allclose(g3[:, 0], np.array([0, 0, 1.065135]), rtol=1e-4) |
| 1168 |
|
|
| 1169 |
|
assert_allclose(f1[:, 1], f1_2) |
| 1170 |
|
assert_allclose(f2[:, 1], f2_2) |
| 1171 |
|
assert_allclose(d1[:, 1], d1_2) |
| 1172 |
|
assert_allclose(d2[:, 1], d2_2) |
| 1173 |
|
assert_allclose(d3[:, 1], d3_2) |
| 1174 |
|
assert_allclose(e1[:, 1], e1_2) |
| 1175 |
|
assert_allclose(e2[:, 1], e2_2) |
| 1176 |
|
assert_allclose(e3[:, 1], e3_2) |
| 1177 |
|
|
| 1178 |
|
assert_allclose(f3[:, 1], np.array([-0.036618, -0.071019, 0.861604]), |
| 1179 |
|
rtol=1e-4) |
| 1180 |
|
assert_allclose(g1[:, 1], np.array([0.844391, 0.015353, 0.037152]), |
| 1181 |
|
rtol=1e-4) |
| 1182 |
|
assert_allclose(g2[:, 1], np.array([0.050808, 1.02131, 0.086342]), |
| 1183 |
|
rtol=1e-4) |
| 1184 |
|
assert_allclose(g3[:, 1], np.array([0, 0, 1.160625]), rtol=1e-4) |
| 1185 |
|
|
| 1186 |
|
|
| 1187 |
|
# test that vectors are calculated correctly |
|
@@ 1072-1113 (lines=42) @@
|
| 1069 |
|
|
| 1070 |
|
|
| 1071 |
|
# test correct vectorization of height |
| 1072 |
|
def test_basevectors_apex_vectorization_height(): |
| 1073 |
|
apex_out = Apex(date=2000, refh=0) |
| 1074 |
|
(f1, f2, f3, g1, g2, g3, d1, d2, d3, e1, e2, |
| 1075 |
|
e3) = apex_out.basevectors_apex(60, 15, [200, 400], coords='geo') |
| 1076 |
|
(_, _, _, _, f1_1, f2_1, _, d1_1, d2_1, d3_1, _, e1_1, e2_1, |
| 1077 |
|
e3_1) = apex_out._geo2apexall(60, 15, 200) |
| 1078 |
|
(_, _, _, _, f1_2, f2_2, _, d1_2, d2_2, d3_2, _, e1_2, e2_2, |
| 1079 |
|
e3_2) = apex_out._geo2apexall(60, 15, 400) |
| 1080 |
|
|
| 1081 |
|
assert_allclose(f1[:, 0], f1_1) |
| 1082 |
|
assert_allclose(f2[:, 0], f2_1) |
| 1083 |
|
assert_allclose(d1[:, 0], d1_1) |
| 1084 |
|
assert_allclose(d2[:, 0], d2_1) |
| 1085 |
|
assert_allclose(d3[:, 0], d3_1) |
| 1086 |
|
assert_allclose(e1[:, 0], e1_1) |
| 1087 |
|
assert_allclose(e2[:, 0], e2_1) |
| 1088 |
|
assert_allclose(e3[:, 0], e3_1) |
| 1089 |
|
|
| 1090 |
|
assert_allclose(f3[:, 0], np.array([-0.088671, -0.018272, 0.993576]), |
| 1091 |
|
rtol=1e-4) |
| 1092 |
|
assert_allclose(g1[:, 0], np.array([0.903098, 0.245273, 0.085107]), |
| 1093 |
|
rtol=1e-4) |
| 1094 |
|
assert_allclose(g2[:, 0], np.array([-0.103495, 1.072078, 0.01048]), |
| 1095 |
|
rtol=1e-4) |
| 1096 |
|
assert_allclose(g3[:, 0], np.array([0, 0, 1.006465]), rtol=1e-4) |
| 1097 |
|
|
| 1098 |
|
assert_allclose(f1[:, 1], f1_2) |
| 1099 |
|
assert_allclose(f2[:, 1], f2_2) |
| 1100 |
|
assert_allclose(d1[:, 1], d1_2) |
| 1101 |
|
assert_allclose(d2[:, 1], d2_2) |
| 1102 |
|
assert_allclose(d3[:, 1], d3_2) |
| 1103 |
|
assert_allclose(e1[:, 1], e1_2) |
| 1104 |
|
assert_allclose(e2[:, 1], e2_2) |
| 1105 |
|
assert_allclose(e3[:, 1], e3_2) |
| 1106 |
|
|
| 1107 |
|
assert_allclose(f3[:, 1], np.array([-0.085415, -0.021176, 0.989645]), |
| 1108 |
|
rtol=1e-4) |
| 1109 |
|
assert_allclose(g1[:, 1], np.array([0.902695, 0.246919, 0.083194]), |
| 1110 |
|
rtol=1e-4) |
| 1111 |
|
assert_allclose(g2[:, 1], np.array([-0.11051, 1.066094, 0.013274]), |
| 1112 |
|
rtol=1e-4) |
| 1113 |
|
assert_allclose(g3[:, 1], np.array([0, 0, 1.010463]), rtol=1e-4) |
| 1114 |
|
|
| 1115 |
|
|
| 1116 |
|
# test scalar return values |
aburrell /
apexpy
