Coverage for src/cell_abm_pipeline/tasks/make_histogram_figure.py: 0%

1import matplotlib.figure as mpl 

2import matplotlib.pyplot as plt 

3from prefect import task 

4 

5 

6@task 

7def make_histogram_figure(keys: list[str], data: dict) -> mpl.Figure: 

8 fig = plt.figure(figsize=(4, 4), constrained_layout=True) 

9 

10 ax = fig.add_subplot() 

11 ax.set_box_aspect(1) 

12 

13 for key in keys: 

14 width = data["*"]["bandwidth"] 

15 edges = [item["x"] for item in data[key]] + [width] 

16 values = [item["y"] for item in data[key]] 

17 ax.stairs(values, edges) 

18 

19 return fig