def test_clear_color():
img = Image(100, 200)
# just to put something in there to clear.
img.draw_rectangle((-10, -10), (50, 100), fill_color='red')
img.draw_rectangle((50, 100), (110, 210), fill_color='blue')
img.save(outfile("test_image_clear_before2.png"), "png")
img.clear(color='white')
img.save(outfile("test_image_clear_after2.png"), "png")
assert np.all(np.asarray(img).flat == img.get_color_index('white'))
def test_add_colors():
img = Image(10, 10, preset_colors='BW')
assert img.get_color_names() == ['transparent', 'black', 'white']
img.add_color('light grey', (220, 220, 220))
assert img.get_color_names() == [
'transparent', 'black', 'white', 'light grey'
]
assert img.get_color_index('light grey') == 3
img.draw_rectangle((2, 2), (7, 7), fill_color='light grey')
img.save(outfile('test_image_grey.bmp'))
with pytest.raises(ValueError):
# color doesn't exist
img.draw_rectangle((2, 2), (7, 7), fill_color='red')
def test_colors():
img = Image(5, 5)
# this shold work
img.get_color_index('black')
# so should this:
img.get_color_index(0)
img.get_color_index(255)
# will round floating point numbers
# shoul dthi sbe changed?
assert img.get_color_index(2.3) == 2
with pytest.raises(ValueError):
# error if index not in 0--255
img.get_color_index(300)
with pytest.raises(ValueError):
# error if color is not in dict
img.get_color_index('something else')
with pytest.raises(ValueError):
# error if color is not anumber
img.get_color_index((1, 2, 3))
with pytest.raises(TypeError):
# error if color is unhasable
img.get_color_index(['a', 'random', 4])
