Merge pull request #1755 from avanwinkle/stepper-yaml-bytes

Bugfix: Pololu Stepper status parsing

Author: avanwinkle

mpf/platforms/pololu/pololu_ticcmd_wrapper.py

@@ -3,7 +3,6 @@
 import logging
 import asyncio
 
-from io import StringIO
 from threading import Thread
 import ruamel.yaml
 
@@ -98,8 +97,10 @@ def _run_subprocess_ticcmd(self, *args):
     async def get_status(self):
         """Return the current status of the TIC device."""
         cmd_return = await self._ticcmd_future('-s', '--full')
+        self.log.debug("get_status() received cmd_return: %s", cmd_return)
+        value = cmd_return.decode('utf-8')
         yaml = ruamel.yaml.YAML(typ='safe')
-        status = yaml.load(StringIO(cmd_return))
+        status = yaml.load(value)
         return status
 
     def halt_and_hold(self):

mpf/tests/test_PololuTic.py

@@ -32,7 +32,7 @@ def _ticcmd_async(self, *args):
                 yaml = ruamel.yaml.YAML()
                 yaml.Dumper = RoundTripDumper
                 yaml.dump(new_status, output)
-                return output.getvalue()
+                return output.getvalue().encode()
 
         elif args == ('--reset-command-timeout',):
             return ""