def testAddAndRemoveEdge(self): agent = Agent(n=10, p = 0, topology='ErdosRenyi') self.assertFalse(agent.graph.are_connected(0,1)) agent.addEdge(0,1) self.assertTrue(agent.graph.are_connected(0,1)) agent.removeEdge(0,1) self.assertFalse(agent.graph.are_connected(0,1))