Including typing checking in example.

LSSTDESC · Jul 11, 2024 · 5c1fbbc · 5c1fbbc
1 parent 0b115ea
commit 5c1fbbc
Showing 1 changed file with 3 additions and 0 deletions.
diff --git 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")