def test_get_files_recursive(self): """ Test that we can find files recusively """ file_finder = FileFinder(path=self.DUD_DIR_PREFIX, recursive=True) files = file_finder.get_files() self.assertEqual(len(files), sum(self.FILES_PER_DIR))
def test_get_files_non_recursive(self): """ Test that we can find files non-recusively """ file_finder = FileFinder(path=self.DUD_DIR_PREFIX, recursive=False) files = file_finder.get_files() self.assertEqual(len(files), self.FILES_PER_DIR[len(self.FILES_PER_DIR) - 1])