Exemple #1
0
    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())
Exemple #2
0
    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())