diff --git a/tests/test_util.py b/tests/test_util.py index 5a036a0..033c745 100644 --- a/tests/test_util.py +++ b/tests/test_util.py @@ -205,7 +205,7 @@ def test_2_2_links(self): ('bx', 'b'), ('Dir/cx', 'Dir/c'), ('Dir/dx', 'Dir/d'), - ('DirX', 'Dir'), + ('DirX', 'Dir', True), ]) results = set(iter_tree_files(self.temp_dir)) self.assertEqual(results, set(map(ospath, [ diff --git a/tests/util.py b/tests/util.py index 301427b..29b95b6 100644 --- a/tests/util.py +++ b/tests/util.py @@ -48,10 +48,11 @@ def make_links(temp_dir: pathlib.Path, links: Iterable[Tuple[str, str]]) -> None the destination link path (:class:`str`) and source node path (:class:`str`). """ - for link, node in links: + for link, node, *opt in links: src = temp_dir / ospath(node) dest = temp_dir / ospath(link) - os.symlink(src, dest) + is_dir = opt[0] if opt else False + os.symlink(src, dest, target_is_directory=is_dir) def mkfile(file: pathlib.Path) -> None: