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

1import matplotlib.figure as mpl 

2import matplotlib.pyplot as plt 

3from prefect import task 

4 

5 

6@task 

7def make_line_figure(data: list[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 item in data: 

14 color = item["color"] if "color" in item else "#000000" 

15 ax.plot(item["x"], item["y"], color=color, linewidth=0.5) 

16 

17 return fig