How To Structure Unit Tests Python