def test_dump_no_overwrite():
d = MinyDict({"a": 1, "b": 2, "u": "x", "r": {"t": 5}})
p = Path(d.to_json("d.json"))
with pytest.raises(FileExistsError):
d.to_json(p, allow_overwrite=False)
p.unlink()
p = Path(d.to_pickle("d.pkl"))
with pytest.raises(FileExistsError):
d.to_pickle(p, allow_overwrite=False)
assert MinyDict.from_pickle(p) == d
p.unlink()
p = Path(d.to_yaml("d.yaml"))
with pytest.raises(FileExistsError):
d.to_yaml(p, allow_overwrite=False)
assert MinyDict.from_yaml(p) == d
p.unlink()
def test_defaults():
examples = Path(__file__).resolve().parent.parent / "examples"
d1 = examples / "demo.json"
d2 = examples / "demo2.json"
y1 = examples / "demo.yaml"
p = MinyDict({"a": "2", "c": 3, "d": {"e": {"f": 4, "g": 5}}})
pkl = p.to_pickle(Path(__file__).resolve().parent / "test.pkl")
with patch.object(sys, "argv", [""]):
args = minydra.resolved_args(defaults=p)
assert args == p
with patch.object(sys, "argv", ["", "d.e.f=2"]):
args = minydra.resolved_args(defaults=p)
assert args.d.e.f == 2
with patch.object(sys, "argv", ["", f"@defaults={str(d1)}"]):
args = minydra.resolved_args()
del args["@defaults"]
assert args.to_dict() == json.loads(d1.read_text())
with patch.object(sys, "argv", ["", f"@defaults={str(y1)}"]):
args = minydra.resolved_args()
del args["@defaults"]
assert args.to_dict() == MinyDict.from_yaml(y1)
with patch.object(sys, "argv", ["", f"@defaults={str(pkl)}"]):
args = minydra.resolved_args()
del args["@defaults"]
assert args.to_dict() == MinyDict.from_pickle(pkl)
Path(pkl).unlink()
with pytest.raises(ValueError):
with patch.object(
sys, "argv", ["", f"@defaults={str(d1).replace('.json', '.py')}"]
):
args = minydra.resolved_args()
with pytest.raises(KeyError):
with patch.object(sys, "argv", ["", f"@defaults={str(d1)}", "new_key=3"]):
args = minydra.resolved_args()
del args["@defaults"]
assert args.to_dict() == json.loads(d1.read_text())
with patch.object(
sys, "argv", ["", f"@defaults={str(d1)}", "@strict=false", "new_key=3"]
):
args = minydra.resolved_args(keep_special_kwargs=False)
target = json.loads(d1.read_text())
target["new_key"] = 3
assert args.to_dict() == target
with patch.object(
sys, "argv", ["", f"@defaults={str(d1)}", "@strict=false", "new_key=3"]
):
args = minydra.resolved_args()
del args["@defaults"]
del args["@strict"]
target = json.loads(d1.read_text())
target["new_key"] = 3
assert args.to_dict() == target
double_defaults = f"['{str(d1)}', '{str(d2)}']"
with patch.object(sys, "argv", ["", f"@defaults={double_defaults}", "new_key=3"]):
args = minydra.resolved_args()
d1d = MinyDict.from_json(d1)
d2d = MinyDict.from_json(d2)
d1d = d1d.update(d2d)
del args["@defaults"]
assert args == d1d
def test_dump_pickle():
d = MinyDict({"a": 1, "b": 2, 4: "x", "r": {"t": 5}})
p = Path(d.to_pickle("d.pkl", verbose=1))
assert MinyDict.from_pickle(p) == d
p.unlink()
