# -*- coding: utf-8 -*-
# ******************************************************************************
#
# Copyright (C) 2009 Olivier Tilloy <olivier@tilloy.net>
#
# This file is part of the pyexiv2 distribution.
#
# pyexiv2 is free software; you can redistribute it and/or
# modify it under the terms of the GNU General Public License
# as published by the Free Software Foundation; either version 2
# of the License, or (at your option) any later version.
#
# pyexiv2 is distributed in the hope that it will be useful,
# but WITHOUT ANY WARRANTY; without even the implied warranty of
# MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
# GNU General Public License for more details.
#
# You should have received a copy of the GNU General Public License
# along with pyexiv2; if not, write to the Free Software
# Foundation, Inc., 51 Franklin Street, 5th Floor, Boston, MA 02110-1301 USA.
#
# Author: Olivier Tilloy <olivier@tilloy.net>
#
# ******************************************************************************
import unittest
from pyexiv2 import ExifTag, ExifValueError, Rational
import datetime
class ImageMetadataMock(object):
tags = {}
def _set_exif_tag_value(self, key, value):
self.tags[key] = value
def _delete_exif_tag(self, key):
try:
del self.tags[key]
except KeyError:
pass
class TestExifTag(unittest.TestCase):
def test_convert_to_python_ascii(self):
xtype = 'Ascii'
# Valid values: datetimes
self.assertEqual(ExifTag._convert_to_python('2009-03-01 12:46:51', xtype, None),
datetime.datetime(2009, 03, 01, 12, 46, 51))
self.assertEqual(ExifTag._convert_to_python('2009:03:01 12:46:51', xtype, None),
datetime.datetime(2009, 03, 01, 12, 46, 51))
self.assertEqual(ExifTag._convert_to_python('2009-03-01T12:46:51Z', xtype, None),
datetime.datetime(2009, 03, 01, 12, 46, 51))
# Valid values: strings
self.assertEqual(ExifTag._convert_to_python('Some text.', xtype, None), u'Some text.')
self.assertEqual(ExifTag._convert_to_python('Some text with exotic chàräctérʐ.', xtype, None),
u'Some text with exotic chàräctérʐ.')
# Invalid values: datetimes
self.assertEqual(ExifTag._convert_to_python('2009-13-01 12:46:51', xtype, None),
u'2009-13-01 12:46:51')
self.assertEqual(ExifTag._convert_to_python('2009-12-01', xtype, None),
u'2009-12-01')
def test_convert_to_string_ascii(self):
xtype = 'Ascii'
# Valid values: datetimes
self.assertEqual(ExifTag._convert_to_string(datetime.datetime(2009, 03, 01, 12, 54, 28), xtype),
'2009:03:01 12:54:28')
self.assertEqual(ExifTag._convert_to_string(datetime.date(2009, 03, 01), xtype),
'2009:03:01 00:00:00')
# Valid values: strings
self.assertEqual(ExifTag._convert_to_string(u'Some text', xtype), 'Some text')
self.assertEqual(ExifTag._convert_to_string(u'Some text with exotic chàräctérʐ.', xtype),
'Some text with exotic chàräctérʐ.')
self.assertEqual(ExifTag._convert_to_string('Some text with exotic chàräctérʐ.', xtype),
'Some text with exotic chàräctérʐ.')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, None, xtype)
def test_convert_to_python_byte(self):
xtype = 'Byte'
# Valid values
self.assertEqual(ExifTag._convert_to_python('Some text', xtype, None), 'Some text')
def test_convert_to_string_byte(self):
xtype = 'Byte'
# Valid values
self.assertEqual(ExifTag._convert_to_string('Some text', xtype), 'Some text')
self.assertEqual(ExifTag._convert_to_string(u'Some text', xtype), 'Some text')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, None, xtype)
def test_convert_to_python_short(self):
xtype = 'Short'
# Valid values
self.assertEqual(ExifTag._convert_to_python('23', xtype, None), 23)
self.assertEqual(ExifTag._convert_to_python('+5628', xtype, None), 5628)
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'abc', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5,64', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '47.0001', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '1E3', xtype, None)
def test_convert_to_string_short(self):
xtype = 'Short'
# Valid values
self.assertEqual(ExifTag._convert_to_string(123, xtype), '123')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, -57, xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 'invalid', xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 3.14, xtype)
def test_convert_to_python_long(self):
xtype = 'Long'
# Valid values
self.assertEqual(ExifTag._convert_to_python('23', xtype, None), 23)
self.assertEqual(ExifTag._convert_to_python('+5628', xtype, None), 5628)
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'abc', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5,64', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '47.0001', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '1E3', xtype, None)
def test_convert_to_string_long(self):
xtype = 'Long'
# Valid values
self.assertEqual(ExifTag._convert_to_string(123, xtype), '123')
self.assertEqual(ExifTag._convert_to_string(678024, xtype), '678024')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, -57, xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 'invalid', xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 3.14, xtype)
def test_convert_to_python_slong(self):
xtype = 'SLong'
# Valid values
self.assertEqual(ExifTag._convert_to_python('23', xtype, None), 23)
self.assertEqual(ExifTag._convert_to_python('+5628', xtype, None), 5628)
self.assertEqual(ExifTag._convert_to_python('-437', xtype, None), -437)
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'abc', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5,64', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '47.0001', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '1E3', xtype, None)
def test_convert_to_string_slong(self):
xtype = 'SLong'
# Valid values
self.assertEqual(ExifTag._convert_to_string(123, xtype), '123')
self.assertEqual(ExifTag._convert_to_string(678024, xtype), '678024')
self.assertEqual(ExifTag._convert_to_string(-437, xtype), '-437')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 'invalid', xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 3.14, xtype)
def test_convert_to_python_rational(self):
xtype = 'Rational'
# Valid values
self.assertEqual(ExifTag._convert_to_python('5/3', xtype, None), Rational(5, 3))
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'invalid', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '-5/3', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5 / 3', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5/-3', xtype, None)
def test_convert_to_string_rational(self):
xtype = 'Rational'
# Valid values
self.assertEqual(ExifTag._convert_to_string(Rational(5, 3), xtype), '5/3')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 'invalid', xtype)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, Rational(-5, 3), xtype)
def test_convert_to_python_srational(self):
xtype = 'SRational'
# Valid values
self.assertEqual(ExifTag._convert_to_python('5/3', xtype, None), Rational(5, 3))
self.assertEqual(ExifTag._convert_to_python('-5/3', xtype, None), Rational(-5, 3))
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'invalid', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5 / 3', xtype, None)
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, '5/-3', xtype, None)
def test_convert_to_string_srational(self):
xtype = 'SRational'
# Valid values
self.assertEqual(ExifTag._convert_to_string(Rational(5, 3), xtype), '5/3')
self.assertEqual(ExifTag._convert_to_string(Rational(-5, 3), xtype), '-5/3')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 'invalid', xtype)
def test_convert_to_python_undefined(self):
xtype = 'Undefined'
# Valid values
self.assertEqual(ExifTag._convert_to_python('48 49 48 48 ', xtype, '1.00'),
u'1.00')
self.assertEqual(ExifTag._convert_to_python('3 ', xtype, 'Digital still camera'),
u'Digital still camera')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_python, 'abc', xtype, None)
def test_convert_to_string_undefined(self):
xtype = 'Undefined'
# Valid values
self.assertEqual(ExifTag._convert_to_string('48 49 48 48 ', xtype), '48 49 48 48 ')
self.assertEqual(ExifTag._convert_to_string(u'48 49 48 48 ', xtype), '48 49 48 48 ')
# Invalid values
self.failUnlessRaises(ExifValueError, ExifTag._convert_to_string, 3, xtype)
def test_set_value_no_metadata(self):
tag = ExifTag('Exif.Thumbnail.Orientation', 'Orientation',
'Orientation', 'The image orientation viewed in terms ' \
'of rows and columns.', 'Short', '1', 'top, left')
old_value = tag.value
tag.value = 2
self.failIfEqual(tag.value, old_value)
def test_set_value_with_metadata(self):
tag = ExifTag('Exif.Thumbnail.Orientation', 'Orientation',
'Orientation', 'The image orientation viewed in terms ' \
'of rows and columns.', 'Short', '1', 'top, left')
tag.metadata = ImageMetadataMock()
old_value = tag.value
tag.value = 2
self.failIfEqual(tag.value, old_value)
self.assertEqual(tag.metadata.tags[tag.key], '2')
def test_del_value_no_metadata(self):
tag = ExifTag('Exif.Thumbnail.Orientation', 'Orientation',
'Orientation', 'The image orientation viewed in terms ' \
'of rows and columns.', 'Short', '1', 'top, left')
del tag.value
self.failIf(hasattr(tag, 'value'))
def test_del_value_with_metadata(self):
tag = ExifTag('Exif.Thumbnail.Orientation', 'Orientation',
'Orientation', 'The image orientation viewed in terms ' \
'of rows and columns.', 'Short', '1', 'top, left')
tag.metadata = ImageMetadataMock()
tag.metadata._set_exif_tag_value(tag.key, tag.to_string())
self.assertEqual(tag.metadata.tags, {tag.key: '1'})
del tag.value
self.failIf(hasattr(tag, 'value'))
self.failIf(tag.metadata.tags.has_key(tag.key))
