Fix beersheba test

next-exp · Feb 2, 2024 · 5039088 · 5039088
1 parent 3f2f9b3
commit 5039088
Showing 1 changed file with 2 additions and 1 deletion.
diff --git 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))