X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/491055ce596b2506c74ca560428814557a4462b8..7fa5002e3aa435859aea7fb4e0c814b4034b6aa5:/apps/picture/tests/picture_import.py

diff --git a/apps/picture/tests/picture_import.py b/apps/picture/tests/picture_import.py
index 91fb35f71..84009e674 100644
--- a/apps/picture/tests/picture_import.py
+++ b/apps/picture/tests/picture_import.py
@@ -1,19 +1,16 @@
 # -*- 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 django.core.files.base import ContentFile, File
-from catalogue.test_utils import *
-from catalogue import models
-from librarian import WLURI
+from os import path
 from picture.models import Picture
+from catalogue.test_utils import WLTestCase
 
-from nose.tools import raises
-import tempfile
-from os import unlink, path, makedirs
 
+class PictureTest(WLTestCase):
 
-class PictureTest(TestCase):
-    
     def test_import(self):
         picture = Picture.from_xml_file(path.join(path.dirname(__file__), "files/kandinsky-composition-viii.xml"))
 
@@ -27,4 +24,13 @@ class PictureTest(TestCase):
                                         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"),
+                                        overwrite=True)
+        cats = set([t.category for t in picture.tags])
+        assert 'genre' in cats
+        assert 'kind' in cats
+