python-pillow/Pillow
1
1
Fork 0
mirror of https://github.com/python-pillow/Pillow.git synced 2025-01-12 10:16:17 +03:00
Code Issues Packages Projects Releases Wiki Activity

Refactor random image

Browse Source
This commit is contained in:
wiredfool 2016-12-03 14:45:05 +00:00
parent 15347b6703
commit 4b57345683
1 changed files with 11 additions and 15 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

26
Tests/test_file_jpeg.py
View File

@ -1,7 +1,6 @@
from helper import unittest, PillowTestCase, hopper, py3 from helper import unittest, PillowTestCase, hopper, py3
from helper import djpeg_available, cjpeg_available from helper import djpeg_available, cjpeg_available
import random
from io import BytesIO from io import BytesIO
import os import os
@ -29,6 +28,15 @@ class TestFileJpeg(PillowTestCase):
im.bytes = test_bytes # for testing only im.bytes = test_bytes # for testing only
return im return im
def gen_random_image(self, size, mode='RGB'):
""" Generates a very hard to compress file
:param size: tuple
:param mode: optional image mode
"""
return Image.frombytes(mode, size,
os.urandom(size[0]*size[1]*len(mode)))
def test_sanity(self): def test_sanity(self):
# internal version number # internal version number
@ -159,12 +167,7 @@ class TestFileJpeg(PillowTestCase):
def test_progressive_large_buffer_highest_quality(self): def test_progressive_large_buffer_highest_quality(self):
f = self.tempfile('temp.jpg') f = self.tempfile('temp.jpg')
if py3: im = self.gen_random_image((255,255))
a = bytes(random.randint(0, 255) for _ in range(256 * 256 * 3))
else:
a = b''.join(chr(random.randint(0, 255)) for _ in range(
256 * 256 * 3))
im = Image.frombuffer("RGB", (256, 256), a, "raw", "RGB", 0, 1)
# this requires more bytes than pixels in the image # this requires more bytes than pixels in the image
im.save(f, format="JPEG", progressive=True, quality=100) im.save(f, format="JPEG", progressive=True, quality=100)
@ -436,14 +439,7 @@ class TestFileJpeg(PillowTestCase):
self.assertEqual(tag_ids['RelatedImageLength'], 0x1002) self.assertEqual(tag_ids['RelatedImageLength'], 0x1002)
def test_MAXBLOCK_scaling(self): def test_MAXBLOCK_scaling(self):
def gen_random_image(size): im = self.gen_random_image((512, 512))
""" Generates a very hard to compress file
:param size: tuple
"""
return Image.frombytes('RGB',
size, os.urandom(size[0]*size[1] * 3))
im = gen_random_image((512, 512))
f = self.tempfile("temp.jpeg") f = self.tempfile("temp.jpeg")
im.save(f, quality=100, optimize=True) im.save(f, quality=100, optimize=True)