FilesExpand file tree

hydit

/

conftest.py

Copy path

Blame

More file actions

Blame

More file actions

Latest commit

 

History

History

History

36 lines (28 loc) · 1.39 KB

hydit

/

conftest.py

Copy path

Top

File metadata and controls

• Code
• Blame

36 lines (28 loc) · 1.39 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

import os

import pytest

# Command line arguments for pytest

def pytest_addoption(parser):

parser.addoption('--output_dir', action="store", default='tests/inference/samples', help='Output directory for generated images')

parser.addoption("--listen", type=str, default="127.0.0.1", metavar="IP", nargs="?", const="0.0.0.0", help="Specify the IP address to listen on (default: 127.0.0.1). If --listen is provided without an argument, it defaults to 0.0.0.0. (listens on all)")

parser.addoption("--port", type=int, default=8188, help="Set the listen port.")

# This initializes args at the beginning of the test session

@pytest.fixture(scope="session", autouse=True)

def args_pytest(pytestconfig):

args = {}

args['output_dir'] = pytestconfig.getoption('output_dir')

args['listen'] = pytestconfig.getoption('listen')

args['port'] = pytestconfig.getoption('port')

os.makedirs(args['output_dir'], exist_ok=True)

return args

def pytest_collection_modifyitems(items):

# Modifies items so tests run in the correct order

LAST_TESTS = ['test_quality']

# Move the last items to the end

last_items = []

for test_name in LAST_TESTS:

for item in items.copy():

print(item.module.__name__, item)

if item.module.__name__ == test_name:

last_items.append(item)

items.remove(item)

items.extend(last_items)