def test_scalability_2(self): # test that the function does not take too long list_of_folders = [str(i) for i in range(100)] dcm_files = ['{}.dcm'.format(i) for i in range(200)] filelist = [(os.path.join(a, b), 0, 1) for a, b in itertools.product(list_of_folders, dcm_files)] s_time = time.time() DataCleaner.clean_files_by_size_optimized(filelist, reduce_size=100000000, pattern='*dcm') e_time = time.time() self.assertLess(e_time - s_time, 1.2)
def test_clean_files_by_size_whitelist(self): filelist = [('file1', 0, 10), ('file2', 0, 10), ('file3', 0, 10), ('file4', 0, 10)] removed = DataCleaner.clean_files_by_size_optimized( filelist, 15, ['file1'], 'file*') self.assertEqual([('file2', 0, 10), ('file3', 0, 10)], removed)
def test_clean_files_by_size_1(self): self.assertEqual([], DataCleaner.clean_files_by_size_optimized([], 1, [], []))