def test_degree(self): # (1) test a-->b and a-->c vertex = Vertex('a') vertex.adj['b'] = 1 self.assertEqual(vertex.degree(), 1) vertex.adj['c'] = 3 assert vertex.degree() == 2 # (2) test a-->letter for all letters in alphabet vertex = Vertex('a') for i, char in enumerate(string.ascii_lowercase): self.assertEqual(vertex.degree(), i) vertex.adj[char] = i
def test_01_vertex_methods(self): ### degree ### vertex = Vertex('a') vertex.adj['b'] = 1 assert vertex.degree() == 1 vertex.adj['c'] = 3 assert vertex.degree() == 2 ### visit/reset ### assert not vertex.visited vertex.visit() assert vertex.visited vertex.reset() assert not vertex.visited ### get_edges ### vertex = Vertex('a') solution = [('b', 1), ('c', 2)] vertex.adj['b'] = 1 vertex.adj['c'] = 2 subject = vertex.get_edges() assert subject == solution ### euclidean_distance ### vertex_a = Vertex('a') vertex_b = Vertex('b', 3, 4) subject = vertex_a.euclidean_distance(vertex_b) assert subject == 5 subject = vertex_b.euclidean_distance(vertex_a) assert subject == 5