def testFloatRGB(self):
lut = numpy.array([[1.0, 0, 0], [0.5, 0, 0], [0, 0, 0]], dtype="float")
colors.registerLUT("test_type", lut)
colormap = colors.Colormap(name="test_type")
lut = colormap.getNColors(3)
self.assertEqual(lut.shape, (3, 4))
self.assertEqual(lut[0, 0], 255)
def testUint8(self):
lut = numpy.array([[255, 0, 0], [200, 0, 0], [150, 0, 0]], dtype="uint")
colors.registerLUT("test_type", lut)
colormap = colors.Colormap(name="test_type")
lut = colormap.getNColors(3)
self.assertEqual(lut.shape, (3, 4))
self.assertEqual(lut[0, 0], 255)
def testFloatRGB(self):
lut = numpy.array([[1.0, 0, 0], [0.5, 0, 0], [0, 0, 0]], dtype="float")
colors.registerLUT("test_type", lut)
colormap = colors.Colormap(name="test_type")
lut = colormap.getNColors(3)
self.assertEqual(lut.shape, (3, 4))
self.assertEqual(lut[0, 0], 255)
def testUint8(self):
lut = numpy.array([[255, 0, 0], [200, 0, 0], [150, 0, 0]], dtype="uint")
colors.registerLUT("test_type", lut)
colormap = colors.Colormap(name="test_type")
lut = colormap.getNColors(3)
self.assertEqual(lut.shape, (3, 4))
self.assertEqual(lut[0, 0], 255)
def setUp(self):
# Save preferred colormaps
lut = numpy.arange(8 * 3)
lut.shape = -1, 3
lut = lut / (8.0 * 3)
colors.registerLUT("test_8", colors=lut, cursor_color='blue')
def setUp(self):
# Save preferred colormaps
lut = numpy.arange(8 * 3)
lut.shape = -1, 3
lut = lut / (8.0 * 3)
colors.registerLUT("test_8", colors=lut, cursor_color='blue')