Skip to content

Commit b9d803c

Browse files
committed
use file output in tests
1 parent eb72919 commit b9d803c

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_run.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -32,17 +32,20 @@ async def test_run(async_flag, record_mode):
3232
output = await replicate.async_run(
3333
f"stability-ai/sdxl:{version}",
3434
input=input,
35+
use_file_output=True,
3536
)
3637
else:
3738
output = replicate.run(
3839
f"stability-ai/sdxl:{version}",
3940
input=input,
41+
use_file_output=True,
4042
)
4143

4244
assert output is not None
4345
assert isinstance(output, list)
4446
assert len(output) > 0
45-
assert output[0].startswith("https://")
47+
assert isinstance(output[0], FileOutput)
48+
assert output[0].url.startswith("https://")
4649

4750

4851
@pytest.mark.vcr("run__concurrently.yaml")

0 commit comments

Comments
 (0)