Example #1
0
def test_meta_is_preserved(tmp_dir, dvc):
    (stage, ) = tmp_dir.dvc_gen("foo", "foo content")

    # Add meta to DVC-file
    data = load_stage_file(stage.path)
    data["meta"] = {"custom_key": 42}
    dump_stage_file(stage.path, data)

    # Loading and dumping to test that it works and meta is retained
    dvcfile = Dvcfile(dvc, stage.path)
    new_stage = dvcfile.load()
    dvcfile.dump(new_stage)

    new_data = load_stage_file(stage.path)
    assert new_data["meta"] == data["meta"]
Example #2
0
    def test(self):
        stages = self.dvc.add(self.FOO)
        self.assertEqual(len(stages), 1)
        stage = stages[0]
        self.assertTrue(stage is not None)

        d = load_stage_file(stage.relpath)

        # NOTE: checking that reloaded stage didn't change its checksum
        md5 = "11111111111111111111111111111111"
        d[stage.PARAM_MD5] = md5
        dump_stage_file(stage.relpath, d)

        dvcfile = Dvcfile(self.dvc, stage.relpath)
        stage = dvcfile.load()

        self.assertTrue(stage is not None)
        dvcfile.dump(stage)

        d = load_stage_file(stage.relpath)
        self.assertEqual(d[stage.PARAM_MD5], md5)