Write Test Cases In Python