mirror of
https://github.com/THU-MIG/yolov10.git
synced 2025-05-23 21:44:22 +08:00
80 lines
2.5 KiB
Python
80 lines
2.5 KiB
Python
# Ultralytics YOLO 🚀, AGPL-3.0 license
|
|
|
|
import shutil
|
|
from pathlib import Path
|
|
|
|
import pytest
|
|
|
|
from ultralytics.utils import ROOT
|
|
from ultralytics.utils.torch_utils import init_seeds
|
|
|
|
TMP = (ROOT / '../tests/tmp').resolve() # temp directory for test files
|
|
|
|
|
|
def pytest_addoption(parser):
|
|
"""Add custom command-line options to pytest.
|
|
|
|
Args:
|
|
parser (pytest.config.Parser): The pytest parser object.
|
|
"""
|
|
parser.addoption('--slow', action='store_true', default=False, help='Run slow tests')
|
|
|
|
|
|
def pytest_configure(config):
|
|
"""Register custom markers to avoid pytest warnings.
|
|
|
|
Args:
|
|
config (pytest.config.Config): The pytest config object.
|
|
"""
|
|
config.addinivalue_line('markers', 'slow: mark test as slow to run')
|
|
|
|
|
|
def pytest_collection_modifyitems(config, items):
|
|
"""Modify collected test items based on custom command-line options.
|
|
|
|
Args:
|
|
config (pytest.config.Config): The pytest config object.
|
|
items (list): List of collected test items.
|
|
"""
|
|
if not config.getoption('--slow'):
|
|
skip_slow = pytest.mark.skip(reason="remove this test because it's slow")
|
|
for item in items:
|
|
if 'slow' in item.keywords:
|
|
item.add_marker(skip_slow)
|
|
|
|
|
|
def pytest_sessionstart(session):
|
|
"""
|
|
Initialize session configurations for pytest.
|
|
|
|
This function is automatically called by pytest after the 'Session' object has been created but before performing
|
|
test collection. It sets the initial seeds and prepares the temporary directory for the test session.
|
|
|
|
Args:
|
|
session (pytest.Session): The pytest session object.
|
|
"""
|
|
init_seeds()
|
|
shutil.rmtree(TMP, ignore_errors=True) # delete any existing tests/tmp directory
|
|
TMP.mkdir(parents=True, exist_ok=True) # create a new empty directory
|
|
|
|
|
|
def pytest_terminal_summary(terminalreporter, exitstatus, config):
|
|
"""
|
|
Cleanup operations after pytest session.
|
|
|
|
This function is automatically called by pytest at the end of the entire test session. It removes certain files
|
|
and directories used during testing.
|
|
|
|
Args:
|
|
terminalreporter (pytest.terminal.TerminalReporter): The terminal reporter object.
|
|
exitstatus (int): The exit status of the test run.
|
|
config (pytest.config.Config): The pytest config object.
|
|
"""
|
|
# Remove files
|
|
for file in ['bus.jpg', 'decelera_landscape_min.mov']:
|
|
Path(file).unlink(missing_ok=True)
|
|
|
|
# Remove directories
|
|
for directory in [ROOT / '../.pytest_cache', TMP]:
|
|
shutil.rmtree(directory, ignore_errors=True)
|