aboutsummaryrefslogtreecommitdiffstats
path: root/rest_framework/tests/files.py
diff options
context:
space:
mode:
Diffstat (limited to 'rest_framework/tests/files.py')
-rw-r--r--rest_framework/tests/files.py3
1 files changed, 2 insertions, 1 deletions
diff --git a/rest_framework/tests/files.py b/rest_framework/tests/files.py
index a69695ca..42e8ed5f 100644
--- a/rest_framework/tests/files.py
+++ b/rest_framework/tests/files.py
@@ -1,6 +1,7 @@
from rest_framework.compat import BytesIO
import datetime
+import six
from django.test import TestCase
@@ -29,7 +30,7 @@ class FileSerializerTests(TestCase):
def test_create(self):
now = datetime.datetime.now()
- file = BytesIO(b'stuff')
+ file = BytesIO(six.b('stuff'))
file.name = 'stuff.txt'
file.size = len(file.getvalue())
serializer = UploadedFileSerializer(data={'created': now}, files={'file': file})