Use Django's six library for Py3 compatibility.

This commit is contained in:
Roberto Rosario
2015-10-21 03:33:10 -04:00
parent 6b950b7ccf
commit 58f73f0b1e

View File

@@ -2,7 +2,6 @@
from __future__ import unicode_literals from __future__ import unicode_literals
import io
import time import time
from json import loads from json import loads
@@ -11,6 +10,7 @@ from django.contrib.auth.models import User
from django.core.files import File from django.core.files import File
from django.core.urlresolvers import reverse from django.core.urlresolvers import reverse
from django.test import override_settings from django.test import override_settings
from django.utils.six import BytesIO
from rest_framework import status from rest_framework import status
from rest_framework.test import APITestCase from rest_framework.test import APITestCase
@@ -259,7 +259,7 @@ class DocumentAPITestCase(APITestCase):
'rest_api:document-download', args=(document.pk,) 'rest_api:document-download', args=(document.pk,)
) )
) )
buf = io.BytesIO() buf = BytesIO()
buf.write(response.content) buf.write(response.content)
self.assertEqual( self.assertEqual(
@@ -280,7 +280,7 @@ class DocumentAPITestCase(APITestCase):
args=(document.latest_version.pk,) args=(document.latest_version.pk,)
) )
) )
buf = io.BytesIO() buf = BytesIO()
buf.write(response.content) buf.write(response.content)
self.assertEqual( self.assertEqual(