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, [],
[]))