Fixed EXIF unit tests.

1 files changed, 319 insertions, 0 deletions

diff --git a/test/exif.py b/test/exif.py
new file mode 100644
index 0000000..d32e30d
--- /dev/null
+++ b/test/exif.py
@@ -0,0 +1,319 @@
+# -*- 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 ExifTagMock(ExifTag):
+
+    def __init__(self, key, type, fvalue=''):
+        super(ExifTagMock, self).__init__(key, '', '', '', type, '', fvalue)
+
+    def _init_values(self):
+        pass
+
+
+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):
+        type = 'Ascii'
+
+        # Valid values: datetimes
+        tag = ExifTagMock('Exif.Image.DateTime', type)
+        self.assertEqual(tag._convert_to_python('2009-03-01 12:46:51'),
+                         datetime.datetime(2009, 03, 01, 12, 46, 51))
+        self.assertEqual(tag._convert_to_python('2009:03:01 12:46:51'),
+                         datetime.datetime(2009, 03, 01, 12, 46, 51))
+        self.assertEqual(tag._convert_to_python('2009-03-01T12:46:51Z'),
+                         datetime.datetime(2009, 03, 01, 12, 46, 51))
+
+        # Valid values: dates
+        tag = ExifTagMock('Exif.GPSInfo.GPSDateStamp', type)
+        self.assertEqual(tag._convert_to_python('2009:08:04'),
+                         datetime.date(2009, 8, 4))
+
+        # Valid values: strings
+        tag = ExifTagMock('Exif.Image.Copyright', type)
+        self.assertEqual(tag._convert_to_python('Some text.'), 'Some text.')
+        self.assertEqual(tag._convert_to_python(u'Some text with exotic chàräctérʐ.'),
+                         u'Some text with exotic chàräctérʐ.')
+
+        # Invalid values: datetimes
+        tag = ExifTagMock('Exif.Image.DateTime', type)
+        self.assertEqual(tag._convert_to_python('2009-13-01 12:46:51'),
+                         u'2009-13-01 12:46:51')
+        self.assertEqual(tag._convert_to_python('2009-12-01'), u'2009-12-01')
+
+        # Invalid values: dates
+        tag = ExifTagMock('Exif.GPSInfo.GPSDateStamp', type)
+        self.assertEqual(tag._convert_to_python('2009:13:01'), u'2009:13:01')
+        self.assertEqual(tag._convert_to_python('2009-12-01'), u'2009-12-01')
+
+    def test_convert_to_string_ascii(self):
+        type = 'Ascii'
+
+        # Valid values: datetimes
+        tag = ExifTagMock('Exif.Image.DateTime', type)
+        self.assertEqual(tag._convert_to_string(datetime.datetime(2009, 03, 01, 12, 54, 28)),
+                         '2009:03:01 12:54:28')
+        self.assertEqual(tag._convert_to_string(datetime.date(2009, 03, 01)),
+                         '2009:03:01 00:00:00')
+
+        # Valid values: dates
+        tag = ExifTagMock('Exif.GPSInfo.GPSDateStamp', type)
+        self.assertEqual(tag._convert_to_string(datetime.date(2009, 03, 01)),
+                         '2009:03:01')
+
+        # Valid values: strings
+        tag = ExifTagMock('Exif.Image.Copyright', type)
+        self.assertEqual(tag._convert_to_string(u'Some text'), 'Some text')
+        self.assertEqual(tag._convert_to_string(u'Some text with exotic chàräctérʐ.'),
+                         'Some text with exotic chàräctérʐ.')
+        self.assertEqual(tag._convert_to_string('Some text with exotic chàräctérʐ.'),
+                         'Some text with exotic chàräctérʐ.')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, None)
+
+    def test_convert_to_python_byte(self):
+        type = 'Byte'
+
+        # Valid values
+        tag = ExifTagMock('Exif.GPSInfo.GPSVersionID', type)
+        self.assertEqual(tag._convert_to_python('D'), 'D')
+
+    def test_convert_to_string_byte(self):
+        type = 'Byte'
+
+        # Valid values
+        tag = ExifTagMock('Exif.GPSInfo.GPSVersionID', type)
+        self.assertEqual(tag._convert_to_string('Some text'), 'Some text')
+        self.assertEqual(tag._convert_to_string(u'Some text'), 'Some text')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, None)
+
+    def test_convert_to_python_short(self):
+        type = 'Short'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.BitsPerSample', type)
+        self.assertEqual(tag._convert_to_python('8'), 8)
+        self.assertEqual(tag._convert_to_python('+5628'), 5628)
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, 'abc')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5,64')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '47.0001')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '1E3')
+
+    def test_convert_to_string_short(self):
+        type = 'Short'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.BitsPerSample', type)
+        self.assertEqual(tag._convert_to_string(123), '123')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, -57)
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 'invalid')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 3.14)
+
+    def test_convert_to_python_long(self):
+        type = 'Long'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.ImageWidth', type)
+        self.assertEqual(tag._convert_to_python('8'), 8)
+        self.assertEqual(tag._convert_to_python('+5628'), 5628)
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, 'abc')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5,64')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '47.0001')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '1E3')
+
+    def test_convert_to_string_long(self):
+        type = 'Long'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.ImageWidth', type)
+        self.assertEqual(tag._convert_to_string(123), '123')
+        self.assertEqual(tag._convert_to_string(678024), '678024')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, -57)
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 'invalid')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 3.14)
+
+    def test_convert_to_python_slong(self):
+        type = 'SLong'
+
+        # Valid values
+        tag = ExifTagMock('Exif.OlympusCs.ManometerReading', type)
+        self.assertEqual(tag._convert_to_python('23'), 23)
+        self.assertEqual(tag._convert_to_python('+5628'), 5628)
+        self.assertEqual(tag._convert_to_python('-437'), -437)
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, 'abc')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5,64')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '47.0001')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '1E3')
+
+    def test_convert_to_string_slong(self):
+        type = 'SLong'
+
+        # Valid values
+        tag = ExifTagMock('Exif.OlympusCs.ManometerReading', type)
+        self.assertEqual(tag._convert_to_string(123), '123')
+        self.assertEqual(tag._convert_to_string(678024), '678024')
+        self.assertEqual(tag._convert_to_string(-437), '-437')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 'invalid')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 3.14)
+
+    def test_convert_to_python_rational(self):
+        type = 'Rational'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.XResolution', type)
+        self.assertEqual(tag._convert_to_python('5/3'), Rational(5, 3))
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, 'invalid')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '-5/3')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5 / 3')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5/-3')
+
+    def test_convert_to_string_rational(self):
+        type = 'Rational'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.XResolution', type)
+        self.assertEqual(tag._convert_to_string(Rational(5, 3)), '5/3')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 'invalid')
+        self.failUnlessRaises(ExifValueError,
+                              tag._convert_to_string, Rational(-5, 3))
+
+    def test_convert_to_python_srational(self):
+        type = 'SRational'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.BaselineExposure', type)
+        self.assertEqual(tag._convert_to_python('5/3'), Rational(5, 3))
+        self.assertEqual(tag._convert_to_python('-5/3'), Rational(-5, 3))
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, 'invalid')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5 / 3')
+        self.failUnlessRaises(ExifValueError, tag._convert_to_python, '5/-3')
+
+    def test_convert_to_string_srational(self):
+        type = 'SRational'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Image.BaselineExposure', type)
+        self.assertEqual(tag._convert_to_string(Rational(5, 3)), '5/3')
+        self.assertEqual(tag._convert_to_string(Rational(-5, 3)), '-5/3')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 'invalid')
+
+    def test_convert_to_python_undefined(self):
+        type = 'Undefined'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Photo.ExifVersion', type, '1.00')
+        self.assertEqual(tag._convert_to_python('48 49 48 48 '), u'1.00')
+        tag = ExifTagMock('Exif.Photo.DeviceSettingDescription', type, 'Digital still camera')
+        self.assertEqual(tag._convert_to_python('3 '), u'Digital still camera')
+
+    def test_convert_to_string_undefined(self):
+        type = 'Undefined'
+
+        # Valid values
+        tag = ExifTagMock('Exif.Photo.ExifVersion', type)
+        self.assertEqual(tag._convert_to_string('48 49 48 48 '), '48 49 48 48 ')
+        self.assertEqual(tag._convert_to_string(u'48 49 48 48 '), '48 49 48 48 ')
+
+        # Invalid values
+        self.failUnlessRaises(ExifValueError, tag._convert_to_string, 3)
+
+    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, str(tag))
+        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))