From 503908835baa8a06cc8fa8be7efc7076a673b677 Mon Sep 17 00:00:00 2001 From: Gonzalo Martinez Lema Date: Fri, 2 Feb 2024 11:38:18 +0100 Subject: [PATCH] Fix beersheba test --- invisible_cities/cities/beersheba_test.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/invisible_cities/cities/beersheba_test.py b/invisible_cities/cities/beersheba_test.py index 62deeebef..0707554ff 100644 --- a/invisible_cities/cities/beersheba_test.py +++ b/invisible_cities/cities/beersheba_test.py @@ -27,7 +27,8 @@ def test_create_deconvolution_df(ICDATADIR): new_dst = pd.concat([create_deconvolution_df(t, t.E.values, (t.X.values, t.Y.values, t.Z.values), CutType.abs, ecut, 3) for _, t in true_dst.groupby('event')]) true_dst = true_dst.loc[true_dst.E > ecut, :].reset_index(drop=True) - + # compare only existing columns + true_dst = true_dst.loc[:, new_dst.columns.values.tolist()] assert_dataframes_close(new_dst .reset_index(drop=True), true_dst.reset_index(drop=True))