FilesExpand file tree

 main

/

test_client.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

21 lines (17 loc) · 638 Bytes

 main

/

test_client.py

Copy path

Top

File metadata and controls

• Code
• Blame

21 lines (17 loc) · 638 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

import responses

from replicate.__about__ import __version__

from replicate.client import Client

from responses import matchers

@responses.activate

def test_client_sets_authorization_token_and_user_agent_headers():

client = Client(api_token="abc123")

model = client.models.get("test/model")

responses.get(

"https://api.replicate.com/v1/models/test/model/versions",

match=[

matchers.header_matcher({"Authorization": "Token abc123"}),

matchers.header_matcher({"User-Agent": f"replicate-python@{__version__}"}),

],

json={"results": []},

)

model.versions.list()