Assert In Python Function