tests: fix flake8 issues