aboutsummaryrefslogtreecommitdiffstats
path: root/test/extractor/image/test_photometrics.py
blob: fb219e23c66fe33cb2bed8a567dd6d44dcf048f7 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143

# standard imports
import unittest

# bsie imports
from bsie.extractor import base
from bsie.utils import bsfs, node as _node, ns

# objects to test
from bsie.extractor.image.photometrics import Exif, _gps_to_dec


## code ##

class TestExif(unittest.TestCase):

    def test_gps_to_dec(self):
        # deg+min+sec format
        self.assertAlmostEqual(_gps_to_dec('29/1 58/1 45/1'.split()), 29.979167, 6)
        self.assertAlmostEqual(_gps_to_dec('31 08 03'.split()), 31.134167, 6)
        self.assertAlmostEqual(_gps_to_dec('20 40 586/10'.split()), 20.682944, 6)
        self.assertAlmostEqual(_gps_to_dec('88/1 34 68/10'.split()), 88.568556, 6)
        # deg+min format
        self.assertAlmostEqual(_gps_to_dec('13 472167/10000 0/1 '.split()), 13.786945, 6)
        self.assertAlmostEqual(_gps_to_dec('104/1 3215/100 0/1'.split()), 104.535833, 6)

    def test_eq(self):
        # identical instances are equal
        self.assertEqual(Exif(), Exif())
        self.assertEqual(hash(Exif()), hash(Exif()))
        # comparison respects type
        class Foo(): pass
        self.assertNotEqual(Exif(), Foo())
        self.assertNotEqual(hash(Exif()), hash(Foo()))
        self.assertNotEqual(Exif(), 1234)
        self.assertNotEqual(hash(Exif()), hash(1234))
        self.assertNotEqual(Exif(), None)
        self.assertNotEqual(hash(Exif()), hash(None))

    def test_schema(self):
        self.assertSetEqual({pred.uri for pred in Exif().schema.predicates()}, {
            ns.bsfs.Predicate,
            ns.bse.exposure,
            ns.bse.aperture,
            ns.bse.iso,
            ns.bse.focal_length,
            ns.bse.width,
            ns.bse.height,
            ns.bse.orientation,
            ns.bse.orientation_label,
            ns.bse.altitude,
            ns.bse.latitude,
            ns.bse.longitude,
            })

    def test_extract(self):
        ext = Exif()
        node = _node.Node(ns.bsfs.File, '') # Blank node
        content = {
            'Exif.Photo.ExposureTime': '10/600',
            'Exif.Photo.FNumber': '48/10',
            'Exif.Photo.ISOSpeedRatings': '400',
            'Exif.Photo.FocalLength': '460/10',
            'Exif.Photo.PixelXDimension': '4288',
            'Exif.Photo.PixelYDimension': '2848',
            'Exif.Image.Orientation': '1',
            'Exif.GPSInfo.GPSAltitude': '431/1',
            'Exif.GPSInfo.GPSLatitude': '46/1 11397/625 0/1',
            'Exif.GPSInfo.GPSLongitude': '7/1 131250/2500 0/1',
            }

        # invalid principals are ignored
        self.assertSetEqual(set(ext.extract(node, content, {ns.bse.filename})), set())
        # extract finds all relevant information
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.exposure)})),
            {(node, ext.schema.predicate(ns.bse.exposure), 60.0)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.aperture)})),
            {(node, ext.schema.predicate(ns.bse.aperture), 4.8)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.iso)})),
            {(node, ext.schema.predicate(ns.bse.iso), 400)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.focal_length)})),
            {(node, ext.schema.predicate(ns.bse.focal_length), 46.0)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.width)})),
            {(node, ext.schema.predicate(ns.bse.width), 4288)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.height)})),
            {(node, ext.schema.predicate(ns.bse.height), 2848)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.orientation)})),
            {(node, ext.schema.predicate(ns.bse.orientation), 1)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.orientation_label)})),
            {(node, ext.schema.predicate(ns.bse.orientation_label), 'landscape')})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.altitude)})),
            {(node, ext.schema.predicate(ns.bse.altitude), 431.0)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.latitude)})),
            {(node, ext.schema.predicate(ns.bse.latitude), 46.30392)})
        self.assertSetEqual(set(ext.extract(node, content, {ext.schema.predicate(ns.bse.longitude)})),
            {(node, ext.schema.predicate(ns.bse.longitude), 7.875)})

        # check orientation label
        self.assertSetEqual(set(ext.extract(
            node, {
                'Exif.Photo.PixelXDimension': '4288',
                'Exif.Photo.PixelYDimension': '2848',
                'Exif.Image.Orientation': '5',
                },
            {ext.schema.predicate(ns.bse.orientation_label)})),
            {(node, ext.schema.predicate(ns.bse.orientation_label), 'portrait')})

        # can pass multiple principals
        self.assertSetEqual(set(ext.extract(node, content, {
            ext.schema.predicate(ns.bse.exposure),
            ext.schema.predicate(ns.bse.iso),
            ext.schema.predicate(ns.bse.focal_length),
            })), {
            (node, ext.schema.predicate(ns.bse.exposure), 60.0),
            (node, ext.schema.predicate(ns.bse.iso), 400),
            (node, ext.schema.predicate(ns.bse.focal_length), 46.0),
            })

        # principals w/o content are ignored
        self.assertSetEqual(set(ext.extract(
            node,
            content={'Exif.Photo.ExposureTime': '10/600'},
            principals={
                ext.schema.predicate(ns.bse.exposure),
                ext.schema.predicate(ns.bse.iso),
                ext.schema.predicate(ns.bse.focal_length),
                })
            ), {
            (node, ext.schema.predicate(ns.bse.exposure), 60.0),
            })

        # empty content is acceptable
        self.assertSetEqual(set(ext.extract(node, {}, set(ext.principals))), set())
        # no principals is acceptable
        self.assertSetEqual(set(ext.extract(node, content, set())), set())


## main ##

if __name__ == '__main__':
    unittest.main()

## EOF ##