cleaned up the tests formatting a bit to match pep8.