def test_make_colormap(self):
make_colormap([([0, 0, 1], 10), ([1, 1, 1], 10), ([1, 0, 0], 10)],
name='french_flag', interpolate=False)
show_colormaps('french_flag')
cmap = make_colormap([('dred', 5), ('blue', 2.0), ('orange', 0)],
name='my_cmap')
assert_almost_equal(cmap["red"][1],
np.array([0.00392157, 0.62400345, 0.62400345]))
assert_almost_equal(cmap["blue"][2],
np.array([0.00784314, 0.01098901, 0.01098901]))
assert_almost_equal(cmap["green"][3], np.array([0.01176471, 0.0, 0.0]))
def test_make_colormap(self):
make_colormap(
[([0, 0, 1], 10), ([1, 1, 1], 10), ([1, 0, 0], 10)],
name="french_flag",
interpolate=False,
)
show_colormaps("french_flag")
cmap = make_colormap([("dred", 5), ("blue", 2.0), ("orange", 0)],
name="my_cmap")
assert_almost_equal(cmap["red"][1],
np.array([0.00392157, 0.62400345, 0.62400345]))
assert_almost_equal(cmap["blue"][2],
np.array([0.00784314, 0.01098901, 0.01098901]))
assert_almost_equal(cmap["green"][3], np.array([0.01176471, 0.0, 0.0]))
def test_show_colormaps(self):
show_colormaps()
show_colormaps(subset=["jet", "cool"])
show_colormaps(subset="yt_native", filename="yt_color_maps.png")
# Test for non-existent color map
with assert_raises(AttributeError) as ex:
show_colormaps(subset="unknown", filename="yt_color_maps.png")
desired = ("show_colormaps requires subset attribute to be 'all', "
"'yt_native', or a list of valid colormap names.")
assert_equal(str(ex.exception), desired)