fix test for OrderedDict

kazmiruk · kazmiruk · commit e4e3f57321e7 · 2015-04-22T13:33:58.000+07:00
diff --git a/tests/test_fields.py b/tests/test_fields.py
@@ -3,14 +3,14 @@
 """
 from __future__ import unicode_literals
 
+from collections import OrderedDict
 import datetime
 import re
 from decimal import Decimal
 from uuid import uuid4
 from django.core import validators
 from django.db import models
 from django.test import TestCase
-from django.utils.datastructures import SortedDict
 from rest_framework import serializers
 from tests.models import RESTFrameworkModel
 
@@ -95,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 = SortedDict()
+        ret = OrderedDict()
         ret['c'] = 1
         ret['b'] = 1
         ret['a'] = 1
diff --git a/tests/test_serializer.py b/tests/test_serializer.py
@@ -1297,7 +1297,7 @@ def test_pickle_simple_model_serializer_data(self):
 
     def test_pickle_inner_serializer(self):
         """
-        Test pickling a serializer whose resulting .data (a SortedDictWithMetadata) will
+        Test pickling a serializer whose resulting .data (a OrderedDictWithMetadata) will
         have unpickleable meta data--in order to make sure metadata doesn't get pulled into the pickle.
         See DictWithMetadata.__getstate__
         """
@@ -1312,13 +1312,13 @@ def test_getstate_method_should_not_return_none(self):
         Regression test for #645.
         """
         data = serializers.DictWithMetadata({1: 1})
-        self.assertEqual(data.__getstate__(), serializers.SortedDict({1: 1}))
+        self.assertEqual(data.__getstate__(), serializers.OrderedDict({1: 1}))
 
     def test_serializer_data_is_pickleable(self):
         """
         Another regression test for #645.
         """
-        data = serializers.SortedDictWithMetadata({1: 1})
+        data = serializers.OrderedDictWithMetadata({1: 1})
         repr(pickle.loads(pickle.dumps(data, 0)))
 
 
