FilesExpand file tree

 main

/

test_identifier.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

57 lines (54 loc) · 1.46 KB

 main

/

test_identifier.py

Copy path

Top

File metadata and controls

• Code
• Blame

57 lines (54 loc) · 1.46 KB

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

34

35

36

37

38

39

40

41

42

43

44

45

46

47

48

49

50

51

52

53

54

55

56

57

import pytest

from replicate.identifier import ModelVersionIdentifier

@pytest.mark.parametrize(

"id, expected",

[

(

"meta/llama-2-70b-chat",

{

"owner": "meta",

"name": "llama-2-70b-chat",

"version": None,

"error": False,

},

),

(

"mistralai/mistral-7b-instruct-v1.4",

{

"owner": "mistralai",

"name": "mistral-7b-instruct-v1.4",

"version": None,

"error": False,

},

),

(

"nateraw/video-llava:a494250c04691c458f57f2f8ef5785f25bc851e0c91fd349995081d4362322dd",

{

"owner": "nateraw",

"name": "video-llava",

"version": "a494250c04691c458f57f2f8ef5785f25bc851e0c91fd349995081d4362322dd",

"error": False,

},

),

(

"",

{"error": True},

),

(

"invalid",

{"error": True},

),

(

"invalid/id/format",

{"error": True},

),

],

)

def test_parse_model_id(id, expected):

try:

result = ModelVersionIdentifier.parse(id)

assert result.owner == expected["owner"]

assert result.name == expected["name"]

assert result.version == expected["version"]

except ValueError:

assert expected["error"]