def test_add_chord(self): M1 = PlanarMap() A, B, C, D = Point(0, 1), Point(1, 1), Point(0, 0), Point(1, 0) # empty map self.assertRaises(AssertionError, M1.add_chord, Segment(B, D)) M1.add_first_edge(Segment(A, B)) # adding leaf self.assertRaises(AssertionError, M1.add_chord, Segment(A, C)) M1.add_leaf(Segment(A, C)) M1.add_chord(Segment(C, B)) # adding the same chord self.assertRaises(ValueError, M1.add_chord, Segment(A, B))
class TestPlanarMap1(unittest.TestCase): def setUp(self): self.M = PlanarMap() A, B, C, D = Point(0, 1), Point(1, 1), Point(0, 0), Point(1, 0) self.M.add_first_edge(Segment(A, B)) self.M.add_leaf(Segment(A, C)) self.M.add_leaf(Segment(B, D)) self.M.add_chord(Segment(C, B)) #self.G.show() def test_add_first_edge(self): M1 = PlanarMap() A, B, C, D = Point(0, 1), Point(1, 1), Point(0, 0), Point(1, 0) # not segment self.assertRaises(AssertionError, M1.add_first_edge, Edge(B, D)) M1.add_first_edge(Segment(A, B)) # second segment self.assertRaises(AssertionError, M1.add_first_edge, Segment(B, D)) def test_add_leaf(self): M1 = PlanarMap() A, B, C, D = Point(0, 1), Point(1, 1), Point(0, 0), Point(1, 0) # empty map self.assertRaises(AssertionError, M1.add_leaf, Segment(B, D)) M1.add_first_edge(Segment(A, B)) M1.add_leaf(Segment(A, C)) # adding chord self.assertRaises(AssertionError, M1.add_leaf, Segment(B, C)) # adding the same leaf self.assertRaises(AssertionError, M1.add_leaf, Segment(A, C)) #M1.show() def test_add_chord(self): M1 = PlanarMap() A, B, C, D = Point(0, 1), Point(1, 1), Point(0, 0), Point(1, 0) # empty map self.assertRaises(AssertionError, M1.add_chord, Segment(B, D)) M1.add_first_edge(Segment(A, B)) # adding leaf self.assertRaises(AssertionError, M1.add_chord, Segment(A, C)) M1.add_leaf(Segment(A, C)) M1.add_chord(Segment(C, B)) # adding the same chord self.assertRaises(ValueError, M1.add_chord, Segment(A, B)) #M1.show() def test_divide_edge(self): M1 = PlanarMap() A, B, C, D = Point(0, 2), Point(2, 2), Point(0, 0), Point(2, 0) AB = Segment(A, B) M1.add_first_edge(AB) #M1.divide_edge(AB, Point(1, 2)) M1.divide_edge(Segment(A, B), Point(1, 2)) # it works self.assertEqual(M1.v(), 3) self.assertEqual(M1.e(), 2) self.assertEqual(M1.f(), 1) #M1.show() def test_parameters(self): self.assertEqual(self.M.v(), 4) self.assertEqual(self.M.e(), 4) self.assertEqual(self.M.f(), 2) def test_iteredges(self): B = Point(1, 1) Bout = list(self.M.iteroutedges(B)) Bin = list(self.M.iterinedges(B)) self.assertEqual(len(Bout), 3) self.assertEqual(len(Bin), 3) for segment in self.M.iteroutedges(B): self.assertEqual(segment.source, B) for segment in self.M.iterinedges(B): self.assertEqual(segment.target, B) def test_iteredges_connected(self): start_edge = next(self.M.iteredges()) A = set([start_edge.source, start_edge.target]) for edge in self.M.iteredges_connected(start_edge): B = set([edge.source, edge.target]) self.assertTrue(len(A & B) > 0) A.update(B) #print ( A ) def test_copy(self): T = self.M.copy() self.assertEqual(T.v(), self.M.v()) self.assertEqual(T.e(), self.M.e()) for node in T.iternodes(): self.assertTrue(self.M.has_node(node)) for edge in T.iteredges(): self.assertTrue(self.M.has_edge(edge)) def test_degree(self): self.assertEqual(self.M.degree(Point(0, 1)), 2) self.assertEqual(self.M.degree(Point(1, 1)), 3) self.assertEqual(self.M.degree(Point(0, 0)), 2) self.assertEqual(self.M.degree(Point(1, 0)), 1) def test_exceptions(self): self.assertRaises(AssertionError, self.M.add_edge, Edge(Point(0, 0), Point(1, 0))) #self.assertRaises(ValueError, self.M.add_edge, Segment(Point(0, 0), Point(0, 0))) self.assertRaises(ValueError, self.M.add_edge, Segment(Point(1, 0), Point(1, 1))) self.assertRaises(AssertionError, self.M.add_node, "A") def test_init_segment(self): M1 = PlanarMap(Segment(0, 1, 2, 3)) self.assertEqual(M1.v(), 2) self.assertEqual(M1.e(), 1) self.assertEqual(M1.f(), 1) def test_init_triangle(self): M1 = PlanarMap(Triangle(0, 0, 1, 0, 1, 1)) self.assertEqual(M1.v(), 3) self.assertEqual(M1.e(), 3) self.assertEqual(M1.f(), 2) def test_init_rectangle(self): M1 = PlanarMap(Rectangle(0, 0, 1, 2)) self.assertEqual(M1.v(), 4) self.assertEqual(M1.e(), 4) self.assertEqual(M1.f(), 2) # D # / \ # E C # | | # A-------B def test_init_polygon(self): M1 = PlanarMap(Polygon(0, 0, 2, 0, 2, 1, 1, 2, 0, 1)) self.assertEqual(M1.v(), 5) self.assertEqual(M1.e(), 5) self.assertEqual(M1.f(), 2) # D # | # A---+---B # | # C def test_mapoverlay1(self): M1 = PlanarMap(Segment(0, 1, 2, 1)) M2 = PlanarMap(Segment(1, 0, 1, 2)) M3 = M1.map_overlay(M2) self.assertEqual(M3.v(), 5) self.assertEqual(M3.e(), 4) self.assertEqual(M3.f(), 1) #print(list(M3.iternodes())) #print(list(M3.iteredges())) def tearDown(self): pass