demo of dataset deserialization

1 files changed, 7 insertions, 1 deletions

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.')