def my_add(a, b):
    return a + b


def my_mul(a, b):
    return a * b


def test_my_add():
    assert my_add(5, 6) == 11
    assert my_add(5, 1) == 6


def test_my_mul():
    assert my_mul(2, 4) == 8
