def test_machinelogs(self):
"""Test anonymization of multiple machine logs."""
logs = MachineLogs()
logs.append(osp.join(self.anon_folder, 'A1234_patientid.dlg'))
logs.append(osp.join(self.anon_folder, 'PatientID_4DC Treatment_JST90_TX_20140712094246.bin'))
logs.anonymize() # shouldn't raise
def test_anonymize(self):
logs = MachineLogs(self.logs_dir, recursive=False)
files = logs.anonymize(inplace=False, suffix='_suffixed')
self.assertIsInstance(files, list)
# cleanup
for pdir, sdir, files in os.walk(self.logs_dir):
to_remove = [file for file in files if 'suffixed' in file]
for file in to_remove:
os.remove(osp.join(pdir, file))
