Pytest Test One Function