def test_get_cmaps_category_descr(self):
cmaps = get_cmaps()
self.assertEqual(
cmaps[0][1],
'For many applications, a perceptually uniform colormap is the best choice - '
'one in which equal steps in data are perceived as equal steps in the color '
'space')
def test_get_cmaps_categories(self):
cmaps = get_cmaps()
self.assertGreaterEqual(len(cmaps), 7)
self.assertEqual(cmaps[0][0], 'Perceptually Uniform Sequential')
self.assertEqual(cmaps[1][0], 'Sequential 1')
self.assertEqual(cmaps[2][0], 'Sequential 2')
self.assertEqual(cmaps[3][0], 'Diverging')
self.assertEqual(cmaps[4][0], 'Qualitative')
self.assertEqual(cmaps[5][0], 'Ocean')
self.assertEqual(cmaps[6][0], 'Miscellaneous')
def test_cmocean_category(self):
cmaps = get_cmaps()
category_tuple = cmaps[5][2]
self.assertEqual(len(category_tuple), 36)
self.assertEqual(category_tuple[0][0], 'thermal')
self.assertEqual(
category_tuple[0][1],
'iVBORw0KGgoAAAANSUhEUgAAAQAAAAACCAYAAAC3zQLZAAAA2klEQVR4nO2S6xHDMAiDP+FROkL3Xy30RwBju52g8V0OIcnyKxqvtwsD5SfAUPZNE6M4VR2hJTdQeBX6UhlY8xgDY8V24A15pMuIXcQHJo4qwOQYIHlojpT6zWnzqDxRo+/+zFZbR7H2Tx3WvMPf1qDvq+17zz/m7TV97YxHbefEW27ve+7Oe9xZZu3cdXCdr17XokurfvYOcmTXxHJkE2P32ei8eVxww1WJecRlBxZr/cndj+T5MKULbzqm5pnY56MFjnkmPH7cb7xXzvR49RRO3njGM57xt+MDC391Pt11tkYAAAAASUVORK5CYII='
)
self.assertEqual(category_tuple[1][0], 'thermal_alpha')
self.assertEqual(category_tuple[2][0], 'haline')
self.assertEqual(category_tuple[3][0], 'haline_alpha')
def test_get_cmaps_category_tuples(self):
cmaps = get_cmaps()
category_tuple = cmaps[0][2]
self.assertEqual(len(category_tuple), 8)
self.assertEqual(category_tuple[0][0], 'viridis')
self.assertEqual(
category_tuple[0][1],
'iVBORw0KGgoAAAANSUhEUgAAAQAAAAACCAYAAAC3zQLZAAAAzklEQVR4nO2TQZLFIAhEX7dXmyPM/Y8SZwEqMcnU3/9QZTU8GszC6Ee/HQlk5FAsJIENqVGv/piZ3uqf3nX6Vtd+l8D8UwNOLhZL3+BLh796OXvMdWaqtrrqnZ/tjvuZT/0XxnN/5f25z9X7tIMTKzV7/5yrME3NHoPlUzvplgOevOcz6ZO5eCqzOmark1nHDQveHuuYaazZkTcdmE110HJu6doR3tgfPHyL51zNc0fd2xjf0vPukUPL36YBTcpcWArFyY0RTca88cYbXxt/gUOJC8yRF1kAAAAASUVORK5CYII='
)
self.assertEqual(category_tuple[1][0], 'viridis_alpha')
self.assertEqual(category_tuple[2][0], 'inferno')
self.assertEqual(category_tuple[3][0], 'inferno_alpha')
def main():
cmaps = get_cmaps()
html_head = '<!DOCTYPE html>\n' + \
'<html lang="en">\n' + \
'<head>' + \
'<meta charset="UTF-8">' + \
'<title>matplotlib Color Maps</title>' + \
'</head>\n' + \
'<body style="padding: 0.2em">\n'
html_body = ''
html_foot = '</body>\n' + \
'</html>\n'
for cmap_cat, cmap_desc, cmap_bars in cmaps:
html_body += ' <h2>%s</h2>\n' % cmap_cat
html_body += ' <p><i>%s</i></p>\n' % cmap_desc
html_body += ' <table style=border: 0">\n'
for cmap_bar in cmap_bars:
cmap_name, cmap_data = cmap_bar
cmap_image = '<img src="data:image/png;base64,%s" width="100%%" height="20px"/>' % cmap_data
html_body += ' <tr><td style="width: 5em">%s:</td><td style="width: 40em">%s</td></tr>\n' % (
cmap_name, cmap_image)
html_body += ' </table>\n'
html_page = html_head + html_body + html_foot
html_filename = 'test_cmaps.html'
with open(html_filename, 'w') as fp:
fp.write(html_page)
import webbrowser
webbrowser.open_new_tab(html_filename)
def test_get_cmaps_returns_singleton(self):
cmaps = get_cmaps()
self.assertIs(cmaps, get_cmaps())
self.assertIs(cmaps, get_cmaps())
def test_get_cmaps_retruns_equal_size_recs(self):
cmaps = get_cmaps()
rec_len = len(cmaps[0])
self.assertEqual(rec_len, 3)
for cmap in cmaps:
self.assertEqual(len(cmap), rec_len)