jupyter#751 fixed python2 kernel output parsing for tests

twosigma · Mar 22, 2018 · a7d8cc3 · a7d8cc3
1 parent a301aa4
commit a7d8cc3
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/nbconvert/preprocessors/tests/test_execute.py b/nbconvert/preprocessors/tests/test_execute.py
@@ -59,6 +59,8 @@ def normalize_output(output):
                 ['model_id'] = '<MODEL_ID>'
         for key, value in output.get('data', {}).items():
             if isinstance(value, string_types):
+                if sys.version_info.major == 2:
+                    value = value.replace('u\'', '\'')
                 output['data'][key] = _normalize_base64(value)
         if 'traceback' in output:
             tb = [