예제 #1
0
    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