diff options
Diffstat (limited to 'archaeological_finds/tests.py')
-rw-r--r-- | archaeological_finds/tests.py | 14 |
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) |