def test_extra_2d_ax0(self):
x = np.ones((4,5))
y = right_shift(x, k=6, axis=0)
t = np.zeros((4,5))
nptest.assert_array_equal(y, t)
def test_extra_1d(self):
x = np.ones(4)
y = right_shift(x, k=5)
t = np.zeros(4)
nptest.assert_array_equal(y, t)
def test_k2_2d_ax0(self):
x = np.ones((4,5))
y = right_shift(x, k=2, axis=0)
t = np.vstack([ np.zeros((2,5)), np.ones((2,5)) ])
nptest.assert_array_equal(y, t)
def test_k2_1d(self):
x = np.ones(4)
y = right_shift(x, k=2)
t = np.array([0, 0, 1, 1])
nptest.assert_array_equal(y, t)
def test_basic_2d_ax1(self):
x = np.ones((4,5))
y = right_shift(x, axis=1)
t = np.hstack([ np.zeros((4,1)), np.ones((4,4)) ])
nptest.assert_array_equal(y, t)
def test_basic_1d(self):
x = np.ones(4)
y = right_shift(x)
t = np.array([0, 1, 1, 1])
nptest.assert_array_equal(y, t)