def test_yaml(self):
options = {"table-is-field-0": True}
fmt = "yaml"
filename = self.tempdir.name + "/" + fmt
fs = FileStorage(fmt, filename, options=options)
fs.store(self.data)
self.check_serialized_data(fs, filename, self.data)
def run_test(self, fmt, lossy=False):
options = {"table-is-field-0": True}
filename = self.tempdir.name + "/" + fmt
fs = FileStorage(fmt, filename, options=options)
fs.store(self.data)
m = fs.load()
self.compare(self.data, m, lossy)
self.check_serialized_data(fs, filename, m)
def compression_test(self, format):
tempdir = tempfile.TemporaryDirectory()
options = {"compression": format}
magic = self.magic[format]
filename = tempdir.name + "/dtb"
fs = FileStorage("json", filename, options=options)
fs.store(self.data)
m = fs.load()
self.compare(self.data, m)
with open(filename, "rb") as fp:
header = fp.read(len(magic))
self.assertEqual(header, magic)
tempdir.cleanup()
def test_default(self):
tempdir = tempfile.TemporaryDirectory()
filename = tempdir.name + "/dtb"
fs = FileStorage("json", filename)
fs.store(self.data)
m = fs.load()
self.compare(self.data, m)
with open(filename, "rb") as fp:
header = fp.read(20)
for k, v in self.magic.items():
self.assertNotEqual(header[0 : len(v)], v)
self.assertEqual(header[0:1], b"[")
tempdir.cleanup()
