six package in test suite (fix)

Ousret · Sep 28, 2017 · ea65bd4 · ea65bd4
1 parent 863a8d5
commit ea65bd4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_tpe.py b/test/test_tpe.py
@@ -29,11 +29,11 @@ def s_name(self):
 
     @staticmethod
     def _has_signal(data, signal):
-        return data[0] == curses.ascii.controlnames.index(signal)
+        return (data[0] if six.PY3 else ord(data[0])) == curses.ascii.controlnames.index(signal)
 
     @staticmethod
     def _create_signal(signal):
-        return bytes([curses.ascii.controlnames.index(signal)])
+        return bytes([curses.ascii.controlnames.index(signal)]) if six.PY3 else bytes(chr(curses.ascii.controlnames.index(signal)))
 
     def _wait_signal(self, signal):
         return FakeTeliumDevice._has_signal(os.read(self._master, 1), signal)