diff --git a/tests/test_relations.py b/tests/test_relations.py index f11b08119..75a6991e7 100644 --- a/tests/test_relations.py +++ b/tests/test_relations.py @@ -304,7 +304,7 @@ class TestSlugRelatedField(APISimpleTestCase): self.queryset = MockQueryset([ MockObject(pk=1, name='foo'), MockObject(pk=2, name='bar'), - MockObject(pk=3, name='baz'), + MockObject(pk=3, name='baz') ]) self.instance = self.queryset.items[2] self.field = serializers.SlugRelatedField( diff --git a/tests/utils.py b/tests/utils.py index 639785951..4ceb35309 100644 --- a/tests/utils.py +++ b/tests/utils.py @@ -1,4 +1,5 @@ from operator import attrgetter + from django.core.exceptions import ObjectDoesNotExist from django.urls import NoReverseMatch