示例#1
0
    def test_del_vertex(self):
        d = DirectedGraph({1: [2, 3], 2: [4], 3: [4]})

        self.assertEqual(d.get_outdegree(1), 2)
        self.assertEqual(d.get_indegree(4), 2)
        self.assertEqual(len(d), 4)
        self.assertTrue(2 in d)

        d.del_vertex(2)

        self.assertEqual(d.get_outdegree(1), 1)
        self.assertEqual(d.get_indegree(4), 1)
        self.assertEqual(len(d), 3)
        self.assertFalse(2 in d)
示例#2
0
def test_del_vertex():
    graph = DirectedGraph({1: [2, 3], 2: [4], 3: [4]})

    assert graph.get_outdegree(1) == 2
    assert graph.get_indegree(4) == 2
    assert len(graph) == 4
    assert 2 in graph

    graph.del_vertex(2)

    assert graph.get_outdegree(1) == 1
    assert graph.get_indegree(4) == 1
    assert len(graph) == 3
    assert 2 not in graph
示例#3
0
    def test_add_edge(self):
        d = DirectedGraph()
        self.assertFalse(d.is_edge(1, 2))
        self.assertEqual(d.get_outdegree(1), 0)

        d.add_edge(1, 2)

        self.assertEqual(d.get_outdegree(1), 1)
        self.assertEqual(d.get_indegree(2), 1)
        self.assertTrue(d.is_edge(1, 2))

        d.add_edge(1, 2)

        self.assertTrue(1, 2)
        self.assertEqual(d.get_indegree(2), 1)
示例#4
0
 def test_add_vertex(self):
     d = DirectedGraph({1: [2, 3], 2: [4], 3: [4]})
     d.add_vertex(5)
     self.assertTrue(5 in d)
     self.assertEqual(d.get_indegree(5), 0)
     self.assertEqual(d.get_outdegree(5), 0)
     self.assertEqual(d[5], [])
示例#5
0
def test_add_vertex():
    graph = DirectedGraph({1: [2, 3], 2: [4], 3: [4]})
    graph.add_vertex(5)
    assert 5 in graph
    assert graph.get_indegree(5) == 0
    assert graph.get_outdegree(5) == 0
    assert graph.get_neighbors(5) == []
示例#6
0
def test_add_edge():
    graph = DirectedGraph()
    assert not graph.is_edge(1, 2)
    assert graph.get_outdegree(1) == 0

    graph.add_edge(1, 2)

    assert graph.get_outdegree(1) == 1
    assert graph.get_indegree(2) == 1
    assert graph.is_edge(1, 2)

    graph.add_edge(1, 2)

    assert graph.get_indegree(1) == 0
    assert graph.get_outdegree(1) == 2
    assert graph.get_indegree(2) == 2