Updating test to use pytest.mark.parametrize