mirror of
https://github.com/Stone-Red-Code/dvc.git
synced 2026-11-04 15:28:27 +01:00
Alexander Schepanovski
fcebad821e
- `tmp_dir` descends from Path - convenient template methods to generate files and dirs, add them to dvc and git and commit to git - independent of old basic_env structures - can go in any order - modular: start with empty `tmp_dir`, initialize git with `scm`, dvc repo with `dvc` add "repo template" with `repo_template`, ...
77 lines
2.0 KiB
Python
77 lines
2.0 KiB
Python
import pytest
|
|
|
|
from dvc.stage import StageCommitError
|
|
from dvc.utils.stage import dump_stage_file
|
|
from dvc.utils.stage import load_stage_file
|
|
|
|
|
|
def test_commit_recursive(dvc_repo, repo_dir):
|
|
stages = dvc_repo.add(repo_dir.DATA_DIR, recursive=True, no_commit=True)
|
|
assert len(stages) == 2
|
|
|
|
assert dvc_repo.status() != {}
|
|
|
|
dvc_repo.commit(repo_dir.DATA_DIR, recursive=True)
|
|
|
|
assert dvc_repo.status() == {}
|
|
|
|
|
|
def test_commit_force(dvc_repo, repo_dir):
|
|
stages = dvc_repo.add(repo_dir.FOO, no_commit=True)
|
|
assert len(stages) == 1
|
|
stage = stages[0]
|
|
|
|
with dvc_repo.state:
|
|
assert stage.outs[0].changed_cache()
|
|
|
|
with open(repo_dir.FOO, "a") as fobj:
|
|
fobj.write(repo_dir.FOO_CONTENTS)
|
|
|
|
with dvc_repo.state:
|
|
assert stage.outs[0].changed_cache()
|
|
|
|
with pytest.raises(StageCommitError):
|
|
dvc_repo.commit(stage.path)
|
|
|
|
with dvc_repo.state:
|
|
assert stage.outs[0].changed_cache()
|
|
|
|
dvc_repo.commit(stage.path, force=True)
|
|
|
|
assert dvc_repo.status([stage.path]) == {}
|
|
|
|
|
|
def test_commit_with_deps(tmp_dir, dvc, run_copy):
|
|
tmp_dir.gen("foo", "foo")
|
|
foo_stage, = dvc.add("foo", no_commit=True)
|
|
assert foo_stage is not None
|
|
assert len(foo_stage.outs) == 1
|
|
|
|
stage = run_copy("foo", "file", no_commit=True)
|
|
assert stage is not None
|
|
assert len(stage.outs) == 1
|
|
|
|
with dvc.state:
|
|
assert foo_stage.outs[0].changed_cache()
|
|
assert stage.outs[0].changed_cache()
|
|
|
|
dvc.commit(stage.path, with_deps=True)
|
|
with dvc.state:
|
|
assert not foo_stage.outs[0].changed_cache()
|
|
assert not stage.outs[0].changed_cache()
|
|
|
|
|
|
def test_commit_changed_md5(dvc_repo, repo_dir):
|
|
stages = dvc_repo.add(repo_dir.FOO, no_commit=True)
|
|
assert len(stages) == 1
|
|
stage = stages[0]
|
|
|
|
stage_file_content = load_stage_file(stage.path)
|
|
stage_file_content["md5"] = "1111111111"
|
|
dump_stage_file(stage.path, stage_file_content)
|
|
|
|
with pytest.raises(StageCommitError):
|
|
dvc_repo.commit(stage.path)
|
|
|
|
dvc_repo.commit(stage.path, force=True)
|