66 lines
1.6 KiB
Python
66 lines
1.6 KiB
Python
# -*- coding:utf-8 -*-
|
|
|
|
import os
|
|
from tempfile import mkdtemp
|
|
from shutil import rmtree
|
|
|
|
try:
|
|
import unittest2 as unittest
|
|
except ImportError:
|
|
import unittest # NOQA
|
|
|
|
try:
|
|
import pyexiv2
|
|
except ImportError:
|
|
pyexiv2 = False
|
|
|
|
from sigal.image import copy_exif, Image
|
|
|
|
CURRENT_DIR = os.path.dirname(__file__)
|
|
TEST_IMAGE = 'exo20101028-b-full.jpg'
|
|
|
|
|
|
class TestImage(unittest.TestCase):
|
|
"Test the Image class."
|
|
|
|
def setUp(self):
|
|
self.temp_path = mkdtemp()
|
|
self.srcfile = os.path.join(CURRENT_DIR, 'sample', 'dir2', TEST_IMAGE)
|
|
self.dstfile = os.path.join(self.temp_path, TEST_IMAGE)
|
|
self.img = Image(self.srcfile)
|
|
|
|
def tearDown(self):
|
|
rmtree(self.temp_path)
|
|
|
|
def test_imgname(self):
|
|
self.assertEqual(self.img.imgname, TEST_IMAGE)
|
|
|
|
def test_save(self):
|
|
self.img.save(self.dstfile)
|
|
self.assertTrue(os.path.isfile(self.dstfile))
|
|
|
|
|
|
@unittest.skipUnless(pyexiv2, "pyexiv2 isn't installed")
|
|
class TestExif(unittest.TestCase):
|
|
"Test the copy of exif metadata with pyexiv2."
|
|
|
|
def setUp(self):
|
|
self.temp_path = mkdtemp()
|
|
self.srcfile = os.path.join(CURRENT_DIR, 'sample', 'dir2', TEST_IMAGE)
|
|
self.dstfile = os.path.join(self.temp_path, TEST_IMAGE)
|
|
|
|
img = Image(self.srcfile)
|
|
img.save(self.dstfile)
|
|
copy_exif(self.srcfile, self.dstfile)
|
|
|
|
def tearDown(self):
|
|
rmtree(self.temp_path)
|
|
|
|
def test_exif(self):
|
|
src = pyexiv2.ImageMetadata(self.srcfile)
|
|
dst = pyexiv2.ImageMetadata(self.dstfile)
|
|
src.read()
|
|
dst.read()
|
|
|
|
self.assertListEqual(src.keys(), dst.keys())
|