FilesExpand file tree

 async

/

test_version.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

33 lines (24 loc) · 986 Bytes

 async

/

test_version.py

Copy path

Top

File metadata and controls

• Code
• Blame

33 lines (24 loc) · 986 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

20

21

22

23

24

25

26

27

28

29

30

31

32

33

import pytest

import replicate

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

@pytest.mark.asyncio

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

async def test_models_versions_get(async_flag):

id = "a00d0b7dcbb9c3fbb34ba87d2d5b46c56969c84a628bf778a7fdaec30b1b99c5"

if async_flag:

version = await replicate.async_models.versions.get("stability-ai", "sdxl", id)

else:

version = replicate.models.versions.get("stability-ai", "sdxl", id)

assert version.id == id

@pytest.mark.vcr("models-versions-list.yaml")

@pytest.mark.asyncio

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

async def test_models_versions_list(async_flag):

if async_flag:

page = await replicate.async_models.versions.list("stability-ai", "sdxl")

else:

page = replicate.models.versions.list("stability-ai", "sdxl")

assert page.previous is None

assert len(page.results) > 0

version = page.results[0]

assert version.id is not None