diff --git a/examples/des_y1_3x2pt/des_y1_3x2pt_PT.py b/examples/des_y1_3x2pt/des_y1_3x2pt_PT.py index 67f1eb9e..c8681791 100755 --- a/examples/des_y1_3x2pt/des_y1_3x2pt_PT.py +++ b/examples/des_y1_3x2pt/des_y1_3x2pt_PT.py @@ -339,6 +339,9 @@ def plot_predicted_and_measured_statistics( fig, ax = plt.subplots(2, 1, sharex=True, figsize=(6, 6)) fig.subplots_adjust(hspace=0) # ax[0].plot(x, y_theory, label="Total") + assert isinstance(ax, np.ndarray) + assert isinstance(ax[0], plt.Axes) + assert isinstance(ax[1], plt.Axes) ax[0].plot(ells, cells.GG, label="GG firecrown") ax[0].plot(ells, cl_GG, ls="--", label="GG CCL") ax[0].plot(ells, -cells.GI, label="-GI firecrown")