Add unit test for Vertex class