eci: only compare non-scalar shape

geospace-code · Nov 20, 2023 · 85fc6ea · 85fc6ea
1 parent 8347339
commit 85fc6ea
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/pymap3d/eci.py b/src/pymap3d/eci.py
@@ -57,7 +57,8 @@ def eci2ecef(x, y, z, time: datetime) -> tuple:
         z = atleast_1d(z)
         gst = atleast_1d(greenwichsrt(juliandate(time)))
         assert x.shape == y.shape == z.shape, f"shape mismatch: x: ${x.shape}  y: {y.shape}  z: {z.shape}"
-        assert x.size == gst.size
+        if gst.size > 1:
+            assert x.size == gst.size, f"shape mismatch: x: {x.shape}  gst: {gst.shape}"
 
         eci = column_stack((x.ravel(), y.ravel(), z.ravel()))
         ecef = empty((x.size, 3))