Update test_energy_dispersion_to_migration

cta-observatory · Dec 13, 2023 · f6434ed · f6434ed
1 parent 0591784
commit f6434ed
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyirf/irf/tests/test_energy_dispersion.py b/pyirf/irf/tests/test_energy_dispersion.py
@@ -143,7 +143,7 @@ def test_energy_dispersion_to_migration():
     assert migration_matrix.shape[2] == dispersion_matrix.shape[2]
 
     # test that all migrations are included for central energies
-    np.allclose(migration_matrix.sum(axis=1).max(), 1, rtol=0.1)
+    assert np.allclose(migration_matrix.sum(axis=2).sum(axis=1), 1, rtol=0.01)
 
     # test that migrations dont always sum to 1 (since some energies are
     # not included in the matrix)