mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-10 07:18:27 +01:00
Ruslan Kuprieiev
8a7ccab82f
Pre-requisite for removing fs.path_info and switching to regular strings as paths in filesystems. With new filesystems we no longer need the zoo of different dependency and output types, as most of them are now just handled by the underlying filesystem. Though deps/outs are still due for a refactor in other aspects.
75 lines
2.0 KiB
Python
75 lines
2.0 KiB
Python
import os
|
|
|
|
from mock import patch
|
|
|
|
from dvc.hash_info import HashInfo
|
|
from dvc.output import Output
|
|
from dvc.stage import Stage
|
|
from dvc.utils import relpath
|
|
from tests.basic_env import TestDvc
|
|
|
|
|
|
def test_str_workdir_outside_repo(tmp_dir, erepo_dir):
|
|
stage = Stage(erepo_dir.dvc)
|
|
output = Output(stage, "path", cache=False)
|
|
|
|
assert relpath("path", erepo_dir.dvc.root_dir) == str(output)
|
|
|
|
|
|
def test_str_workdir_inside_repo(dvc):
|
|
stage = Stage(dvc)
|
|
output = Output(stage, "path", cache=False)
|
|
|
|
assert "path" == str(output)
|
|
|
|
stage = Stage(dvc, wdir="some_folder")
|
|
output = Output(stage, "path", cache=False)
|
|
|
|
assert os.path.join("some_folder", "path") == str(output)
|
|
|
|
|
|
def test_str_on_local_absolute_path(dvc):
|
|
stage = Stage(dvc)
|
|
|
|
rel_path = os.path.join("path", "to", "file")
|
|
abs_path = os.path.abspath(rel_path)
|
|
output = Output(stage, abs_path, cache=False)
|
|
|
|
assert output.def_path == rel_path
|
|
assert output.path_info.fspath == abs_path
|
|
assert str(output) == rel_path
|
|
|
|
|
|
def test_str_on_external_absolute_path(dvc):
|
|
stage = Stage(dvc)
|
|
|
|
rel_path = os.path.join("..", "path", "to", "file")
|
|
abs_path = os.path.abspath(rel_path)
|
|
output = Output(stage, abs_path, cache=False)
|
|
|
|
assert output.def_path == abs_path
|
|
assert output.path_info.fspath == abs_path
|
|
assert str(output) == abs_path
|
|
|
|
|
|
class TestGetFilesNumber(TestDvc):
|
|
def _get_output(self):
|
|
stage = Stage(self.dvc)
|
|
return Output(stage, "path")
|
|
|
|
def test_return_0_on_no_cache(self):
|
|
o = self._get_output()
|
|
o.use_cache = False
|
|
self.assertEqual(0, o.get_files_number())
|
|
|
|
def test_return_multiple_for_dir(self):
|
|
o = self._get_output()
|
|
o.hash_info = HashInfo("md5", "12345678.dir", nfiles=2)
|
|
self.assertEqual(2, o.get_files_number())
|
|
|
|
@patch.object(Output, "is_dir_checksum", False)
|
|
def test_return_1_on_single_file_cache(self):
|
|
o = self._get_output()
|
|
o.hash_info = HashInfo("md5", "12345678")
|
|
self.assertEqual(1, o.get_files_number())
|