def test_unique_colors_palette(self): from jicbioimage.core.util.color import unique_color_palette color_dict = unique_color_palette([0, 1, 16777215]) self.assertEqual(len(color_dict), 3) self.assertEqual(color_dict[0], (0, 0, 0)) self.assertEqual(color_dict[1], (0, 0, 1)) self.assertEqual(color_dict[16777215], (255, 255, 255))
def test_unique_colors_palette(self): from jicbioimage.core.util.color import unique_color_palette color_dict = unique_color_palette([0, 1, 16777215]) self.assertEqual(len(color_dict), 3) self.assertEqual(color_dict[0], (0, 0, 0)) self.assertEqual(color_dict[1], (0, 0, 1)) self.assertEqual(color_dict[16777215], (255, 255, 255))
def unique_color_array(array): """Return a RGB unique color array. Assigning a unique RGB color value to each unique element of the input array and return an array of shape (array.shape, 3). :param array: input numpy.array :returns: numpy.array """ unique_identifiers = set(np.unique(array)) color_dict = unique_color_palette(unique_identifiers) return color_array(array, color_dict)