| 22 | class AstScaningTest(unittest.TestCase): |
| 23 | |
| 24 | def setUp(self): |
| 25 | print(('Testing %s.%s' % (type(self).__name__, self._testMethodName))) |
| 26 | self.tmp_dir = tempfile.TemporaryDirectory().name |
| 27 | self.tmp_dir2 = tempfile.TemporaryDirectory().name |
| 28 | self.test_file = os.path.join(self.tmp_dir, 'test.py') |
| 29 | if not os.path.exists(self.tmp_dir): |
| 30 | os.makedirs(self.tmp_dir) |
| 31 | |
| 32 | fnames = ['1.py', '2.py', '3.py', '__init__.py'] |
| 33 | self.folders = ['.', 'a', 'b', 'c'] |
| 34 | dir_path = self.tmp_dir2 |
| 35 | folder_dirs = [ |
| 36 | os.path.join(dir_path, folder) for folder in self.folders |
| 37 | ] |
| 38 | for folder in folder_dirs: |
| 39 | os.makedirs(folder, exist_ok=True) |
| 40 | for fname in fnames: |
| 41 | fpath = os.path.join(folder, fname) |
| 42 | with open(fpath, 'w') as f: |
| 43 | f.write('hello world') |
| 44 | |
| 45 | for folder in folder_dirs: |
| 46 | print(f'folder: {os.listdir(folder)}') |
| 47 | |
| 48 | def tearDown(self): |
| 49 | super().tearDown() |