refs #62: fix broken image tests

* TODO: we need to add test cases for build/exec stages.
lablup · Oct 22, 2017 · 08ae9eb · 08ae9eb
1 parent 1ff88e1
commit 08ae9eb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/tests/test.py b/tests/test.py
@@ -162,7 +162,7 @@ def execute(self, code, user_input=None):
             codecs.getincrementaldecoder('utf8')(),
         )
         try:
-            msg_in = (b'', code.encode('utf8'))
+            msg_in = (b'code', code.encode('utf8'))
             kin.send_multipart(msg_in)
             while True:
                 if kout.poll(timeout=10_000) == 0:  # 10 sec
@@ -174,7 +174,7 @@ def execute(self, code, user_input=None):
                     break
                 elif reply_type == b'waiting-input':
                     assert isinstance(user_input, str), 'User input required'
-                    msg_in = (b'', user_input.encode('utf8'))
+                    msg_in = (b'code', user_input.encode('utf8'))
                     time.sleep(0.1)
                     kin.send_multipart(msg_in)
                 elif reply_type == b'stdout':