fixes tests for renamed SortedDict

kazmiruk · kazmiruk · commit 73e433ed5c68 · 2015-04-23T16:32:36.000+07:00
diff --git a/tests/test_fields.py b/tests/test_fields.py
@@ -11,13 +11,9 @@
 from django.db import models
 from django.test import TestCase
 from rest_framework import serializers
+from rest_framework.compat import SortedDict
 from tests.models import RESTFrameworkModel
 
-try:
-    from collections import OrderedDict
-except ImportError:
-    from django.utils.datastructures import SortedDict as OrderedDict
-
 
 class TimestampedModel(models.Model):
     added = models.DateTimeField(auto_now_add=True)
@@ -99,7 +95,7 @@ def test_dict_field_ordering(self):
         Field should preserve dictionary ordering, if it exists.
         See: https://github.com/tomchristie/django-rest-framework/issues/832
         """
-        ret = OrderedDict()
+        ret = SortedDict()
         ret['c'] = 1
         ret['b'] = 1
         ret['a'] = 1
diff --git a/tests/test_serializer.py b/tests/test_serializer.py
@@ -1312,7 +1312,7 @@ def test_getstate_method_should_not_return_none(self):
         Regression test for #645.
         """
         data = serializers.DictWithMetadata({1: 1})
-        self.assertEqual(data.__getstate__(), serializers.OrderedDict({1: 1}))
+        self.assertEqual(data.__getstate__(), serializers.SortedDict({1: 1}))
 
     def test_serializer_data_is_pickleable(self):
         """
