ensure Block.wcs works for non-2D data

GeminiDRSoftware · Aug 12, 2020 · 7f6c47f · 7f6c47f
1 parent 958b5e2
commit 7f6c47f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gempy/library/transform.py b/gempy/library/transform.py
@@ -176,8 +176,8 @@ def shape(self):
     @property
     def wcs(self):
         """WCS object"""
-        if self.corners[0] != (0, 0):
-            raise ValueError("First element does not start at (0, 0)")
+        if not all (c == 0 for c in self.corners[0]):
+            raise ValueError("First element does not start at origin")
         return self._elements[0].wcs
 
     def _return_array(self, elements):