Test A Function With Pytest