diff --git a/tests/unit/utils/test_utils.py b/tests/unit/utils/test_utils.py index ec99d8de71..15b2c541f4 100644 --- a/tests/unit/utils/test_utils.py +++ b/tests/unit/utils/test_utils.py @@ -12,6 +12,7 @@ from dvc.utils import relpath from dvc.utils import to_chunks from dvc.utils import tmp_fname +from dvc.utils import walk_files from tests.basic_env import TestDir @@ -147,3 +148,7 @@ def test_relpath(): path_info = PathInfo(path) assert relpath(path) == relpath(path_info) + + +def test_walk_files(tmp_dir): + assert list(walk_files(".")) == list(walk_files(tmp_dir))