/
test_Vertex.py
62 lines (56 loc) · 3.48 KB
/
test_Vertex.py
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
import unittest
import Vertex
class MyTestCase(unittest.TestCase):
def test_AddVertex(self):#вершина имеется, связи с ней отсутствуют
graff = Vertex.SimpleGraph(4)
graff.AddVertex(1)
self.assertEqual(graff.vertex.count(None) == len(graff.vertex) - 1, True)#считает количкество Нан в масисиве
self.assertEqual(graff.m_adjacency.count(1) == 0, True) #проверяет количество связей (едениц) в матрице
graff.AddVertex(3)
self.assertEqual(graff.vertex.count(None) == len(graff.vertex) - 2, True)
self.assertEqual(graff.m_adjacency.count(1) == 0, True)
def test_AddEdge(self):
graff = Vertex.SimpleGraph(4)# до добавления связи между вершинами не было, после добавления появилась
graff.AddVertex(1)
graff.AddVertex(2)
graff.AddVertex(3)
graff.AddVertex(4)
self.assertEqual(graff.m_adjacency[1][2] == 0, True)
self.assertEqual(graff.m_adjacency[2][1] == 0, True)
graff.AddEdge(1, 2)
self.assertEqual(graff.m_adjacency[1].count(1) == 1, True)#считает количкество едениц в масисиве
self.assertEqual(graff.m_adjacency[2].count(1) == 1, True)
self.assertEqual(graff.m_adjacency[1][2] == 1, True)
self.assertEqual(graff.m_adjacency[2][1] == 1, True)
def test_RemoveEdge(self):
graff = Vertex.SimpleGraph(4) # до удаления связь между вершинами была, после удаления отсутствует
graff.AddVertex(1)
graff.AddVertex(2)
graff.AddVertex(3)
graff.AddVertex(4)
graff.AddEdge(1, 2)
self.assertEqual(graff.m_adjacency[1][2] == 1, True)
self.assertEqual(graff.m_adjacency[2][1] == 1, True)
graff.RemoveEdge(1, 2)
self.assertEqual(graff.m_adjacency[1][2] == 0, True)
self.assertEqual(graff.m_adjacency[2][1] == 0, True)
def test_RemoveVertex(self):
graff = Vertex.SimpleGraph(4)# до удаления некоторые вершины имеют связи с удаляемой вершиной, после удаления этих связей нету
graff.AddVertex(1)
graff.AddVertex(2)
graff.AddVertex(3)
graff.AddVertex(4)
graff.AddEdge(1, 2)
#до удаления
self.assertEqual(graff.vertex.count(None) == 0, True) # считает количкество Нан в масисиве
self.assertEqual(graff.m_adjacency[1].count(0) == 3, True) # считает количкество едениц в масисиве до добавления связей
self.assertEqual(graff.m_adjacency[2][1] == 1, True)
self.assertEqual(graff.m_adjacency[1][2] == 1, True)
graff.RemoveVertex(1)
#после удаления
self.assertEqual(graff.vertex.count(None) == 1, True) # считает количкество Нан в масисиве
self.assertEqual(graff.m_adjacency[1].count(0) == 4, True) # считает количкество едениц в масисиве до добавления связей
self.assertEqual(graff.m_adjacency[2][1] == 0, True)
self.assertEqual(graff.m_adjacency[1][2] == 0, True)
if __name__ == '__main__':
unittest.main()