def test_connect_edges_to_vertex_should_be_1(self): v1 = Vertex(0, 0) v2 = Vertex(1, 2) e = Edge(v1, v2) am.connect_edges_to_vertices([e]) self.assertEqual(1, len([v for v in v1.connected_edges if v is not None]))
def test_connect_edges_to_vertex_should_be_2(self): v1 = Vertex(0, 0) v2 = Vertex(1, 0) v3 = Vertex(-1, 0) e1 = Edge(v1, v2) e2 = Edge(v3, v1) am.connect_edges_to_vertices([e1, e2]) self.assertEqual(2, len([v for v in v1.connected_edges if v is not None]))
def test_connect_edges_to_vertex_should_be_sorted_right_left_up_down(self): v = Vertex(0, 0) e_left = Edge(Vertex(-1, 0), v) e_right = Edge(v, Vertex(1, 0)) e_up = Edge(v, Vertex(0, 1)) e_down = Edge(Vertex(0, -1), v) edges = [e_left, e_right, e_down, e_up] am.connect_edges_to_vertices(edges) ce = v.connected_edges self.assertEqual(ce[DIRECTION['RIGHT']], e_right) self.assertEqual(ce[DIRECTION['LEFT']], e_left) self.assertEqual(ce[DIRECTION['UP']], e_up) self.assertEqual(ce[DIRECTION['DOWN']], e_down)