mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-02 07:18:25 +01:00
71 lines
2.0 KiB
Python
71 lines
2.0 KiB
Python
import os
|
|
|
|
from dvc.stage.cache import RunCacheNotSupported
|
|
from dvc.utils.fs import remove
|
|
|
|
|
|
def test_repro_pulls_missing_data_source(tmp_dir, dvc, mocker, local_remote):
|
|
(foo,) = tmp_dir.dvc_gen("foo", "foo")
|
|
|
|
dvc.push()
|
|
|
|
dvc.stage.add(name="copy-foo", cmd="cp foo bar", deps=["foo"], outs=["bar"])
|
|
remove("foo")
|
|
remove(foo.outs[0].cache_path)
|
|
|
|
assert dvc.reproduce(pull=True)
|
|
|
|
|
|
def test_repro_pulls_missing_import(tmp_dir, dvc, mocker, erepo_dir, local_remote):
|
|
with erepo_dir.chdir():
|
|
erepo_dir.dvc_gen("foo", "foo", commit="first")
|
|
|
|
foo_import = dvc.imp(os.fspath(erepo_dir), "foo")
|
|
|
|
dvc.push()
|
|
|
|
dvc.stage.add(name="copy-foo", cmd="cp foo bar", deps=["foo"], outs=["bar"])
|
|
remove("foo")
|
|
remove(foo_import.outs[0].cache_path)
|
|
|
|
assert dvc.reproduce(pull=True)
|
|
|
|
|
|
def test_repro_pulls_continue_without_run_cache(tmp_dir, dvc, mocker, local_remote):
|
|
(foo,) = tmp_dir.dvc_gen("foo", "foo")
|
|
|
|
dvc.push()
|
|
mocker.patch.object(
|
|
dvc.stage_cache, "pull", side_effect=RunCacheNotSupported("foo")
|
|
)
|
|
dvc.stage.add(name="copy-foo", cmd="cp foo bar", deps=["foo"], outs=["bar"])
|
|
remove("foo")
|
|
remove(foo.outs[0].cache_path)
|
|
|
|
assert dvc.reproduce(pull=True)
|
|
|
|
|
|
def test_repro_skip_pull_if_no_run_cache_is_passed(tmp_dir, dvc, mocker, local_remote):
|
|
(foo,) = tmp_dir.dvc_gen("foo", "foo")
|
|
|
|
dvc.push()
|
|
spy_pull = mocker.spy(dvc.stage_cache, "pull")
|
|
dvc.stage.add(name="copy-foo", cmd="cp foo bar", deps=["foo"], outs=["bar"])
|
|
remove("foo")
|
|
remove(foo.outs[0].cache_path)
|
|
|
|
assert dvc.reproduce(pull=True, run_cache=False)
|
|
assert not spy_pull.called
|
|
|
|
|
|
def test_repro_skip_pull_if_single_item_is_passed(tmp_dir, dvc, mocker, local_remote):
|
|
(foo,) = tmp_dir.dvc_gen("foo", "foo")
|
|
|
|
dvc.push()
|
|
|
|
dvc.stage.add(name="copy-foo", cmd="cp foo bar", deps=["foo"], outs=["bar"])
|
|
remove("foo")
|
|
remove(foo.outs[0].cache_path)
|
|
|
|
assert dvc.reproduce(pull=True, single_item=True)
|