def test_add_remove_edge(self): node1 = Node(0, 0) node2 = Node(0, 1) node1.connect(node2) assert node1.is_connected_to(node2) assert node2.is_connected_to(node1) node1.disconnect(node2) assert not node1.is_connected_to(node2) assert not node2.is_connected_to(node1)