pyo3-dlpack 0.3.0

Zero-copy DLPack tensor interop for PyO3
Documentation 
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
37
38
39

"""
Pytest configuration for pyo3-dlpack integration tests.
"""

import pytest


def pytest_configure(config):
    """Configure custom markers."""
    config.addinivalue_line("markers", "cpu: mark test as CPU-only")
    config.addinivalue_line("markers", "gpu: mark test as requiring CUDA GPU")
    config.addinivalue_line("markers", "slow: mark test as slow")


def pytest_collection_modifyitems(config, items):
    """Add markers based on test class names."""
    for item in items:
        # Add gpu marker to GPU test classes
        if "Gpu" in item.nodeid or "gpu" in item.nodeid:
            item.add_marker(pytest.mark.gpu)
        # Add cpu marker to CPU test classes
        elif "Cpu" in item.nodeid or "cpu" in item.nodeid:
            item.add_marker(pytest.mark.cpu)


def pytest_addoption(parser):
    """Add custom command line options."""
    parser.addoption(
        "--run-gpu",
        action="store_true",
        default=False,
        help="Run GPU tests (requires CUDA)",
    )
    parser.addoption(
        "--cpu-only",
        action="store_true",
        default=False,
        help="Run only CPU tests",
    )