Use six.text_type instead of unicode().
[mediagoblin.git] / mediagoblin / tests / test_storage.py
index f6f1d18f532004b9e4280638fab10e1dfc70554c..2194741596135a2039ab941b101701a95e7b4b6d 100644 (file)
@@ -19,6 +19,8 @@ import os
 import tempfile
 
 import pytest
+import six
+
 from werkzeug.utils import secure_filename
 
 from mediagoblin import storage
@@ -78,7 +80,7 @@ def test_storage_system_from_config():
              'mediagoblin.tests.test_storage:FakeStorageSystem'})
     assert this_storage.foobie == 'eiboof'
     assert this_storage.blech == 'hcelb'
-    assert unicode(this_storage.__class__) == \
+    assert six.text_type(this_storage.__class__) == \
         u'mediagoblin.tests.test_storage.FakeStorageSystem'