def test_save_history(model_name: str, new_history: History):
save_history(new_history, model_name)
history = load_history(model_name)
assert history == new_history
def model_history(model):
"""groom images to remove inaccurate masks"""
plot_history(model, load_history(model))
def test_load_history(model_name: str):
history = load_history(model_name)
for key in ('accuracy', 'loss', 'val_accuracy', 'val_loss'):
assert key in history
def plot_history_wrapper(args):
plot_history(args.model, load_history(args.model))