How To Test Class In Python