import numpy as np
from metacausal.aggregation import BootstrapResult
from metacausal.plots import forest

boot = BootstrapResult(
    ate=5.0, ate_ci_lower=1.0, ate_ci_upper=9.0,
    boot_ates=np.array([3.0, 5.0, 7.0]),
    cate=None, cate_ci_lower=None, cate_ci_upper=None,
    component_boot_ates={
        "a": np.array([2.0, 4.0, 6.0]),
        "b": np.array([4.0, 6.0, 8.0]),
    },
)
forest(boot)