Finds: fix material col in tables

1 files changed, 6 insertions, 2 deletions

diff --git a/archaeological_context_records/tests.py b/archaeological_context_records/tests.py
index c56964670..87eb80c57 100644
--- a/archaeological_context_records/tests.py
+++ b/archaeological_context_records/tests.py
@@ -231,7 +231,10 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase):
         c.login(username=self.username, password=self.password)
         response = c.get(reverse('get-contextrecord'), search)
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(json.loads(response.content)['total'], 1)
+        res = json.loads(response.content)
+        self.assertTrue(res['total'] == 1)
+        self.assertEqual(res["rows"][0]["unit"],
+                         unicode(dig))
 
         # no result for the brother
         search = {'unit': dest.pk}
@@ -268,7 +271,8 @@ class ContextRecordSearchTest(ContextRecordInit, TestCase):
         c.login(username=self.username, password=self.password)
         response = c.get(reverse('get-contextrecord'), search)
         self.assertEqual(response.status_code, 200)
-        self.assertEqual(json.loads(response.content)['total'], 1)
+        res = json.loads(response.content)
+        self.assertTrue(res['total'] == 1)
 
         # no result for the brother
         search = {'datings__period': recent_neo.pk}