def test_set_colors_updates_existing_colors(self):
call_graph = CallGraph()
call_graph.set_colors(yellow=2, red=3)
call_graph.set_colors(yellow=5, red=6)
self.assertEqual(call_graph.get_yellow(), 5)
self.assertEqual(call_graph.get_red(), 6)
def test_set_colors_sets_returned_colors(self):
call_graph = CallGraph()
call_graph.set_colors(yellow=2, red=3)
self.assertEqual(call_graph.get_yellow(), 2)
self.assertEqual(call_graph.get_red(), 3)