def test_at_root_yields_only_once(self, mock_getcwd): """We should only get root once. # noqa""" mock_getcwd.return_value = '/' path_walker = walk_path() self.assertEqual(next(path_walker), '/') with self.assertRaises(StopIteration): next(path_walker)
def test_really_long_path(self, mock_getcwd): directories = [ ''.join( [choice(ascii_letters + '_-') for _ in range(randint(1, 10))]) for __ in range(randint(10, 30)) ] cwd = '/' + '/'.join(directories) mock_getcwd.return_value = cwd path_walker = walk_path() paths_walked = [x for x in path_walker] self.assertEqual( len(paths_walked), len(directories) + 1, 'Should have had {} but had {} paths.'.format( len(directories), len(paths_walked) + 1, ))