add unit tests

Author: scivision

src/pymap3d/tests/test_aer.py

@@ -16,6 +16,9 @@ def test_aer2ecef(aer, lla, xyz):
     assert x == approx(xyz[0])
     assert y == approx(xyz[1])
     assert z == approx(xyz[2])
+    assert isinstance(x, float)
+    assert isinstance(y, float)
+    assert isinstance(z, float)
 
     raer = (radians(aer[0]), radians(aer[1]), aer[2])
     rlla = (radians(lla[0]), radians(lla[1]), lla[2])

src/pymap3d/tests/test_vincenty_vreckon.py

@@ -30,7 +30,10 @@ def test_unit(lat, lon, srange, az, lato, lono):
     lat1, lon1 = vincenty.vreckon(lat, lon, srange, az)
 
     assert lat1 == approx(lato)
+    assert isinstance(lat1, float)
+
     assert lon1 == approx(lono, rel=0.001)
+    assert isinstance(lon1, float)
 
 
 def test_az_vector():

src/pymap3d/vincenty.py

@@ -450,7 +450,10 @@ def vreckon(
 
     lon2 = lon2 % 360  # follow [0, 360) convention
 
-    return degrees(lat2), lon2
+    try:
+        return degrees(lat2).squeeze()[()], lon2.squeeze()[()]
+    except AttributeError:
+        return degrees(lat2), lon2
 
 
 def track2(