really fix profiling this time I promise

Author: pipermerriam

ethereum/tests/profile_vm.py

@@ -3,14 +3,19 @@
 import ethereum.testutils as testutils
 import cProfile
 import pstats
-from io import BytesIO
 import time
 from rlp.utils import encode_hex
 from ethereum.utils import sha3, to_string
 from ethereum.slogging import get_logger
 logger = get_logger()
 
 
+if sys.version_info.major == 2:
+    from io import BytesIO as StringIO
+else:
+    from io import StringIO
+
+
 def do_test_vm(filename, testname=None, testdata=None, limit=99999999, profiler=None):
     logger.debug('running test:%r in %r' % (testname, filename))
     testutils.run_vm_test(testutils.fixture_to_bytes(testdata), testutils.VERIFY, profiler=profiler)
@@ -39,7 +44,7 @@ def run(profiler=None):
     if len(sys.argv) == 1:
         pr = cProfile.Profile()
         run(pr)
-        s = BytesIO()
+        s = StringIO()
         sortby = 'tottime'
         ps = pstats.Stats(pr, stream=s).sort_stats(sortby)
         ps.print_stats(50)