pep8 improvements on bok-choy tests.

@jzoldak @muhammad-ammar @clytwynec