X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/3726f9b225fb5c39cbf067f53d4cf6f7d8a9f326..0e79f07515678e3e678c3637124a27cb81ee8a94:/apps/picture/tests/picture_import.py?ds=inline diff --git a/apps/picture/tests/picture_import.py b/apps/picture/tests/picture_import.py index 68c875391..1e17289c3 100644 --- a/apps/picture/tests/picture_import.py +++ b/apps/picture/tests/picture_import.py @@ -1,4 +1,7 @@ # -*- coding: utf-8 -*- +# This file is part of Wolnelektury, licensed under GNU Affero GPLv3 or later. +# Copyright © Fundacja Nowoczesna Polska. See NOTICE for more information. +# from __future__ import with_statement from os import path @@ -7,12 +10,19 @@ from catalogue.test_utils import WLTestCase class PictureTest(WLTestCase): - + def test_import(self): picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml")) - motifs = set([tag.name for tag in picture.tags if tag.category == 'theme']) - assert motifs == set([u'nieporządek']), 'theme tags are wrong. %s' % motifs + themes = set() + for area in picture.areas.all(): + themes.update([(tag.category, tag.name) + for tag in area.tags if tag.category in (u'theme', u'thing')]) + assert themes == set([(u'theme', u'nieporządek'), (u'thing', u'Kosmos')]), \ + 'Bad themes on Picture areas: %s' % themes + + pic_themes = set([tag.name for tag in picture.tags if tag.category in ('theme', 'thing')]) + assert not pic_themes, 'Unwanted themes set on Pictures: %s' % pic_themes picture.delete() @@ -21,13 +31,13 @@ class PictureTest(WLTestCase): path.join(path.dirname(__file__), "files/kandinsky-composition-viii.png")) picture.delete() - + def test_import_2(self): - picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/pejzaz-i-miasto-krzyzanowski-chmury.xml"), - path.join(path.dirname(__file__), "files/pejzaz-i-miasto-krzyzanowski-chmury.jpg"), + picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"), + path.join(path.dirname(__file__), "files/kandinsky-composition-viii.png"), overwrite=True) cats = set([t.category for t in picture.tags]) - assert 'genre' in cats + assert 'epoch' in cats assert 'kind' in cats