compabibility with python2 (unicode)

test/test_main.py

@@ -16,22 +16,28 @@
 # along with phonemizer. If not, see <http://www.gnu.org/licenses/>.
 """Test of the phonemizer.Phonemizer class"""
 
+import codecs
 import pytest
 import tempfile
 import shlex
 
 from phonemizer import main, festival
 
 
-def _test(input, output, args=''):
+def _test(input, expected_output, args=''):
     with tempfile.NamedTemporaryFile('w', delete=False) as finput:
         finput.write(input)
         finput.seek(0)
 
         with tempfile.NamedTemporaryFile('w+', delete=False) as foutput:
             opts = '{} -o {} {}'.format(finput.name, foutput.name, args)
             main.main(shlex.split(opts))
-            assert foutput.read() == output + '\n'
+            output = foutput.read()
+            try:  # python2 needs additional utf8 decoding
+                output = output.decode('utf8')
+            except AttributeError:  # python3 is OK
+                pass
+            assert output == expected_output + '\n'
 
 def test_help():
     with pytest.raises(SystemExit):