How To Test A Function In Python