diff options
Diffstat (limited to 'archaeological_finds/tests.py')
| -rw-r--r-- | archaeological_finds/tests.py | 50 |
1 files changed, 24 insertions, 26 deletions
diff --git a/archaeological_finds/tests.py b/archaeological_finds/tests.py index ad78646b9..8c5feb7a2 100644 --- a/archaeological_finds/tests.py +++ b/archaeological_finds/tests.py @@ -877,10 +877,7 @@ class ImportFindTest(ImportTest, FindInit, TestCase): _("The division {} {} do not exist for the location {}.") ).format("Area", "43", "Warehouse test") for error in impt.errors: - self.assertEqual( - error["error"], - error_msg - ) + self.assertEqual(error["error"], error_msg) def tearDown(self): self.tmpdir.cleanup() @@ -1088,8 +1085,9 @@ class FindBasketTest(FindInit, TestCase): self.create_finds(force=True) self.username = "myuser" self.password = "mypassword" - u = User.objects.create_superuser(self.username, "myemail@test.com", - self.password) + u = User.objects.create_superuser( + self.username, "myemail@test.com", self.password + ) self.client = Client() b = models.FindBasket.objects.create(label="test basket", user_id=u.pk) b.items.add(self.finds[0]) @@ -1113,7 +1111,7 @@ class FindBasketTest(FindInit, TestCase): content = response.content.decode() res = json.loads(content) self.assertEqual(res["recordsTotal"], 2) - search = {"search_vector" : 'other'} + search = {"search_vector": "other"} response = c.get(reverse("get-findbasket"), search) content = response.content.decode() res = json.loads(content) @@ -2526,21 +2524,18 @@ class GeomaticTest(FindInit, TestCase): base_find = models.BaseFind.objects.get(pk=base_find.pk) dic = { "type": "Feature", - "geometry": { - "type": "Point", - "coordinates": [2.0, 43.0] - } + "geometry": {"type": "Point", "coordinates": [2.0, 43.0]}, } res = base_find.get_geo_items(get_polygons=True) self.assertEqual(dic, res) res = base_find.get_geo_items(get_polygons=False) self.assertEqual(dic, res) - # test API + # test API find = base_find.find.all()[0] response = self.client.get(reverse("api-get-geo-items", kwargs={"pk": find.pk})) self.assertEqual(response.status_code, 200) - self.assertIn(json.dumps(dic).encode('utf-8'), response.content) + self.assertIn(json.dumps(dic).encode("utf-8"), response.content) # with multi-polygon base_find = models.BaseFind.objects.get(pk=base_find.pk) @@ -2564,31 +2559,34 @@ class GeomaticTest(FindInit, TestCase): [[1.0, 1.0], [5.0, 1.0], [5.0, 5.0], [1.0, 5.0], [1.0, 1.0]], [[2.0, 2.0], [2.0, 3.0], [3.0, 3.0], [3.0, 2.0], [2.0, 2.0]], ], - [ - [[6.0, 3.0], [9.0, 2.0], [9.0, 4.0], [6.0, 3.0]] - ] - ] + [[[6.0, 3.0], [9.0, 2.0], [9.0, 4.0], [6.0, 3.0]]], + ], }, } res_poly = base_find.get_geo_items(get_polygons=True) self.assertEqual(dict_poly, res_poly) dict_centroid = { "type": "Feature", - "geometry": { - "type": "Point", - "coordinates": [3.86111, 3.02778] - } + "geometry": {"type": "Point", "coordinates": [3.86111, 3.02778]}, } res_centroid = base_find.get_geo_items(get_polygons=False) self.assertEqual(dict_centroid, res_centroid) - # test API - response = self.client.get(reverse("api-get-geo-items", kwargs={"pk": base_find.pk, "get_polygons": True})) + # test API + response = self.client.get( + reverse( + "api-get-geo-items", kwargs={"pk": base_find.pk, "get_polygons": True} + ) + ) self.assertEqual(response.status_code, 200) - self.assertIn(json.dumps(dict_poly).encode('utf-8'), response.content) - response = self.client.get(reverse("api-get-geo-items", kwargs={"pk": base_find.pk, "get_polygons": False})) + self.assertIn(json.dumps(dict_poly).encode("utf-8"), response.content) + response = self.client.get( + reverse( + "api-get-geo-items", kwargs={"pk": base_find.pk, "get_polygons": False} + ) + ) self.assertEqual(response.status_code, 200) - self.assertIn(json.dumps(dict_centroid).encode('utf-8'), response.content) + self.assertIn(json.dumps(dict_centroid).encode("utf-8"), response.content) |