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_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_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)
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)
