Binary Image Classification Using Python Assert Function To Validate