@@ -145,11 +145,7 @@ async def test_async_list_operations(codebox: CodeBox):
145
145
def test_sync_stream_exec (codebox : CodeBox ):
146
146
chunks : list [tuple [ExecChunk , float ]] = []
147
147
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
153
149
for chunk in codebox .stream_exec (
154
150
f"import time;\n for i in range(3): time.sleep({ sleep } ); print(i)"
155
151
):
@@ -181,11 +177,7 @@ def test_sync_stream_exec(codebox: CodeBox):
181
177
async def test_sync_stream_exec_ipython (codebox : CodeBox ):
182
178
chunks = []
183
179
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
189
181
for chunk in codebox .stream_exec (
190
182
f"python -u -c 'import time\n for i in range(3): time.sleep({ sleep } ); print(i)'" ,
191
183
kernel = "bash" ,
@@ -218,11 +210,7 @@ async def test_sync_stream_exec_ipython(codebox: CodeBox):
218
210
async def test_async_stream_exec_ipython (codebox : CodeBox ):
219
211
chunks : list [tuple [ExecChunk , float ]] = []
220
212
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
226
214
async for chunk in codebox .astream_exec (
227
215
f"import time;\n for i in range(3): time.sleep({ sleep } ); print(i)" ,
228
216
):
@@ -254,11 +242,7 @@ async def test_async_stream_exec_ipython(codebox: CodeBox):
254
242
async def test_async_stream_exec_bash (codebox : CodeBox ):
255
243
chunks = []
256
244
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
262
246
async for chunk in codebox .astream_exec (
263
247
f"python -u -c 'import time\n for i in range(3): time.sleep({ sleep } ); print(i)'" ,
264
248
kernel = "bash" ,
0 commit comments