Update tests for Python 3

StatisKit · Sep 6, 2017 · 3a07c37 · 3a07c37
1 parent db3644c
commit 3a07c37
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/test/test_feedback.py b/test/test_feedback.py
@@ -15,6 +15,7 @@
 ##################################################################################
 
 import autowig
+import six
 
 import unittest
 from nose.plugins.attrib import attr
@@ -143,9 +144,10 @@ def test_with_none_overload_export(self, overload="none"):
                              stdout=subprocess.PIPE, stderr=subprocess.PIPE)
         prev, curr = s.communicate()
 
-        while not prev == curr:
+        while curr and not prev == curr:
             prev = curr
-            curr = curr.decode('ascii', 'ignore')
+            if six.PY3:
+                curr = curr.decode('ascii', 'ignore')
             code = autowig.feedback(curr, '.', asg)
             if code:
                 exec(code, locals())