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"]
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)