def test_init_with_val_range():
"""
Can we initialize with a val_range?
"""
cm = ColorMap("hsv", val_range=(0, 511))
assert np.all(cm.get_colors((0, 200, 511)) == [[255, 0, 0],
[0, 255, 81],
[255, 0, 24]])
def test_init():
"""
Can we even initialize one?
"""
cm = ColorMap("hsv")
assert np.all(cm.get_colors((0, 100, 255)) == [[255, 0, 0],
[0, 255, 81],
[255, 0, 24]])
def test_init_with_val_range():
"""
Can we initialize with a val_range?
"""
cm = ColorMap("hsv", val_range=(0, 511))
assert np.all(
cm.get_colors((0, 200,
511)) == [[255, 0, 0], [0, 255, 81], [255, 0, 24]])
def test_init():
"""
Can we even initialize one?
"""
cm = ColorMap("hsv")
assert np.all(
cm.get_colors((0, 100,
255)) == [[255, 0, 0], [0, 255, 81], [255, 0, 24]])
def test_reversed_float_val_range():
"""
Can we set a reversed floating point val_range, and does that range
accurately map to the color positions?
"""
cm = ColorMap("hsv")
# testing our entire range.
cm.val_range = (20.0, 10.0)
for i, v in enumerate(np.linspace(20.0, 10.0, 256)):
assert np.all(NamedColorMaps['hsv'][i] == cm.get_colors((v,)))
def test_reversed_float_val_range():
"""
Can we set a reversed floating point val_range, and does that range
accurately map to the color positions?
"""
cm = ColorMap("hsv")
# testing our entire range.
cm.val_range = (20.0, 10.0)
for i, v in enumerate(np.linspace(20.0, 10.0, 256)):
assert np.all(NamedColorMaps['hsv'][i] == cm.get_colors((v, )))
def test_integer_val_range():
"""
Can we set an integer val_range, and does that range accurately map
to the color positions?
"""
cm = ColorMap("hsv")
# testing our entire range.
# our range is 1000 + (256 - 1) * 4 which should match the size of the
# colormap
cm.val_range = (1000, 2020)
for i, v in enumerate(range(1000, 2024, 4)):
assert np.all(NamedColorMaps['hsv'][i] == cm.get_colors((v,)))
def test_integer_val_range():
"""
Can we set an integer val_range, and does that range accurately map
to the color positions?
"""
cm = ColorMap("hsv")
# testing our entire range.
# our range is 1000 + (256 - 1) * 4 which should match the size of the
# colormap
cm.val_range = (1000, 2020)
for i, v in enumerate(range(1000, 2024, 4)):
assert np.all(NamedColorMaps['hsv'][i] == cm.get_colors((v, )))
