(dir_to_check, kind='test')
| 65 | |
| 66 | |
| 67 | def get_files(dir_to_check, kind='test'): |
| 68 | files = dict() |
| 69 | patterns = { |
| 70 | 'test': f'{dir_to_check}/**/test_*.py', |
| 71 | 'stub': f'{dir_to_check}/**/*.pyi', |
| 72 | } |
| 73 | for path in glob.glob(patterns[kind], recursive=True): |
| 74 | relpath = os.path.relpath(path, dir_to_check) |
| 75 | files[relpath] = path |
| 76 | |
| 77 | return files |
| 78 | |
| 79 | |
| 80 | if __name__ == '__main__': |