Example of chex.variants with pytest

[chMoussa]

Hi,

How would someone use chex.variants for a pytest function like this?

@pytest.mark.parametrize(
    "n",
    [
       1,2,3
    ],
)
def test_example(n: int) -> None:

[shreed27]

Hi @chMoussa,

Good question — this is a common point of confusion for me also when combining
pytest.mark.parametrize with chex.variants.

I think this would be best addressed with a small, concrete pytest example
showing how chex.variants can wrap a parametrized test and run it across
different execution modes (e.g. with and without jax.jit), with comments
explaining the pattern.

I’m happy to take this issue up and add a minimal, runnable example test if that
sounds useful. Please let me know if you have any preference on where such an
example should live (new test vs existing test file).

Thanks!