diff options
author | Étienne Loks <etienne.loks@proxience.com> | 2015-09-21 23:28:36 +0200 |
---|---|---|
committer | Étienne Loks <etienne.loks@proxience.com> | 2015-09-21 23:28:36 +0200 |
commit | 8eee2250b8f140eb689663a056389f20a966a887 (patch) | |
tree | 53df01f8bdeeab4aa94e081a381c2939962f0d66 /archaeological_context_records/tests.py | |
parent | b1f63beb09085b0a8d7a238d23234d9f44f5d4ed (diff) | |
download | Ishtar-8eee2250b8f140eb689663a056389f20a966a887.tar.bz2 Ishtar-8eee2250b8f140eb689663a056389f20a966a887.zip |
Flake8
Diffstat (limited to 'archaeological_context_records/tests.py')
-rw-r--r-- | archaeological_context_records/tests.py | 64 |
1 files changed, 34 insertions, 30 deletions
diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py index 00854ad7c..bccd859aa 100644 --- a/archaeological_context_records/tests.py +++ b/archaeological_context_records/tests.py @@ -20,7 +20,6 @@ """ Unit tests """ -import json from django.conf import settings from django.core.exceptions import ValidationError @@ -29,11 +28,11 @@ from django.test import TestCase from ishtar_common.models import ImporterType from archaeological_operations.tests import OperationInitTest, \ - ImportOperationTest -from archaeological_operations.models import Parcel + ImportOperationTest from archaeological_context_records import models from ishtar_common import forms_common + class ImportContextRecordTest(ImportOperationTest): test_operations = False test_context_records = True @@ -50,13 +49,14 @@ class ImportContextRecordTest(ImportOperationTest): old_nb = models.ContextRecord.objects.count() MCC = ImporterType.objects.get(name=u"MCC - UE") - mcc_file = open(settings.ROOT_PATH + \ - '../archaeological_context_records/tests/'\ - 'MCC-context-records-example.csv', 'rb') + mcc_file = open( + settings.ROOT_PATH + + '../archaeological_context_records/tests/' + 'MCC-context-records-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() @@ -91,7 +91,7 @@ class ContextRecordInit(OperationInitTest): if not getattr(self, 'context_records', None): self.context_records = [] - default = {'label':"Context record"} + default = {'label': "Context record"} if not data.get('operation'): data['operation'] = self.get_default_operation() if not data.get('parcel'): @@ -101,12 +101,13 @@ class ContextRecordInit(OperationInitTest): default.update(data) self.context_records.append(models.ContextRecord.objects.create( - **default)) + **default)) return self.context_records def get_default_context_record(self): return self.create_context_record()[0] + class RecordRelationsTest(TestCase, ContextRecordInit): fixtures = [settings.ROOT_PATH + '../fixtures/initial_data.json', @@ -120,22 +121,22 @@ class RecordRelationsTest(TestCase, ContextRecordInit): def setUp(self): # two different context records - self.create_context_record({"label":u"CR 1"}) - self.create_context_record({"label":u"CR 2"}) + self.create_context_record({"label": u"CR 1"}) + self.create_context_record({"label": u"CR 2"}) def testRelations(self): - sym_rel_type = models.RelationType.objects.create(symmetrical=True, - txt_idx='sym') - rel_type_1 = models.RelationType.objects.create(symmetrical=False, - txt_idx='rel_1') + sym_rel_type = models.RelationType.objects.create( + symmetrical=True, txt_idx='sym') + rel_type_1 = models.RelationType.objects.create( + symmetrical=False, txt_idx='rel_1') # cannot be symmetrical and have an inverse_relation with self.assertRaises(ValidationError): - rel_test = models.RelationType.objects.create(symmetrical=True, - inverse_relation=rel_type_1, txt_idx='rel_3') + rel_test = models.RelationType.objects.create( + symmetrical=True, inverse_relation=rel_type_1, txt_idx='rel_3') rel_test.full_clean() # auto fill inverse relations - rel_type_2 = models.RelationType.objects.create(symmetrical=False, - inverse_relation=rel_type_1, txt_idx='rel_2') + rel_type_2 = models.RelationType.objects.create( + symmetrical=False, inverse_relation=rel_type_1, txt_idx='rel_2') self.assertEqual(rel_type_1.inverse_relation, rel_type_2) cr_1 = self.context_records[0] @@ -143,19 +144,22 @@ class RecordRelationsTest(TestCase, ContextRecordInit): # inserting a new symmetrical relation automatically creates the same # relation for the second context record - rel = models.RecordRelations.objects.create(left_record=cr_1, - right_record=cr_2, relation_type=sym_rel_type) - self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2, - right_record=cr_1, relation_type=sym_rel_type).count(), 1) + rel = models.RecordRelations.objects.create( + left_record=cr_1, right_record=cr_2, relation_type=sym_rel_type) + self.assertEqual(models.RecordRelations.objects.filter( + left_record=cr_2, right_record=cr_1, + relation_type=sym_rel_type).count(), 1) # removing one symmetrical relation removes the other rel.delete() - self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2, - right_record=cr_1, relation_type=sym_rel_type).count(), 0) + self.assertEqual(models.RecordRelations.objects.filter( + left_record=cr_2, right_record=cr_1, + relation_type=sym_rel_type).count(), 0) # for non-symmetrical relation, adding one relation automatically # adds the inverse - rel = models.RecordRelations.objects.create(left_record=cr_1, - right_record=cr_2, relation_type=rel_type_1) - self.assertEqual(models.RecordRelations.objects.filter(left_record=cr_2, - right_record=cr_1, relation_type=rel_type_2).count(), 1) + rel = models.RecordRelations.objects.create( + left_record=cr_1, right_record=cr_2, relation_type=rel_type_1) + self.assertEqual(models.RecordRelations.objects.filter( + left_record=cr_2, right_record=cr_1, + relation_type=rel_type_2).count(), 1) |