def test_entity_net_is_sparse_returns_false_for_numpy_matrix(self): e = EntityNet(self.net_a, 'Net', self.node_names, precomputed=self.net_a_precomp) self.assertFalse(e.is_sparse())
def test_entity_net_is_sparse_returns_true_for_csr_matrix(self): sp = sparse.csr_matrix(self.net_a) e = EntityNet(sp, 'Net', self.node_names, precomputed=self.net_a_precomp) self.assertTrue(e.is_sparse())