def test_three_color(self):
generator = GradientPaletteGenerator('#f00', '#000', '#fff')
np.testing.assert_equal(generator.getRGB(.5), (0, 0, 0))
np.testing.assert_equal(generator.getRGB(np.nan), NAN_GREY)
np.testing.assert_equal(generator.getRGB([0, .5, 1]),
[(255, 0, 0),
(0, 0, 0),
(255, 255, 255)])
def test_two_color(self):
generator = GradientPaletteGenerator('#000', '#fff')
np.testing.assert_equal(generator.getRGB(.5), (128, 128, 128))
np.testing.assert_equal(generator.getRGB(np.nan), NAN_GREY)
np.testing.assert_equal(generator.getRGB([0, .5, np.nan]),
[(0, 0, 0),
(128, 128, 128),
NAN_GREY])
def test_three_color(self):
generator = GradientPaletteGenerator('#f00', '#000', '#fff')
for float_values, rgb in ((.5, (0, 0, 0)),
(np.nan, NAN_GREY),
((0, .5, 1), ((255, 0, 0),
(0, 0, 0),
(255, 255, 255)))):
np.testing.assert_equal(generator.getRGB(float_values), rgb)
def test_two_color(self):
generator = GradientPaletteGenerator('#000', '#fff')
for float_values, rgb in ((.5, (128, 128, 128)),
(np.nan, NAN_GREY),
((0, .5, np.nan), ((0, 0, 0),
(128, 128, 128),
NAN_GREY))):
np.testing.assert_equal(generator.getRGB(float_values), rgb)
def test_three_color(self):
generator = GradientPaletteGenerator("#f00", "#000", "#fff")
for float_values, rgb in (
(0.5, (0, 0, 0)),
(np.nan, NAN_GREY),
((0, 0.5, 1), ((255, 0, 0), (0, 0, 0), (255, 255, 255))),
):
np.testing.assert_equal(generator.getRGB(float_values), rgb)
def test_two_color(self):
generator = GradientPaletteGenerator("#000", "#fff")
for float_values, rgb in (
(0.5, (128, 128, 128)),
(np.nan, NAN_GREY),
((0, 0.5, np.nan), ((0, 0, 0), (128, 128, 128), NAN_GREY)),
):
np.testing.assert_equal(generator.getRGB(float_values), rgb)
