import WrightTools as wt
from matplotlib import cm
import matplotlib.pyplot as plt
cmap = cm.get_cmap('viridis')
wt.artists.plot_colormap_components(cmap)
wt.artists.savefig('viridis_cmap.png')