def test_unsigned_type_error(self):
x = sparse.random((5, 5, 5), density=0.3, idx_dtype=np.uint8)
with pytest.raises(ValueError):
sparse.roll(x, -1)
def test_valerr(self, args):
x = sparse.random((2, 2, 2), density=1)
with pytest.raises(ValueError):
sparse.roll(x, *args)
def test_dtype_errors(self, dtype, shift):
x = sparse.random((5, 5, 5), density=0.2, idx_dtype=dtype)
with pytest.raises(ValueError):
sparse.roll(x, shift)
def test_empty(self):
x = np.array([])
assert_eq(np.roll(x, 1), sparse.roll(sparse.as_coo(x), 1))
def test_original_is_copied(self, shift, ax):
xs = sparse.random((10, 10), density=0.5)
xc = COO(np.copy(xs.coords), np.copy(xs.data), shape=xs.shape)
sparse.roll(xs, shift, axis=ax)
assert_eq(xs, xc)
def test_multiaxis(self, shift, ax):
xs = sparse.random((9, 9, 9), density=0.5)
x = xs.todense()
assert_eq(np.roll(x, shift, axis=ax), sparse.roll(xs, shift, axis=ax))
assert_eq(np.roll(x, shift, axis=ax), sparse.roll(x, shift, axis=ax))
def test_2d(self, shift, ax):
xs = sparse.random((10, 10), density=0.5)
x = xs.todense()
assert_eq(np.roll(x, shift, axis=ax), sparse.roll(xs, shift, axis=ax))
assert_eq(np.roll(x, shift, axis=ax), sparse.roll(x, shift, axis=ax))
def test_1d(self, shift):
xs = sparse.random((100, ), density=0.5)
x = xs.todense()
assert_eq(np.roll(x, shift), sparse.roll(xs, shift))
assert_eq(np.roll(x, shift), sparse.roll(x, shift))
