qubed/tree_compresser/tests/open_climate_dt.py
2024-12-13 14:48:33 +00:00

18 lines
596 B
Python

import json
from pathlib import Path
from tree_traverser import CompressedTree
data_path = Path("/home/eouser/qubed/config/climate-dt/compressed_tree.json")
# Print size of file
print(f"climate dt compressed tree: {data_path.stat().st_size // 1e6:.1f} MB")
print("Opening json file")
compressed_tree = CompressedTree.load(data_path)
print(compressed_tree.reconstruct_compressed_ecmwf_style())
# print("Outputting compressed tree ecmwf style")
# with open("data/compressed_tree_climate_dt_ecmwf_style.json", "w") as f:
# json.dump(compressed_tree.reconstruct_compressed_ecmwf_style(), f)