Shared via CrossHair Playground

main.py

import json

BYTES_PER_ENTITY = 507

def estimate_storage(entity_count: int) -> str:
    """
    pre: 0 <= entity_count < 1000
    post: len(__return__) < 19
    """
    
    kib = entity_count * BYTES_PER_ENTITY
    size = f"{kib}kib" if kib < 1024 else f"{kib // 1024}mib"
    return json.dumps({"size": size})