Add flake8 check on test files
Add the missing max line length check
Use assertEquals