pytest parametrize

inline-snapshot can also be used with pytest.mark.parametrize.

One parametrize

You can use snapshot() as a parameter for your test.

import pytest
from inline_snapshot import snapshot


@pytest.mark.parametrize(
    "a,b,result",
    [
        (1, 2, snapshot()),
        (3, 4, snapshot()),
    ],
)
def test_param(a, b, result):
    assert a + b == result

The missing values will be created on the next run.

import pytest
from inline_snapshot import snapshot


@pytest.mark.parametrize(
    "a,b,result",
    [
        (1, 2, snapshot(3)),
        (3, 4, snapshot(7)),
    ],
)
def test_param(a, b, result):
    assert a + b == result

Multiple parametrize

Stacking @pytest.mark.parametrize creates a cross-product of all combinations, so snapshot() can no longer be used as an argument. Instead, use a single snapshot object that stores all combinations indexed by a stable key.

import pytest
from inline_snapshot import snapshot

snapshot_results = snapshot()


@pytest.mark.parametrize("a", [1, 2, 3])
@pytest.mark.parametrize("b", [10, 20, 30])
def test_parametrize_stack(a, b):
    assert snapshot_results[f"{a}+{b}"] == a + b

Once the snapshots are created, it looks like this:

import pytest
from inline_snapshot import snapshot

snapshot_results = snapshot(
    {
        "1+10": 11,
        "2+10": 12,
        "3+10": 13,
        "1+20": 21,
        "2+20": 22,
        "3+20": 23,
        "1+30": 31,
        "2+30": 32,
        "3+30": 33,
    }
)


@pytest.mark.parametrize("a", [1, 2, 3])
@pytest.mark.parametrize("b", [10, 20, 30])
def test_parametrize_stack(a, b):
    assert snapshot_results[f"{a}+{b}"] == a + b

Store values externally

You can use outsource() when you want to store larger values.

import pytest
from inline_snapshot import external, outsource, snapshot

external_snapshots = snapshot()


@pytest.mark.parametrize("a", [1, 2])
@pytest.mark.parametrize("b", [10, 20])
def test_parametrize_stack_external(a, b):
    assert external_snapshots[f"[{a}]*{b}"] == outsource([a] * b)

The values will be stored in external JSON files when you run pytest.

import pytest
from inline_snapshot import external, outsource, snapshot

external_snapshots = snapshot(
    {
        "[1]*10": external("uuid:e443df78-9558-467f-9ba9-1faf7a024204.json"),
        "[2]*10": external("uuid:23a7711a-8133-4876-b7eb-dcd9e87a1613.json"),
        "[1]*20": external("uuid:1846d424-c17c-4279-a3c6-612f48268673.json"),
        "[2]*20": external("uuid:fcbd04c3-4021-4ef7-8ca5-a5a19e4d6e3c.json"),
    }
)


@pytest.mark.parametrize("a", [1, 2])
@pytest.mark.parametrize("b", [10, 20])
def test_parametrize_stack_external(a, b):
    assert external_snapshots[f"[{a}]*{b}"] == outsource([a] * b)