FilesExpand file tree

 async

/

test_model.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

19 lines (15 loc) · 534 Bytes

 async

/

test_model.py

Copy path

Top

File metadata and controls

• Code
• Blame

19 lines (15 loc) · 534 Bytes

Raw

Copy raw file

Download raw file

Open symbols panel

Edit and raw actions

1

2

3

4

5

6

7

8

9

10

11

12

13

14

15

16

17

18

19

import pytest

import replicate

@pytest.mark.vcr("models-get.yaml")

@pytest.mark.asyncio

@pytest.mark.parametrize("async_flag", [True, False])

async def test_models_get(async_flag):

if async_flag:

model = await replicate.async_models.get("stability-ai", "sdxl")

else:

model = replicate.models.get("stability-ai", "sdxl")

assert model is not None

assert model.owner == "stability-ai"

assert model.name == "sdxl"

assert model.visibility == "public"

assert model.latest_version is not None