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))))
