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)