65 lines
2.0 KiB
Python
65 lines
2.0 KiB
Python
# -*- coding:utf-8 -*-
|
|
|
|
import os
|
|
import pytest
|
|
from PIL import Image
|
|
|
|
from sigal import init_logging
|
|
from sigal.image import generate_image, generate_thumbnail, get_exif_tags
|
|
|
|
CURRENT_DIR = os.path.dirname(__file__)
|
|
TEST_IMAGE = 'exo20101028-b-full.jpg'
|
|
SRCFILE = os.path.join(CURRENT_DIR, 'sample', 'pictures', 'dir2', TEST_IMAGE)
|
|
|
|
|
|
def test_generate_image(tmpdir):
|
|
"Test the generate_image function."
|
|
|
|
dstfile = str(tmpdir.join(TEST_IMAGE))
|
|
for size in [(600, 600), (300, 200)]:
|
|
generate_image(SRCFILE, dstfile, size, None, method='ResizeToFill')
|
|
im = Image.open(dstfile)
|
|
assert im.size == size
|
|
|
|
|
|
def test_generate_image_processor(tmpdir):
|
|
"Test generate_image with a wrong processor name."
|
|
|
|
init_logging()
|
|
dstfile = str(tmpdir.join(TEST_IMAGE))
|
|
with pytest.raises(SystemExit):
|
|
generate_image(SRCFILE, dstfile, (200, 200), None,
|
|
method='WrongMethod')
|
|
|
|
|
|
def test_generate_thumbnail(tmpdir):
|
|
"Test the generate_thumbnail function."
|
|
|
|
dstfile = str(tmpdir.join(TEST_IMAGE))
|
|
for size in [(200, 150), (150, 200)]:
|
|
generate_thumbnail(SRCFILE, dstfile, size, None)
|
|
im = Image.open(dstfile)
|
|
assert im.size == size
|
|
|
|
for size, thumb_size in [((200, 150), (185, 150)),
|
|
((150, 200), (150, 122))]:
|
|
generate_thumbnail(SRCFILE, dstfile, size, None, fit=False)
|
|
im = Image.open(dstfile)
|
|
assert im.size == thumb_size
|
|
|
|
|
|
def test_exif_copy(tmpdir):
|
|
"Test if EXIF data can transfered copied to the resized image."
|
|
|
|
test_image = '11.jpg'
|
|
src_file = os.path.join(CURRENT_DIR, 'sample', 'pictures', 'dir1', 'test1',
|
|
test_image)
|
|
dst_file = str(tmpdir.join(test_image))
|
|
|
|
generate_image(src_file, dst_file, (300, 400), None, copy_exif_data=True)
|
|
tags = get_exif_tags(dst_file)
|
|
assert tags['simple']['iso'] == 50
|
|
|
|
generate_image(src_file, dst_file, (300, 400), None, copy_exif_data=False)
|
|
assert not get_exif_tags(dst_file)
|