def test_create_legend(self):
"""Test that this can run and that file is generated
"""
cpt = CPT(test_filename)
legend_filename = cpt.create_legend_for_colour_coding(-10, 20,
'Test legend')
assert os.path.exists(legend_filename)
# Cleanup
os.remove(legend_filename)
