pre_commit_hooks/tests_should_end_in_test.py

from __future__ import print_function

import sys

from pre_commit_hooks.util import entry


@entry
def validate_files(argv):
    retcode = 0
    for filename in argv:
        if (
            not filename.endswith('_test.py') and
            not filename.endswith('__init__.py') and
            not filename.endswith('/conftest.py')
        ):
            retcode = 1
            print('{0} does not end in _test.py'.format(filename))

    return retcode


if __name__ == '__main__':
    sys.exit(entry())