Files
pre-commit-hooks/pre_commit_hooks/tests_should_end_in_test.py
2015-01-04 15:03:56 -08:00

27 lines
634 B
Python

from __future__ import print_function
import argparse
import sys
def validate_files(argv=None):
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*')
args = parser.parse_args(argv)
retcode = 0
for filename in args.filenames:
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(validate_files())