Merge pull request #333 from chadlung/master

Fix PEP8 in test