FilesExpand file tree

 main

/

conftest.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

31 lines (23 loc) · 801 Bytes

 main

/

conftest.py

Copy path

Top

File metadata and controls

• Code
• Blame

31 lines (23 loc) · 801 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

import os

from unittest import mock

import pytest

skip_if_no_token = pytest.mark.skipif(

os.environ.get("REPLICATE_API_TOKEN") is None, reason="REPLICATE_API_TOKEN not set"

)

@pytest.fixture(scope="session")

def mock_replicate_api_token(scope="class"):

if os.environ.get("REPLICATE_API_TOKEN", "") != "":

yield

else:

with mock.patch.dict(

os.environ,

{"REPLICATE_API_TOKEN": "test-token", "REPLICATE_POLL_INTERVAL": "0.0"},

):

yield

@pytest.fixture(scope="module")

def vcr_config():

return {"allowed_hosts": ["api.replicate.com"], "filter_headers": ["authorization"]}

@pytest.fixture(scope="module")

def vcr_cassette_dir(request):

module = request.node.fspath

return os.path.join(module.dirname, "cassettes")