add tests
autopep8 tests