added new tests for n_colors and get_color_dict