Small cleanup in test library generator

R=sgjesse@google.com

Review URL: https://codereview.chromium.org//486593002