Stone_Red/dvc
1
0
Fork 0
mirror of https://github.com/Stone-Red-Code/dvc.git synced 2026-11-14 07:18:27 +01:00
Code Issues Packages Projects Releases Wiki Activity

get_files_number test fix

Browse Source
This commit is contained in:
Casper da Costa-Luis
2019-08-20 14:27:20 +01:00
parent 8faa02e837
commit 4bd2ce25e1
1 changed files with 7 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files
tests/unit/output/test_local.py
+7 -2
View File
@@ -2,6 +2,7 @@ from mock import patch
from dvc.stage import Stage
from dvc.output import OutputLOCAL
from dvc.remote.local import RemoteLOCAL
from tests.basic_env import TestDvc
@@ -32,8 +33,12 @@ class TestGetFilesNumber(TestDvc):
self.assertEqual(0, o.get_files_number())
@patch.object(OutputLOCAL, "checksum", "12345678.dir")
@patch.object(OutputLOCAL, "dir_cache", [{"md5": "asdf"}, {"md5": "qwe"}])
def test_return_mutiple_for_dir(self):
@patch.object(
RemoteLOCAL,
"get_dir_cache",
return_value=[{"md5": "asdf"}, {"md5": "qwe"}],
)
def test_return_mutiple_for_dir(self, mock_get_dir_cache):
o = self._get_output()
self.assertEqual(2, o.get_files_number())