def test_save_load(): path = 'myvars.pkl' vars = {'a': 1, 'b': '2'} save_vars(path, vars) vars2 = load_vars(path, list(vars.keys())) assert vars == vars2 os.remove(path)
def test_load_fail(): path = 'myvars.pkl' load_vars(path, ['a', 'b'])