From 58f73f0b1e3c93dbaff4d4596835b1a895a39aea Mon Sep 17 00:00:00 2001 From: Roberto Rosario Date: Wed, 21 Oct 2015 03:33:10 -0400 Subject: [PATCH] Use Django's six library for Py3 compatibility. --- mayan/apps/documents/tests/test_api.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/mayan/apps/documents/tests/test_api.py b/mayan/apps/documents/tests/test_api.py index 0b5b01807e..75d668342f 100644 --- a/mayan/apps/documents/tests/test_api.py +++ b/mayan/apps/documents/tests/test_api.py @@ -2,7 +2,6 @@ from __future__ import unicode_literals -import io import time from json import loads @@ -11,6 +10,7 @@ from django.contrib.auth.models import User from django.core.files import File from django.core.urlresolvers import reverse from django.test import override_settings +from django.utils.six import BytesIO from rest_framework import status from rest_framework.test import APITestCase @@ -259,7 +259,7 @@ class DocumentAPITestCase(APITestCase): 'rest_api:document-download', args=(document.pk,) ) ) - buf = io.BytesIO() + buf = BytesIO() buf.write(response.content) self.assertEqual( @@ -280,7 +280,7 @@ class DocumentAPITestCase(APITestCase): args=(document.latest_version.pk,) ) ) - buf = io.BytesIO() + buf = BytesIO() buf.write(response.content) self.assertEqual(