X-Git-Url: https://git.mdrn.pl/wolnelektury.git/blobdiff_plain/cfe30a8b506379030bd36bfabd5a17efe83c3100..3b7dc968f4a15d41a63cd7bb220a7a57e089954b:/apps/picture/tests/picture_import.py

diff --git a/apps/picture/tests/picture_import.py b/apps/picture/tests/picture_import.py
index 9e965f91d..785aa6d28 100644
--- a/apps/picture/tests/picture_import.py
+++ b/apps/picture/tests/picture_import.py
@@ -1,22 +1,24 @@
 # -*- coding: utf-8 -*-
 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(TestCase):
+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
-    
+
+        picture.delete()
+
+    def test_import_with_explicit_image(self):
+        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"))
+
+        picture.delete()
+