How To Write Unit Tests Python