def test_walk_fs_with_git(tmp_dir, scm): tmp_dir.gen({ "foo": "foo", "bar": "bar", "тест": "проверка", "code.py": "import sys\nimport shutil\n" "shutil.copyfile(sys.argv[1], sys.argv[2])", "data_dir": { "data": "data", "data_sub_dir": { "data_sub": "data_sub" }, }, }) fs = LocalFileSystem(url=str(tmp_dir)) walk_result = [] for root, dirs, files in fs.walk("."): dirs[:] = [i for i in dirs if i != ".git"] walk_result.append((root, dirs, files)) assert convert_to_sets(walk_result) == [ (".", {"data_dir"}, {"bar", "тест", "code.py", "foo"}), (join("data_dir"), {"data_sub_dir"}, {"data"}), (join("data_dir", "data_sub_dir"), set(), {"data_sub"}), ] walk_result = fs.walk(join("data_dir", "data_sub_dir")) assert convert_to_sets(walk_result) == [(join("data_dir", "data_sub_dir"), set(), {"data_sub"})]
def test_walk_no_scm(tmp_dir): tmp_dir.gen({ "foo": "foo", "bar": "bar", "тест": "проверка", "code.py": "import sys\nimport shutil\n" "shutil.copyfile(sys.argv[1], sys.argv[2])", "data_dir": { "data": "data", "data_sub_dir": { "data_sub": "data_sub" }, }, }) fs = LocalFileSystem() walk_results = fs.walk(str(tmp_dir)) assert convert_to_sets(walk_results) == [ (str(tmp_dir), {"data_dir"}, {"code.py", "bar", "тест", "foo"}), (str(tmp_dir / "data_dir"), {"data_sub_dir"}, {"data"}), (str(tmp_dir / "data_dir" / "data_sub_dir"), set(), {"data_sub"}), ] walk_results = fs.walk(join("data_dir", "data_sub_dir")) assert convert_to_sets(walk_results) == [ (join("data_dir", "data_sub_dir"), set(), {"data_sub"}), ]
def test_nobranch(self): fs = LocalFileSystem(None, {"url": self._root_dir}, use_dvcignore=True) self.assertWalkEqual( fs.walk("."), [ (".", ["data_dir"], ["bar", "тест", "code.py", "foo"]), (join("data_dir"), ["data_sub_dir"], ["data"]), (join("data_dir", "data_sub_dir"), [], ["data_sub"]), ], ) self.assertWalkEqual( fs.walk(join("data_dir", "data_sub_dir")), [(join("data_dir", "data_sub_dir"), [], ["data_sub"])], )
def test_nobranch(self): fs = LocalFileSystem(url=self._root_dir) walk_result = [] for root, dirs, files in fs.walk("."): dirs[:] = [i for i in dirs if i != ".git"] walk_result.append((root, dirs, files)) self.assertWalkEqual( walk_result, [ (".", ["data_dir"], ["bar", "тест", "code.py", "foo"]), (join("data_dir"), ["data_sub_dir"], ["data"]), (join("data_dir", "data_sub_dir"), [], ["data_sub"]), ], ) self.assertWalkEqual( fs.walk(join("data_dir", "data_sub_dir")), [(join("data_dir", "data_sub_dir"), [], ["data_sub"])], )
def test(self): fs = LocalFileSystem(None, {"url": self._root_dir}) self.assertWalkEqual( fs.walk(self._root_dir), [ ( self._root_dir, ["data_dir"], ["code.py", "bar", "тест", "foo"], ), (join(self._root_dir, "data_dir"), ["data_sub_dir"], ["data"]), ( join(self._root_dir, "data_dir", "data_sub_dir"), [], ["data_sub"], ), ], )
def test_subdir(self): fs = LocalFileSystem(None, {"url": self._root_dir}) self.assertWalkEqual( fs.walk(join("data_dir", "data_sub_dir")), [(join("data_dir", "data_sub_dir"), [], ["data_sub"])], )
def test_subdir(self): fs = LocalFileSystem() self.assertWalkEqual( fs.walk(join("data_dir", "data_sub_dir")), [(join("data_dir", "data_sub_dir"), [], ["data_sub"])], )