From 5c1fbbc71e9fe93867f25f742006aa625d03a60e Mon Sep 17 00:00:00 2001 From: Sandro Dias Pinto Vitenti Date: Thu, 11 Jul 2024 13:38:39 -0300 Subject: [PATCH] Including typing checking in example. --- examples/des_y1_3x2pt/des_y1_3x2pt_PT.py | 3 +++ 1 file changed, 3 insertions(+) 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")