diff --git a/src/ralph/data_importer/tests/test_fields.py b/src/ralph/data_importer/tests/test_fields.py index 0074ab16d2..f5474d2818 100644 --- a/src/ralph/data_importer/tests/test_fields.py +++ b/src/ralph/data_importer/tests/test_fields.py @@ -134,7 +134,7 @@ def test_through_field_only_add(self): self.assertEqual(self.licence.base_objects.all().count(), 4) self.assertCountEqual( - self.licence.base_objects.values_list('pk', flat=True), ids + [bo.pk for bo in self.licence.base_objects.all()], ids ) # Make sure it doesn't touch other licences self.assertEqual(self.licence2.base_objects.all().count(), 2) @@ -151,7 +151,7 @@ def test_through_field_only_remove(self): self.assertEqual(self.licence.base_objects.all().count(), 1) self.assertCountEqual( - self.licence.base_objects.values_list('pk', flat=True), ids + [bo.pk for bo in self.licence.base_objects.all()], ids ) # Make sure it doesn't touch other licences self.assertEqual(self.licence2.base_objects.all().count(), 2) @@ -172,7 +172,7 @@ def test_through_field_add_and_remove(self): ) self.assertEqual(self.licence.base_objects.all().count(), 3) self.assertCountEqual( - self.licence.base_objects.values_list('pk', flat=True), ids + [bo.pk for bo in self.licence.base_objects.all()], ids ) # Make sure it doesn't touch other licences self.assertEqual(self.licence2.base_objects.all().count(), 2)