mark tests as flaky

blink1073 · Nov 9, 2023 · 89f62bb · 89f62bb
1 parent 1e31abd
commit 89f62bb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 0 deletions.
diff --git a/tests/services/kernels/test_api.py b/tests/services/kernels/test_api.py
@@ -7,6 +7,7 @@
 import jupyter_client
 import pytest
 import tornado
+from flaky import flaky
 from jupyter_client.kernelspec import NATIVE_KERNEL_NAME
 from tornado.httpclient import HTTPClientError
 
@@ -256,6 +257,7 @@ async def test_kernel_handler_startup_error_pending(
         await jp_ws_fetch("api", "kernels", kid, "channels")
 
 
+@flaky
 @pytest.mark.timeout(TEST_TIMEOUT)
 async def test_connection(jp_fetch, jp_ws_fetch, jp_http_port, jp_auth_header):
     # Create kernel

diff --git a/tests/services/sessions/test_api.py b/tests/services/sessions/test_api.py
@@ -9,6 +9,7 @@
 import jupyter_client
 import pytest
 import tornado
+from flaky import flaky
 from jupyter_client.ioloop import AsyncIOLoopKernelManager
 from nbformat import writes
 from nbformat.v4 import new_notebook
@@ -506,6 +507,7 @@ async def test_modify_kernel_id(session_client, jp_fetch, jp_serverapp, session_
         assert kernel_list == [kernel]
 
 
+@flaky
 @pytest.mark.timeout(TEST_TIMEOUT)
 async def test_restart_kernel(session_client, jp_base_url, jp_fetch, jp_ws_fetch, session_is_ready):
     # Create a session.