def test_id(self):
temp1 = Vertex("A")
temp1.add_neighbours({"A": 10})
self.assertEqual(temp1.id, "A")
temp2 = Vertex("B")
temp2.add_neighbours({"A": 20})
self.assertEqual(temp2.id, "B")
def test_add_neighbours(self):
temp = Vertex("C")
new_neighbour = {"B": 10}
temp.add_neighbours(new_neighbour)
self.assertDictEqual(temp.neighbours, new_neighbour)
