lightning/tests/tests_fabric/utilities/test_runif.py

import pytest

from tests_fabric.helpers.runif import RunIf


@RunIf(min_torch="99")
def test_always_skip():
    exit(1)


@pytest.mark.parametrize("arg1", [0.5, 1.0, 2.0])
@RunIf(min_torch="0.0")
def test_wrapper(arg1: float):
    assert arg1 > 0.0