From fe9d02cf7191c0ab1fe949cb2fa4770a478c0340 Mon Sep 17 00:00:00 2001 From: Adam Janovsky Date: Mon, 16 Nov 2020 11:25:08 +0100 Subject: demo of dataset deserialization --- oop_demo.py | 8 +++++++- 1 file changed, 7 insertions(+), 1 deletion(-) diff --git a/oop_demo.py b/oop_demo.py index 2d76a6c4..455c0ad7 100644 --- a/oop_demo.py +++ b/oop_demo.py @@ -1,4 +1,4 @@ -from sec_certs.dataset import CCDataset, DatasetJSONEncoder +from sec_certs.dataset import CCDataset, DatasetJSONEncoder, DatasetJSONDecoder from pathlib import Path from datetime import datetime import logging @@ -20,6 +20,12 @@ def main(): with open('./debug_dataset/cc_full_dataset.json', 'w') as handle: json.dump(dset, handle, cls=DatasetJSONEncoder, indent=4) + # Load dataset from JSON + with open('./debug_dataset/cc_full_dataset.json', 'r') as handle: + new_dset = json.load(handle, cls=DatasetJSONDecoder) + + assert dset == new_dset + end = datetime.now() logging.info(f'The computation took {(end-start)} seconds.') -- cgit v1.3.1