|
@@ -186,13 +186,13 @@ class GeneratorTest(unittest.TestCase):
|
|
|
image = PIL.Image.open(image_stream)
|
|
|
|
|
|
# Verify image size, format, and mode.
|
|
|
self.assertEqual(image.size[0], 240)
|
|
|
self.assertEqual(image.size[1], 240)
|
|
|
self.assertEqual(image.format, "PNG")
|
|
|
self.assertEqual(image.mode, "RGB")
|
|
|
self.assertEqual(image.mode, "RGBA")
|
|
|
|
|
|
def test_generate_ascii(self):
|
|
|
"""
|
|
|
Tests the generated identicon in ASCII format.
|
|
|
"""
|
|
|
|
|
@@ -356,16 +356,16 @@ class GeneratorTest(unittest.TestCase):
|
|
|
|
|
|
# Set-up parameters equivalent as used for samples.
|
|
|
width = 200
|
|
|
height = 200
|
|
|
padding = (20, 20, 20, 20)
|
|
|
|
|
|
# Load the reference images, making sure they're in RGB mode.
|
|
|
test1_ref = PIL.Image.open("tests/samples/test1.png").convert(mode="RGB")
|
|
|
test2_ref = PIL.Image.open("tests/samples/test2.png").convert(mode="RGB")
|
|
|
test3_ref = PIL.Image.open("tests/samples/test3.png").convert(mode="RGB")
|
|
|
# Load the reference images, making sure they're in RGBA mode.
|
|
|
test1_ref = PIL.Image.open("tests/samples/test1.png").convert(mode="RGBA")
|
|
|
test2_ref = PIL.Image.open("tests/samples/test2.png").convert(mode="RGBA")
|
|
|
test3_ref = PIL.Image.open("tests/samples/test3.png").convert(mode="RGBA")
|
|
|
|
|
|
# Set-up the Generator.
|
|
|
generator = Generator(5, 5, foreground=foreground, background=background)
|
|
|
|
|
|
# Generate first test identicon.
|
|
|
raw_image = generator.generate("test1", width, height, padding=padding)
|
|
@@ -386,13 +386,13 @@ class GeneratorTest(unittest.TestCase):
|
|
|
diff1 = PIL.ImageChops.difference(test1, test1_ref)
|
|
|
diff2 = PIL.ImageChops.difference(test2, test2_ref)
|
|
|
diff3 = PIL.ImageChops.difference(test3, test3_ref)
|
|
|
|
|
|
# Verify that all the diffs are essentially black (i.e. no differences
|
|
|
# between generated identicons and reference samples).
|
|
|
expected_extrema = ((0, 0), (0, 0), (0, 0))
|
|
|
expected_extrema = ((0, 0), (0, 0), (0, 0), (0, 0))
|
|
|
|
|
|
self.assertEqual(diff1.getextrema(), expected_extrema)
|
|
|
self.assertEqual(diff2.getextrema(), expected_extrema)
|
|
|
self.assertEqual(diff3.getextrema(), expected_extrema)
|
|
|
|
|
|
if __name__ == '__main__':
|