refactor(tests): add comments to some test packages and clean up with goimports