def test_remove(self):
"""test_remove: Test removing a node
"""
node_1 = WeightedNode(1, [2], {(1, 2): 2})
node_2 = WeightedNode(2, [1], {(2, 1): 2})
graph = WeightedGraph([node_1, node_2])
graph.remove_node(node_1)
self.assertEqual(graph.vertices, set({2}))
self.assertEqual(graph.edges, {2: set({})})
self.assertEqual(graph.weights, {})
