def test_extract(self):
indices, values, shape = sparse.extract(self.data)
self.assertEqual(indices[0].tolist(),
sparse.indices(self.data)[0].tolist())
self.assertEqual(indices[1].tolist(),
sparse.indices(self.data)[1].tolist())
self.assertEqual(values.tolist(), sparse.values(self.data).tolist())
self.assertEqual(shape, sparse.shape(self.data))
def test_indices(self):
i0, = sparse.indices(self.data)
self.assertEqual(i0.tolist(), [4, 4, 3, 2, 1, 2, 3, 0, 0])
