assign_edge_colors function

Assign edge colors based on source node colors