Fix archaeological finds tests

1 files changed, 8 insertions, 6 deletions

diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py
index f61807637..34cacfe42 100644
--- a/archaeological_finds/tests.py
+++ b/archaeological_finds/tests.py
@@ -27,6 +27,7 @@ from archaeological_context_records.tests import ImportContextRecordTest
 
 from ishtar_common import forms_common
 
+
 class ImportFindTest(ImportContextRecordTest):
     test_context_records = False
 
@@ -40,12 +41,13 @@ class ImportFindTest(ImportContextRecordTest):
 
         old_nb = models.BaseFind.objects.count()
         MCC = ImporterType.objects.get(name=u"MCC - Mobilier")
-        mcc_file = open(settings.ROOT_PATH + \
-              '../archaeological_finds/tests/MCC-finds-example.csv', 'rb')
+        mcc_file = open(
+            settings.ROOT_PATH +
+            '../archaeological_finds/tests/MCC-finds-example.csv', 'rb')
         file_dict = {'imported_file': SimpleUploadedFile(mcc_file.name,
                                                          mcc_file.read())}
-        post_dict = {'importer_type':MCC.pk, 'skip_lines':1,
-                     "encoding":'utf-8'}
+        post_dict = {'importer_type': MCC.pk, 'skip_lines': 1,
+                     "encoding": 'utf-8'}
         form = forms_common.NewImportForm(data=post_dict, files=file_dict,
                                           instance=None)
         form.is_valid()
@@ -56,7 +58,7 @@ class ImportFindTest(ImportContextRecordTest):
 
         # doing manual connections
         ceram = models.MaterialType.objects.get(txt_idx='ceramic').pk
-        self.setTargetKey('find__material_type', 'ceramique', ceram)
+        self.setTargetKey('find__material_types', 'ceramique', ceram)
         impt.importation()
         if not test:
             return
@@ -64,4 +66,4 @@ class ImportFindTest(ImportContextRecordTest):
         current_nb = models.BaseFind.objects.count()
         self.assertTrue(current_nb == (old_nb + 4))
         self.assertEqual(
-            models.Find.objects.filter(material_type__pk=ceram).count(), 4)
+            models.Find.objects.filter(material_types__pk=ceram).count(), 4)