boards: simulator: Improve spi.write() simulation

Signed-off-by: Daniel Thompson <daniel@redfelineninja.org.uk>

1 files changed, 5 insertions, 0 deletions

diff --git a/wasp/boards/simulator/display.py b/wasp/boards/simulator/display.py
index 8efbb9c..379760c 100644
--- a/wasp/boards/simulator/display.py
+++ b/wasp/boards/simulator/display.py
@@ -37,6 +37,11 @@ class ST7789Sim(object):
         self.cmd = 0
 
     def write(self, data):
+        # Converting data to a memoryview ensures we act more like spi.write()
+        # when running in a real device (e.g. data must be  bytes-like object
+        # that implements the buffer protocol)
+        data = memoryview(data)
+
         if len(data) == 1:
             # Assume if we get a byte at a time then it is command.
             # This is a simplification do we don't have to track