Add linting for tests and corresponding fixes