def test__pull_axis():
a = Axis('x', 0, None)
b = Axis('y', 1, None)
c = Axis('z', 2, None)
t_pos = Axis('y', 1, None)
t_neg = Axis('x', 5, None)
axes = [a, b, c]
yield nt.assert_true, t_pos in axes
yield nt.assert_false, t_neg in axes
yield nt.assert_equal, axes, _pull_axis(axes, t_neg)
yield nt.assert_equal, axes[:-1], _pull_axis(axes, c)
new_axes = [a, Axis('z', 1, None)]
yield nt.assert_equal, new_axes, _pull_axis(axes, t_pos)
def test__pull_axis():
a = Axis('x', 0, None)
b = Axis('y', 1, None)
c = Axis('z', 2, None)
t_pos = Axis('y', 1, None)
t_neg = Axis('x', 5, None)
axes = [a, b, c]
yield nt.assert_true, t_pos in axes
yield nt.assert_false, t_neg in axes
yield nt.assert_equal, axes, _pull_axis(axes, t_neg)
yield nt.assert_equal, axes[:-1], _pull_axis(axes, c)
new_axes = [a, Axis('z', 1, None)]
yield nt.assert_equal, new_axes, _pull_axis(axes, t_pos)
def test__pull_axis():
a = Axis("x", 0, None)
b = Axis("y", 1, None)
c = Axis("z", 2, None)
t_pos = Axis("y", 1, None)
t_neg = Axis("x", 5, None)
axes = [a, b, c]
nt.assert_true(t_pos in axes)
nt.assert_false(t_neg in axes)
nt.assert_equal(axes, _pull_axis(axes, t_neg))
nt.assert_equal(axes[:-1], _pull_axis(axes, c))
new_axes = [a, Axis("z", 1, None)]
nt.assert_equal(new_axes, _pull_axis(axes, t_pos))
def test__pull_axis():
a = Axis('x', 0, None)
b = Axis('y', 1, None)
c = Axis('z', 2, None)
t_pos = Axis('y', 1, None)
t_neg = Axis('x', 5, None)
axes = [a, b, c]
nt.assert_true(t_pos in axes)
nt.assert_false(t_neg in axes)
nt.assert_equal(axes, _pull_axis(axes, t_neg))
nt.assert_equal(axes[:-1], _pull_axis(axes, c))
new_axes = [a, Axis('z', 1, None)]
nt.assert_equal(new_axes, _pull_axis(axes, t_pos))
def assert_axes_correct(d_arr, op, axis):
from datarray.datarray import _names_to_numbers, _pull_axis
opr = getattr(d_arr, op)
d = opr(axis=axis)
axis_idx = _names_to_numbers(d_arr.axes, [axis])[0]
if op not in accumulations:
axes = _pull_axis(d_arr.axes, d_arr.axes[axis_idx])
else:
axes = d_arr.axes
assert all( [ax1==ax2 for ax1, ax2 in zip(d.axes, axes)] ), \
'mislabeled axes from operation %s'%op
def assert_axes_correct(d_arr, op, axis):
from datarray.datarray import _names_to_numbers, _pull_axis
opr = getattr(d_arr, op)
d = opr(axis=axis)
axis_idx = _names_to_numbers(d_arr.axes, [axis])[0]
if op not in accumulations:
axes = _pull_axis(d_arr.axes, d_arr.axes[axis_idx])
else:
axes = d_arr.axes
assert all( [ax1==ax2 for ax1, ax2 in zip(d.axes, axes)] ), \
'mislabeled axes from operation %s'%op
