def test_shift3():
for offset in ( (3,), (3,4), (3,4,5) ):
for axis in range(len(offset),4):
s = np.random.random_integers(-2,2,offset)
shift = ShiftOperator([s], axis=axis, shapein=(3,4,5,6))
yield assert_equal, shift.todense().T, shift.T.todense()
def test_shift2():
for axis in range(1,4):
shift = ShiftOperator(((1,2,3),), axis=axis, shapein=(3,4,5,6))
yield assert_equal, shift.todense().T, shift.T.todense()