def test_from_dense(): t = fromdense(tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2)) assert isinstance(t, SparseTensor) assert isinstance(t.op, DenseToSparse) assert t.issparse() is True t = tile(t) assert t.chunks[0].index == (0, 0) assert isinstance(t.op, DenseToSparse)
def testFromDense(self): t = fromdense(tensor([[0, 0, 1], [1, 0, 0]], chunk_size=2)) self.assertIsInstance(t, SparseTensor) self.assertIsInstance(t.op, DenseToSparse) self.assertTrue(t.issparse()) t = t.tiles() self.assertEqual(t.chunks[0].index, (0, 0)) self.assertIsInstance(t.op, DenseToSparse)