def test_is_a_generator(self): """Walk the tree top-down.""" result = walk(self.basedir) for _ in xrange(len(list(os.walk(self.basedir)))): dirpath, dirnames, filenames = result.next() self.assertRaises(StopIteration, result.next)
def keys(self): """ returns a iterator over the keys """ splitext = os.path.splitext for dirpath, dirnames, filenames in walk(self._path): for filename in filenames: # just in case a .new file was left around ext = splitext(filename)[1] if ext != ".new" and ext != ".old": yield filename
def test_top_down(self, topdown=True, expected=None): """Walk the tree top-down.""" if expected is None: result = os.walk(self.basedir, topdown) expected = self._build_dict_from_walk(result) actual = self._build_dict_from_walk(walk(self.basedir, topdown)) self.assertEqual(sorted(actual.keys()), sorted(expected.keys())) for dirpath, values in expected.iteritems(): self.assertEqual(values['dirnames'], actual[dirpath]['dirnames']) self.assertEqual(values['filenames'], actual[dirpath]['filenames'])
def test_len_result(self): """Lenght of result is correct.""" result = list(walk(self.basedir)) self.assertEqual(len(result), len(list(os.walk(self.basedir))))