How To Input Test Cases In Python