@@ -145,11 +145,7 @@ async def test_async_list_operations(codebox: CodeBox):
145145def test_sync_stream_exec (codebox : CodeBox ):
146146 chunks : list [tuple [ExecChunk , float ]] = []
147147 t0 = time .perf_counter ()
148- sleep = 0.5
149- if codebox .api_key == "local" :
150- sleep = 0.01
151- if codebox .api_key == "docker" :
152- sleep = 0.05
148+ sleep = 0.01 if codebox .api_key == "local" else 1
153149 for chunk in codebox .stream_exec (
154150 f"import time;\n for i in range(3): time.sleep({ sleep } ); print(i)"
155151 ):
@@ -181,11 +177,7 @@ def test_sync_stream_exec(codebox: CodeBox):
181177async def test_sync_stream_exec_ipython (codebox : CodeBox ):
182178 chunks = []
183179 t0 = time .perf_counter ()
184- sleep = 0.5
185- if codebox .api_key == "local" :
186- sleep = 0.01
187- if codebox .api_key == "docker" :
188- sleep = 0.05
180+ sleep = 0.01 if codebox .api_key == "local" else 1
189181 for chunk in codebox .stream_exec (
190182 f"python -u -c 'import time\n for i in range(3): time.sleep({ sleep } ); print(i)'" ,
191183 kernel = "bash" ,
@@ -218,11 +210,7 @@ async def test_sync_stream_exec_ipython(codebox: CodeBox):
218210async def test_async_stream_exec_ipython (codebox : CodeBox ):
219211 chunks : list [tuple [ExecChunk , float ]] = []
220212 t0 = time .perf_counter ()
221- sleep = 0.5
222- if codebox .api_key == "local" :
223- sleep = 0.01
224- if codebox .api_key == "docker" :
225- sleep = 0.05
213+ sleep = 0.01 if codebox .api_key == "local" else 1
226214 async for chunk in codebox .astream_exec (
227215 f"import time;\n for i in range(3): time.sleep({ sleep } ); print(i)" ,
228216 ):
@@ -254,11 +242,7 @@ async def test_async_stream_exec_ipython(codebox: CodeBox):
254242async def test_async_stream_exec_bash (codebox : CodeBox ):
255243 chunks = []
256244 t0 = time .perf_counter ()
257- sleep = 0.5
258- if codebox .api_key == "local" :
259- sleep = 0.01
260- if codebox .api_key == "docker" :
261- sleep = 0.05
245+ sleep = 0.01 if codebox .api_key == "local" else 1
262246 async for chunk in codebox .astream_exec (
263247 f"python -u -c 'import time\n for i in range(3): time.sleep({ sleep } ); print(i)'" ,
264248 kernel = "bash" ,
0 commit comments