X-Git-Url: https://git.mdrn.pl/librarian.git/blobdiff_plain/5fed78856949474a36bc5e268517775a9a802e27..2c15db814c7f40406b6a86383e3e4bc8825b7faf:/tests/test_picture.py diff --git a/tests/test_picture.py b/tests/test_picture.py index 71a77dc..7d12b1d 100644 --- a/tests/test_picture.py +++ b/tests/test_picture.py @@ -3,17 +3,17 @@ # This file is part of Librarian, licensed under GNU Affero GPLv3 or later. # Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. # +from __future__ import unicode_literals + from librarian import picture, dcparser -from lxml import etree -from nose.tools import * -from os.path import splitext from tests.utils import get_all_fixtures, get_fixture -import codecs from os import path + def test_wlpictureuri(): uri = picture.WLPictureURI('http://wolnelektury.pl/katalog/obraz/angelus-novus') + def check_load(xml_file): pi = dcparser.parse(xml_file, picture.PictureInfo) assert pi is not None @@ -31,22 +31,23 @@ def test_wlpicture(): # from nose.tools import set_trace; set_trace() assert pi.type[0] == u"Image" - assert pi.mime_type == u'image/png' == wlp.mime_type + assert pi.mime_type == u'image/jpeg' == wlp.mime_type assert wlp.slug == 'angelus-novus' assert path.exists(wlp.image_path) - f = wlp.image_file('r') + f = wlp.image_file() f.close() + def test_picture_parts(): wlp = picture.WLPicture.from_file(open(get_fixture('picture', 'angelus-novus.xml'))) parts = list(wlp.partiter()) - assert len(parts) == 5, "there should be %d parts of the picture" % 5 + expect_parts = 4 + assert len(parts) == expect_parts, "there should be %d parts of the picture" % expect_parts motifs = set() names = set() - print parts for p in parts: for m in p['themes']: motifs.add(m) @@ -54,7 +55,5 @@ def test_picture_parts(): if p['object']: names.add(p['object']) - assert motifs == set([u'anioł historii', u'spojrzenie']), "missing motifs, got: %s" % motifs - assert names == set([u'obraz cały', u'skrzydło']), 'missing objects, got: %s' % names - - + assert motifs == {u'anioł historii', u'spojrzenie'}, "missing motifs, got: %s" % motifs + assert names == {u'obraz cały', u'skrzydło'}, 'missing objects, got: %s' % names