Add tests.
Automatic style corrections courtesy of autopep8.