Fix return for LegacyCoordinates

Author: astrofrog

glue/core/coordinate_helpers.py

@@ -63,9 +63,8 @@ def pixel2world_single_axis(wcs, *pixel, world_axis=None):
         result = result.reshape(pixel_shape)
     else:
         result = wcs.pixel_to_world_values(*pixel)
-
-    if len(pixel) > 1:
-        result = result[world_axis]
+        if len(pixel) > 1:
+            result = result[world_axis]
 
     return np.broadcast_to(result, original_shape)
 
@@ -121,9 +120,8 @@ def world2pixel_single_axis(wcs, *world, pixel_axis=None):
         result = result.reshape(world_shape)
     else:
         result = wcs.world_to_pixel_values(*world)
-
-    if len(world) > 1:
-        result = result[pixel_axis]
+        if len(world) > 1:
+            result = result[pixel_axis]
 
     return np.broadcast_to(result, original_shape)
 

glue/core/coordinates.py

@@ -90,13 +90,13 @@ def __init__(self):
         super().__init__(pixel_n_dim=10, world_n_dim=10)
 
     def pixel_to_world_values(self, *pixel):
-        if self.pixel_n_dim == 1:
+        if len(pixel) == 1:
             return pixel[0]
         else:
             return pixel
 
     def world_to_pixel_values(self, *world):
-        if self.world_n_dim == 1:
+        if len(world) == 1:
             return world[0]
         else:
             return world