def test_yaml(): data = {'bob': 1, 'tom': 2, 'sam': 3} fname = 'test.yaml' fs = FileStorage() fs.writeYaml(fname, data) fs.clear() fs.readYaml(fname) # print fs.db os.remove(fname) assert fs.db == data
def test_json(): data = {'bob': 1, 'tom': 2, 'sam': 3} fname = 'test.json' fs = FileStorage() fs.writeJson(fname, data) fs.clear() fs.readJson(fname) # print fs.db os.remove(fname) assert fs.db == data