From cbcf9d645e57f6401fc91c4ad20952ee2d89aacb Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Tue, 14 Dec 2021 13:31:24 +0100
Subject: [PATCH] Test with rtol of 5%
---
km3buu/tests/test_physics.py | 30 ++++++++----------------------
1 file changed, 8 insertions(+), 22 deletions(-)
diff --git a/km3buu/tests/test_physics.py b/km3buu/tests/test_physics.py
index 6d19ed8..7d2bd24 100644
--- a/km3buu/tests/test_physics.py
+++ b/km3buu/tests/test_physics.py
@@ -28,53 +28,39 @@ class TestVisibleEnergyWeights(unittest.TestCase):
def test_ngamma_elec(self):
vfunc = np.vectorize(number_photons_per_electron)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[1, :], val)
+ assert np.allclose(self.ref_values[1, :], val, rtol=0.05)
def test_pion_weight(self):
vfunc = np.vectorize(pion_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[2, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[2, :], val, rtol=0.05)
def test_kaon_weight(self):
vfunc = np.vectorize(kaon_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[3, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[3, :], val, rtol=0.05)
def test_kshort_weight(self):
vfunc = np.vectorize(kshort_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[4, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[4, :], val, rtol=0.05)
def test_klong_weight(self):
vfunc = np.vectorize(klong_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[5, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[5, :], val, rtol=0.05)
def test_proton_weight(self):
vfunc = np.vectorize(proton_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[6, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[6, :], val, rtol=0.05)
def test_neutron_weight(self):
vfunc = np.vectorize(neutron_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[7, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[7, :], val, rtol=0.05)
def test_high_ene_weights(self):
vfunc = np.vectorize(high_energy_weight)
val = vfunc(self.ref_values[0, :])
- np.testing.assert_array_almost_equal(self.ref_values[8, :],
- val,
- decimal=3)
+ assert np.allclose(self.ref_values[8, :], val, rtol=0.05)
--
GitLab