summaryrefslogtreecommitdiff
path: root/factory/compat.py
diff options
context:
space:
mode:
Diffstat (limited to 'factory/compat.py')
-rw-r--r--factory/compat.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/factory/compat.py b/factory/compat.py
index 7c9b845..f77458f 100644
--- a/factory/compat.py
+++ b/factory/compat.py
@@ -31,10 +31,15 @@ PY2 = (sys.version_info[0] == 2)
if PY2: # pragma: no cover
def is_string(obj):
return isinstance(obj, (str, unicode))
+
+ from StringIO import StringIO as BytesIO
+
else: # pragma: no cover
def is_string(obj):
return isinstance(obj, str)
+ from io import BytesIO
+
try: # pragma: no cover
# Python >= 3.2
UTC = datetime.timezone.utc