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()