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) | 
