Made pylint work for the test class