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))