If you want to test your function with different examples, use the pytest.mark.parametrize decorator.
pytest.mark.parametrize
My previous tips on testing.