def setUp(self): self.graph = Graph(range(5)) self.graph.add_edges({(0, 1), (1, 2), (1, 3), (2, 3), (3, 4), (2, 4)}) self.layout = Layout(self.graph, default_location=(0.42, 0.42))