A little cleanup to payout tests.

* PEP8, some docstrings, remove dead imports