FIX: fixed array.sum(incompatible_axis) (fixes larray-project#1138) (BUT I am unsure I should fix it, or disallow/warn on array[axis] because it makes aggregates a bit ambiguous (and might break current code)

Author: gdementen

larray/core/array.py

@@ -2937,7 +2937,12 @@ def to_labelgroup(key, stack_depth=1):
 
         def standardise_arg(arg, stack_depth=1):
             if self.axes.isaxis(arg):
-                return self.axes[arg]
+                real_axis = self.axes[arg]
+                if (isinstance(arg, Axis) and not isinstance(arg, AxisReference)
+                        and not arg.equals(real_axis)):
+                    return arg[:].retarget_to(real_axis)
+                else:
+                    return real_axis
             else:
                 return to_labelgroup(arg, stack_depth + 1)
 

larray/tests/test_array.py

@@ -1879,6 +1879,12 @@ def test_sum_full_axes(array):
     # filter on aggregated
     assert aggregated.filter(b=b_group1).shape == (7, 6)
 
+    # using an Axis object from another array, we must filter the axis
+    array = ndtest(3)
+    smaller_a_axis = Axis('a=a0,a1')
+    res = array.sum(smaller_a_axis)
+    assert res == 1
+
 
 def test_sum_full_axes_with_nan(array):
     array['c0', 'd2', 'b1', 0] = nan