blob: b8cf9152a1761fd2dcb1522744e767d6a9621ba6 [file] [log] [blame]
import argparse
import os.path
import re
from typing import Optional
from typing import Sequence
def main(argv: Optional[Sequence[str]] = None) -> int:
parser = argparse.ArgumentParser()
parser.add_argument('filenames', nargs='*')
'--django', default=False, action='store_true',
help='Use Django-style test naming pattern (test*.py)',
args = parser.parse_args(argv)
retcode = 0
test_name_pattern = r'test.*\.py' if args.django else r'.*_test\.py'
for filename in args.filenames:
base = os.path.basename(filename)
if (
not re.match(test_name_pattern, base) and
not base == '' and
not base == ''
retcode = 1
print(f'{filename} does not match pattern "{test_name_pattern}"')
return retcode
if __name__ == '__main__':