def test_sparse_repeat_1d(self):
sparse_1d = sparse_getitem(self.sparse, 1)
actual = sparse_1d.to_dense().repeat(3, 1)
res = sparse_repeat(sparse_1d, 3, 1)
self.assertTrue(torch.equal(actual, res.to_dense()))
actual = sparse_1d.to_dense().repeat(2, 3)
res = sparse_repeat(sparse_1d, 2, 3)
self.assertTrue(torch.equal(actual, res.to_dense()))
def test_sparse_repeat_1d():
sparse_1d = sparse_getitem(sparse, 1)
actual = sparse_1d.to_dense().repeat(3, 1)
res = sparse_repeat(sparse_1d, 3, 1)
assert torch.equal(actual, res.to_dense())
actual = sparse_1d.to_dense().repeat(2, 3)
res = sparse_repeat(sparse_1d, 2, 3)
assert torch.equal(actual, res.to_dense())
def test_sparse_repeat_2d(self):
actual = self.sparse.to_dense().repeat(3, 2)
res = sparse_repeat(self.sparse, 3, 2)
self.assertTrue(torch.equal(actual, res.to_dense()))
actual = self.sparse.to_dense().repeat(1, 2)
res = sparse_repeat(self.sparse, 1, 2)
self.assertTrue(torch.equal(actual, res.to_dense()))
actual = self.sparse.to_dense().repeat(3, 1)
res = sparse_repeat(self.sparse, 3, 1)
self.assertTrue(torch.equal(actual, res.to_dense()))
def test_sparse_repeat_2d():
actual = sparse.to_dense().repeat(3, 2)
res = sparse_repeat(sparse, 3, 2)
assert torch.equal(actual, res.to_dense())
actual = sparse.to_dense().repeat(1, 2)
res = sparse_repeat(sparse, 1, 2)
assert torch.equal(actual, res.to_dense())
actual = sparse.to_dense().repeat(3, 1)
res = sparse_repeat(sparse, 3, 1)
assert torch.equal(actual, res.to_dense())
