def test_shift(self):
cube = Cube((2, 2), np.array([0, 1, 2, 3]))
np.testing.assert_equal(
np.array([[0., 0., 1., 0.], [0., 0., 0., 1.], [1., 0., 0., 0.],
[0., 1., 0., 0.]]), cube.shift((1, 0)))
def test_shift(self):
cube = Cube((2, 2), np.array([0, 1, 2, 3]))
np.testing.assert_equal(
np.array([[0., 0., 1., 0.], [0., 0., 0., 1.], [1., 0., 0., 0.],
[0., 1., 0., 0.]]), cube.shift((1, 0)))
