Skip to content

Commit 1cce923

Browse files
committed
fix test chunk sep sleep time
1 parent b7b5651 commit 1cce923

File tree

1 file changed

+4
-20
lines changed

1 file changed

+4
-20
lines changed

tests/test_v02.py

Lines changed: 4 additions & 20 deletions
Original file line numberDiff line numberDiff line change
@@ -145,11 +145,7 @@ async def test_async_list_operations(codebox: CodeBox):
145145
def 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;\nfor i in range(3): time.sleep({sleep}); print(i)"
155151
):
@@ -181,11 +177,7 @@ def test_sync_stream_exec(codebox: CodeBox):
181177
async 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\nfor 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):
218210
async 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;\nfor i in range(3): time.sleep({sleep}); print(i)",
228216
):
@@ -254,11 +242,7 @@ async def test_async_stream_exec_ipython(codebox: CodeBox):
254242
async 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\nfor i in range(3): time.sleep({sleep}); print(i)'",
264248
kernel="bash",

0 commit comments

Comments
 (0)