From 13ca5bdf28045e44503b73bc2f09abd6246881e2 Mon Sep 17 00:00:00 2001 From: Aman Sharma Date: Thu, 2 Jan 2020 01:16:32 +0530 Subject: [PATCH] Write test for `walk_files` to check if it accepts Path-like and str objects --- tests/unit/utils/test_utils.py | 5 +++++ 1 file changed, 5 insertions(+) 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))