summaryrefslogtreecommitdiff
path: root/archaeological_context_records/tests.py
diff options
context:
space:
mode:
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
commita23307e29cc5a4f0c6ad03aa300b0c07a680751b (patch)
tree53df01f8bdeeab4aa94e081a381c2939962f0d66 /archaeological_context_records/tests.py
parentde51f7b9be16bbc260ea73cdaaef406976c5a8da (diff)
downloadIshtar-a23307e29cc5a4f0c6ad03aa300b0c07a680751b.tar.bz2
Ishtar-a23307e29cc5a4f0c6ad03aa300b0c07a680751b.zip
Flake8
Diffstat (limited to 'archaeological_context_records/tests.py')
-rw-r--r--archaeological_context_records/tests.py64
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)