assert that some input results in an exception being raised

test_parameterized_exception.py

import pytest
from typing import List

def method_under_test(values: List[int]) -> int:
    """dummy method"""
    if 0 in values:
        raise(ValueError("Zeros are not supported"))
    return sum(values)

@pytest.mark.parametrize(
    "flat,expected",
    [
        ...
        ([0,2,3], ValueError),
        ([1,2,3], 6),
        ... ,
    ],
)
def test_collapse(value, expected):
    if type(expected) == type and issubclass(expected, Exception):
        with pytest.raises(expected):
            method_under_test(value)
    else:
        assert method_under_test(value) == expected