From 883db6a1ab090be4283e63d2533ba493efaeb8d2 Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Wed, 5 May 2021 16:46:14 +0200
Subject: [PATCH] Add missing outgoing lepton from neutrino vertex to tests
---
km3buu/tests/test_output.py | 31 +++++++++++++++++--------------
1 file changed, 17 insertions(+), 14 deletions(-)
diff --git a/km3buu/tests/test_output.py b/km3buu/tests/test_output.py
index 3e41fb1..edf96cc 100644
--- a/km3buu/tests/test_output.py
+++ b/km3buu/tests/test_output.py
@@ -97,20 +97,23 @@ class TestAANET(unittest.TestCase):
def test_firstevent(self):
evt = self.fobj.events[0]
np.testing.assert_array_equal(evt.mc_tracks.pdgid,
- [12, 2212, 111, 211, -211])
- np.testing.assert_array_equal(evt.mc_tracks.status, [0, 1, 1, 1, 1])
- np.testing.assert_array_almost_equal(
- evt.mc_tracks.E,
- [11.90433897, 1.45689677, 0.49284856, 8.33975778, 0.28362369])
- np.testing.assert_array_almost_equal(
- evt.mc_tracks.dir_x,
- [0.18255849, 0.48623089, 0.23767571, 0.24971059, 0.11284916])
- np.testing.assert_array_almost_equal(
- evt.mc_tracks.dir_y,
- [-0.80816248, -0.49241334, -0.84679953, -0.83055629, -0.82624071])
- np.testing.assert_array_almost_equal(
- evt.mc_tracks.dir_z,
- [0.55995162, 0.72187854, 0.47585798, 0.4978161, -0.55189796])
+ [12, 11, 2212, 111, 211, -211])
+ np.testing.assert_array_equal(evt.mc_tracks.status, [0, 1, 1, 1, 1, 1])
+ np.testing.assert_array_almost_equal(evt.mc_tracks.E, [
+ 11.90433897, 2.1818, 1.45689677, 0.49284856, 8.33975778, 0.28362369
+ ])
+ np.testing.assert_array_almost_equal(evt.mc_tracks.dir_x, [
+ 0.18255849, -0.309836, 0.48623089, 0.23767571, 0.24971059,
+ 0.11284916
+ ])
+ np.testing.assert_array_almost_equal(evt.mc_tracks.dir_y, [
+ -0.80816248, -0.377270, -0.49241334, -0.84679953, -0.83055629,
+ -0.82624071
+ ])
+ np.testing.assert_array_almost_equal(evt.mc_tracks.dir_z, [
+ 0.55995162, 0.872737, 0.72187854, 0.47585798, 0.4978161,
+ -0.55189796
+ ])
# Test dataset is elec CC -> outgoing particles are placed at vertex pos
np.testing.assert_allclose(evt.mc_tracks.t, 8603022.62272017)
np.testing.assert_allclose(evt.mc_tracks.pos_x, -127.07940486)
--
GitLab