From 47458b74046cd083ae48f21fb133b59d1ee5e7f9 Mon Sep 17 00:00:00 2001 From: Étienne Loks Date: Fri, 22 Sep 2017 21:10:16 +0200 Subject: Tests: fix V3tov2 test --- chimere/tests.py | 10 ++++++---- 1 file changed, 6 insertions(+), 4 deletions(-) diff --git a/chimere/tests.py b/chimere/tests.py index e8b280f..641a9cd 100644 --- a/chimere/tests.py +++ b/chimere/tests.py @@ -754,11 +754,13 @@ class V3ToV2Test(TestCase): sys.stdout = StringIO() call_command('v3_to_v2_markers') vals = sys.stdout.getvalue() - markers = json.loads(vals) - self.assertEqual(len(markers), 4) + values = json.loads(vals) + markers = [k for k in values + if k['model'] == 'chimere.marker'] + self.assertEqual(len(markers), len(self.markers)) self.assertNotIn('weight', markers[0]['fields']) self.assertNotIn('normalised_weight', markers[0]['fields']) - self.assertEqual(markers[-1]['fields']['value'], 'myproperty') + self.assertEqual(values[-1]['fields']['value'], 'myproperty') self.assertNotIn('polygon', markers[-1]['fields']) self.assertNotIn('route', markers[-1]['fields']) sys.stdout = sysout @@ -768,7 +770,7 @@ class V3ToV2Test(TestCase): sys.stdout = StringIO() call_command('v3_to_v2_routes') vals = sys.stdout.getvalue() - routes = json.loads(vals) + routes = [k for k in json.loads(vals) if k['model'] == 'chimere.route'] self.assertEqual(len(routes), len(self.routes)) self.assertNotIn('weight', routes[0]['fields']) self.assertNotIn('normalised_weight', routes[0]['fields']) -- cgit v1.2.3