diff options
| -rw-r--r-- | archaeological_finds/tests.py | 6 |
1 files changed, 4 insertions, 2 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index a0c05dbe2..097038d8c 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -277,7 +277,7 @@ class ImportFindTest(ImportTest, TestCase): importer_type_id=MCC.pk) formater = FormaterType.objects.filter( formater_type='FileFormater').all()[0] - ImportTarget.objects.create(target='image', + ImportTarget.objects.create(target='images__image', formater_type_id=formater.pk, column_id=col.pk) mcc_file = open( @@ -314,7 +314,9 @@ class ImportFindTest(ImportTest, TestCase): models.Find.objects.filter(material_types__pk=ceram).count(), 4) self.assertEqual( models.Find.objects.filter(material_types__pk=glass).count(), 1) - images = [f.image for f in models.Find.objects.all() if f.image.name] + images = [] + for find in models.Find.objects.all(): + images += [im.image for im in find.images.all() if im.image.name] self.assertEqual(len(images), 1) # check index |