Reproduce and fix #11

nbrmd/nbrmd.py

@@ -14,6 +14,7 @@
 # -----------------------------------------------------------------------------
 
 import os
+import io
 import re
 from enum import Enum
 from nbformat.v4.rwbase import NotebookReader, NotebookWriter
@@ -350,7 +351,7 @@ def readf(nb_file):
     :return: the notebook
     """
     file, ext = os.path.splitext(nb_file)
-    with open(nb_file) as fp:
+    with io.open(nb_file, encoding='utf-8') as fp:
         if ext == '.Rmd':
             return read(fp)
         elif ext == '.md':
@@ -372,7 +373,7 @@ def writef(nb, nb_file):
     """
 
     file, ext = os.path.splitext(nb_file)
-    with open(nb_file, 'w') as fp:
+    with io.open(nb_file, 'w', encoding='utf-8') as fp:
         if ext == '.Rmd':
             write(nb, fp)
         elif ext == '.md':

tests/test_unicode.py

@@ -0,0 +1,19 @@
+# coding: utf-8
+import nbrmd
+import pytest
+from .utils import list_all_notebooks
+
+
+@pytest.mark.parametrize('nb_file', list_all_notebooks('.ipynb') +
+                         list_all_notebooks('.Rmd'))
+def test_notebook_contents_is_unicode(nb_file):
+    nb = nbrmd.readf(nb_file)
+
+    for cell in nb.cells:
+        assert cell.source == '' or isinstance(cell.source, unicode)
+
+
+def test_write_non_ascii(tmpdir):
+    nb = nbrmd.reads(u'Non-ascii contênt')
+    nbrmd.writef(nb, str(tmpdir.join('notebook.Rmd')))
+    nbrmd.writef(nb, str(tmpdir.join('notebook.ipynb')))