PEP8 fixes for aggregation tests

I recently worked on `test_aggregate_by.py` and found it in need of some PEP8 love. Pretty much every method had very long lines making it hard to read.