Merge pull request #604 from ccordoba12/flaky-test

Mark a couple of tests as flaky

qtconsole/tests/test_comms.py

@@ -1,8 +1,9 @@
 import time
 from queue import Empty
-
 import unittest
 
+from flaky import flaky
+
 from qtconsole.manager import QtKernelManager
 
 
@@ -29,7 +30,6 @@ def setUp(self):
             self._get_next_msg()
             self._get_next_msg()
 
-
     def tearDown(self):
         """Close the kernel."""
         if self.kernel_manager:
@@ -51,6 +51,7 @@ def _get_next_msg(self, timeout=10):
                 pass
         return msg
 
+    @flaky(max_runs=10)
     def test_kernel_to_frontend(self):
         """Communicate from the kernel to the frontend."""
         comm_manager = self.comm_manager
@@ -104,6 +105,7 @@ def comm_message(self, msg):
         msg = self._get_next_msg()
         assert msg['header']['msg_type'] == 'stream'
 
+    @flaky(max_runs=10)
     def test_frontend_to_kernel(self):
         """Communicate from the frontend to the kernel."""
         comm_manager = self.comm_manager
@@ -144,5 +146,6 @@ def test_frontend_to_kernel(self):
         assert msg['msg_type'] == 'stream'
         assert msg['content']['text'] == 'close\n'
 
+
 if __name__ == "__main__":
     unittest.main()