Pytest parameters

import mymath
import pytest

@pytest.mark.parametrize("a,b,expected", [
    (1, 1, 2),
    (2, 3, 5),
    (-1, 1, 0),
    (-1, -1, -2),
])
def test_add(a, b, expected):
    assert mymath.add(a, b)  == expected