def test_get_weight(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     vertex.add_neighbor(data='b')
     self.assertEqual(vertex.get_weight('b'), 1)
     vertex.add_neighbor(data='b', weight=5)
     self.assertEqual(vertex.get_weight('b'), 5)
 def test_remove_neighbor(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     vertex.add_neighbor(data='b')
     self.assertEqual(set(list(vertex.adjacent.keys())), set(['b']))
     self.assertEqual(vertex.adjacent.get('b'), 1)
     self.assertEqual(vertex.remove_neighbor(data='b'), True)
 def test_remove_neighbor_key_error(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     vertex.add_neighbor(data='b')
     self.assertEqual(set(list(vertex.adjacent.keys())), set(['b']))
     self.assertEqual(vertex.adjacent.get('b'), 1)
     with self.assertRaises(KeyError) as error:
         self.assertEqual(vertex.remove_neighbor(data='invalid_key'), False)
 def test_get_weight_keyerror(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     vertex.add_neighbor(data='b')
     with self.assertRaises(KeyError) as error:
         self.assertEqual(vertex.get_weight('c'), None)
 def test_get_neighbor(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     vertex.add_neighbor(data='b')
     self.assertEqual(vertex.get_neighbors(), ['b'])
 def test_add_neighbor_negative_weight(self):
     vertex = Vertex(data='a')
     self.assertEqual(set(list(vertex.adjacent.keys())), set([]))
     with self.assertRaises(TypeError) as error:
         vertex.add_neighbor(data='b', weight=-5)