Import force_bytes on django >= 1.5

1 files changed, 13 insertions, 0 deletions

diff --git a/rest_framework/tests/test_compat.py b/rest_framework/tests/test_compat.py
new file mode 100644
index 00000000..4916d19b
--- /dev/null
+++ b/rest_framework/tests/test_compat.py
@@ -0,0 +1,13 @@
+import django
+from django.test import TestCase
+
+
+class TestCompat(TestCase):
+    def test_force_bytes_or_smart_bytes(self):
+        from rest_framework.compat import force_bytes_or_smart_bytes
+        if django.VERSION >= (1, 5):
+            from django.utils.encoding import force_bytes
+            self.assertEqual(force_bytes_or_smart_bytes, force_bytes)
+        else:
+            from django.utils.encoding import smart_str
+            self.assertEqual(force_bytes_or_smart_bytes, smart_str)