Spaces:
Running
Running
| # Ultralytics 🚀 AGPL-3.0 License - https://ultralytics.com/license | |
| import shutil | |
| from pathlib import Path | |
| from tests import TMP | |
| def pytest_addoption(parser): | |
| """ | |
| Add custom command-line options to pytest. | |
| Args: | |
| parser (pytest.config.Parser): The pytest parser object for adding custom command-line options. | |
| Returns: | |
| (None) | |
| """ | |
| parser.addoption("--slow", action="store_true", default=False, help="Run slow tests") | |
| def pytest_collection_modifyitems(config, items): | |
| """ | |
| Modify the list of test items to exclude tests marked as slow if the --slow option is not specified. | |
| Args: | |
| config (pytest.config.Config): The pytest configuration object that provides access to command-line options. | |
| items (list): The list of collected pytest item objects to be modified based on the presence of --slow option. | |
| Returns: | |
| (None) The function modifies the 'items' list in place, and does not return a value. | |
| """ | |
| if not config.getoption("--slow"): | |
| # Remove the item entirely from the list of test items if it's marked as 'slow' | |
| items[:] = [item for item in items if "slow" not in item.keywords] | |
| 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. | |
| Returns: | |
| (None) | |
| """ | |
| from ultralytics.utils.torch_utils import init_seeds | |
| 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 used for terminal output. | |
| exitstatus (int): The exit status of the test run. | |
| config (pytest.config.Config): The pytest config object. | |
| Returns: | |
| (None) | |
| """ | |
| from ultralytics.utils import WEIGHTS_DIR | |
| # Remove files | |
| models = [path for x in ["*.onnx", "*.torchscript"] for path in WEIGHTS_DIR.rglob(x)] | |
| for file in ["decelera_portrait_min.mov", "bus.jpg", "yolo11n.onnx", "yolo11n.torchscript"] + models: | |
| Path(file).unlink(missing_ok=True) | |
| # Remove directories | |
| models = [path for x in ["*.mlpackage", "*_openvino_model"] for path in WEIGHTS_DIR.rglob(x)] | |
| for directory in [WEIGHTS_DIR / "path with spaces", TMP.parents[1] / ".pytest_cache", TMP] + models: | |
| shutil.rmtree(directory, ignore_errors=True) | |