def test_valid_type_organization(self): file_list = ['file.doc', 'file.docx', 'file.pdf', 'song.mp3', 'song.wav', 'some torrent.torrent', 'picture.png', 'picture.jpg', 'exec.exe', 'exec.bin', 'exec.app', 'exec.osx', 'exec.msi', 'presentation.pptx'] organizer = Organizer('C:') organizer.file_list = file_list files_by_type = organizer.organize_by_type() self.assertCountEqual(files_by_type['Documents'], ['file.doc', 'file.docx', 'file.pdf']) self.assertCountEqual(files_by_type['Music'], ['song.wav', 'song.mp3']) self.assertCountEqual(files_by_type['Torrents'], ['some torrent.torrent']) self.assertCountEqual(files_by_type['Pictures'], ['picture.png', 'picture.jpg']) self.assertCountEqual(files_by_type['Executables'], ['exec.exe', 'exec.bin', 'exec.app', 'exec.osx', 'exec.msi']) self.assertCountEqual(files_by_type['Presentations'], ['presentation.pptx'])
def test_valid_extension_organization(self): file_list = ['file1.docx', 'file2.exe.exe', 'file3.docx.exe', 'file5.sdx', 'some.power'] organizer = Organizer('C:') organizer.file_list = file_list files_by_extensions = organizer.organize_by_extension() self.assertCountEqual(files_by_extensions['exe'], ['file2.exe.exe', 'file3.docx.exe']) self.assertCountEqual(files_by_extensions['docx'], ['file1.docx']) self.assertCountEqual(files_by_extensions['power'], ['some.power'])