Python DenseAdjacencyMatrixGraphの例

コード例 #1

ファイル: test_adj.py プロジェクト: ckanu13k/graphs

class TestAdjacencyMatrixGraphs(unittest.TestCase):
  def setUp(self):
    self.G = DenseAdjacencyMatrixGraph(ADJ)
    self.S = SparseAdjacencyMatrixGraph(csr_matrix(ADJ))

  def test_pairs(self):
    assert_array_equal(self.G.pairs(), PAIRS)
    assert_array_equal(self.S.pairs(), PAIRS)

  def test_matrix(self):
    M = self.G.matrix()
    assert_array_equal(M, ADJ)
    M = self.G.matrix(csr=True)
    self.assertEqual(M.format, 'csr')
    assert_array_equal(M.toarray(), ADJ)
    M = self.S.matrix()
    self.assertEqual(M.format, 'csr')
    assert_array_equal(M.toarray(), ADJ)

  def test_matrix_copy(self):
    M = self.G.matrix(dense=True, copy=False)
    assert_array_equal(M, ADJ)
    M2 = self.G.matrix(dense=True, copy=True)
    assert_array_equal(M, M2)
    self.assertIsNot(M, M2)
    # Sparse case
    M = self.S.matrix(csr=True, copy=False)
    assert_array_equal(M.toarray(), ADJ)
    M2 = self.S.matrix(csr=True, copy=True)
    assert_array_equal(M.toarray(), M2.toarray())
    self.assertIsNot(M, M2)

コード例 #2

ファイル: test_adj.py プロジェクト: dhengkt/CourseProjects

class TestAdjacencyMatrixGraphs(unittest.TestCase):
    def setUp(self):
        self.G = DenseAdjacencyMatrixGraph(ADJ)
        self.S = SparseAdjacencyMatrixGraph(csr_matrix(ADJ))

    def test_pairs(self):
        assert_array_equal(self.G.pairs(), PAIRS)
        assert_array_equal(self.S.pairs(), PAIRS)

    def test_matrix(self):
        M = self.G.matrix()
        assert_array_equal(M, ADJ)
        M = self.G.matrix('csr')
        self.assertEqual(M.format, 'csr')
        assert_array_equal(M.toarray(), ADJ)
        M = self.S.matrix()
        self.assertEqual(M.format, 'csr')
        assert_array_equal(M.toarray(), ADJ)

    def test_matrix_copy(self):
        M = self.G.matrix('dense', copy=False)
        assert_array_equal(M, ADJ)
        M2 = self.G.matrix('dense', copy=True)
        assert_array_equal(M, M2)
        self.assertIsNot(M, M2)
        # Sparse case
        M = self.S.matrix('csr', copy=False)
        assert_array_equal(M.toarray(), ADJ)
        M2 = self.S.matrix('csr', copy=True)
        assert_array_equal(M.toarray(), M2.toarray())
        self.assertIsNot(M, M2)

コード例 #3

ファイル: test_adj.py プロジェクト: ckanu13k/graphs

 def setUp(self):
   self.G = DenseAdjacencyMatrixGraph(ADJ)
   self.S = SparseAdjacencyMatrixGraph(csr_matrix(ADJ))

コード例 #4

ファイル: test_adj.py プロジェクト: dhengkt/CourseProjects

 def setUp(self):
     self.G = DenseAdjacencyMatrixGraph(ADJ)
     self.S = SparseAdjacencyMatrixGraph(csr_matrix(ADJ))