From b97f9f542d00cd1e1c40407f897094315ad874e9 Mon Sep 17 00:00:00 2001 From: Tarek Ziade Date: Tue, 21 Feb 2012 10:37:12 +0100 Subject: [PATCH] io under py3 --- flake8/tests/test_mccabe.py | 5 ++++- 1 file changed, 4 insertions(+), 1 deletion(-) diff --git a/flake8/tests/test_mccabe.py b/flake8/tests/test_mccabe.py index f1ae82b..5d3f4c1 100644 --- a/flake8/tests/test_mccabe.py +++ b/flake8/tests/test_mccabe.py @@ -1,6 +1,9 @@ import unittest import sys -from StringIO import StringIO +try: + from StringIO import StringIO +except ImportError: + from io import StringIO from flake8.mccabe import get_code_complexity