assert conversion to correctionlib

ikrommyd · Jun 11, 2024 · 3afd586 · 3afd586
1 parent 339b061
commit 3afd586
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/src/egamma_tnp/utils/pileup.py b/src/egamma_tnp/utils/pileup.py
@@ -51,6 +51,7 @@ def create_correction(pu_data_histogram, pu_mc_array, outfile=None, normalize_pu
 
     ceval = cset.to_evaluator()
     corr = ceval["Pileup"]
+    np.testing.assert_allclose(corr.evaluate(np.arange(100, dtype=float)), (pu_data / pu_mc), err_msg="Pileup correction does not match input data")
     return correctionlib_wrapper(corr)