def test_assert_rectangle_at_passes(self):
backCol = (123, 45, 6)
image = Image.new('RGB', (20, 10), backCol)
rectCol = (234, 56, 7)
draw_rectangle(image, 2, 2, 17, 7, rectCol)
assert_rectangle_at(image, (2, 2, 17, 7), rectCol, backCol)
# and again, with rogue pixels just off each corner
image.putpixel((1, 1), rectCol)
image.putpixel((18, 1), rectCol)
image.putpixel((1, 8), rectCol)
image.putpixel((18, 8), rectCol)
assert_rectangle_at(image, (2, 2, 17, 7), rectCol, backCol)
def test_assert_rectangle_at_fails_if_colors_same(self):
color = (123, 45, 6)
image = Image.new('RGB', (20, 10), color)
assertion = lambda: assert_rectangle_at(
image, (1, 1, 18, 8), color, color)
expectedMsg = "colors are same %s. Broken test?" % (color,)
self.assertRaises(assertion, AssertionError, expectedMsg)
def test_assert_rectangle_at_fails_if_degenerate(self):
rectCol = (255, 255, 0)
backCol = (0, 0, 255)
image = Image.new('RGB', (20, 10), backCol)
window_ords = [
(1, 1, 1, 8),
(1, 1, 2, 8),
(1, 1, 8, 1),
(1, 1, 8, 2),
]
for ords in window_ords:
assertion = lambda: assert_rectangle_at(
image, ords, rectCol, backCol)
expectedMsg = "degenerate rect %d,%d %d,%d. Broken test?" % ords
self.assertRaises(assertion, AssertionError, expectedMsg)
def test_assert_rectangle_at_fails_if_touch_edge(self):
backCol = (0, 0, 255)
rectCol = (255, 255, 0)
image = Image.new('RGB', (20, 10), backCol)
window_ords = [
(0, 1, 18, 8),
(-1, 1, 18, 8),
(1, 0, 18, 8),
(1, -1, 18, 8),
(1, 1, 19, 8),
(1, 1, 20, 8),
(1, 1, 18, 9),
(1, 1, 18, 10),
]
for ords in window_ords:
assertion = lambda: assert_rectangle_at(
image, ords, rectCol, backCol)
expectedMsg = "rect %d,%d %d,%d touches edge of (20, 10). " \
"Broken test?" % ords
self.assertRaises(assertion, AssertionError, expectedMsg)
def test_assert_rectangle_at_fails(self):
backCol = (123, 45, 6)
rectCol = (234, 56, 7)
# if any one of these pixels are set, the assert should raise
roguePixels = [
(2, 2, backCol),
(2, 4, backCol),
(2, 7, backCol),
(9, 2, backCol),
(9, 7, backCol),
(17, 2, backCol),
(17, 4, backCol),
(17, 7, backCol),
(2, 1, rectCol),
(9, 1, rectCol),
(17, 1, rectCol),
(2, 8, rectCol),
(9, 8, rectCol),
(17, 8, rectCol),
(1, 2, rectCol),
(1, 4, rectCol),
(1, 7, rectCol),
(18, 2, rectCol),
(18, 4, rectCol),
(18, 7, rectCol),
]
for x, y, col in roguePixels:
image = Image.new('RGB', (20, 10), backCol)
draw_rectangle(image, 2, 2, 17, 7, rectCol)
image.putpixel((x, y), col)
assertion = lambda: assert_rectangle_at( \
image, (2, 2, 17, 7), rectCol, backCol)
expectedMsg = "rectangle 2,2 17,7 bad, eg at %d,%d" % (x, y)
e = self.assertRaises(assertion, AssertionError)
self.assertTrue(e.message.startswith(expectedMsg), \
'assert_rectange_at raised with bad message')
def test_assert_rectangle_at_with_rgba(self):
backCol = (123, 45, 6)
rectCol = (234, 56, 7)
image = Image.new('RGBA', (20, 10), backCol)
draw_rectangle(image, 2, 2, 17, 7, rectCol)
assert_rectangle_at(image, (2, 2, 17, 7), rectCol, backCol)
