From 01cdb9b05f97146115fdab9bd33578e501423410 Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Sun, 2 Apr 2023 21:16:22 +0200
Subject: [PATCH] Add test for muon propagation from outside the can
---
km3buu/tests/test_propagation.py | 42 +++++++++++++++++++++++++++++---
1 file changed, 38 insertions(+), 4 deletions(-)
diff --git a/km3buu/tests/test_propagation.py b/km3buu/tests/test_propagation.py
index 17f82ff..403b79e 100644
--- a/km3buu/tests/test_propagation.py
+++ b/km3buu/tests/test_propagation.py
@@ -47,10 +47,10 @@ class TestTauPropagation(unittest.TestCase):
[3.271, 3.696, -0.009],
decimal=3)
np.testing.assert_array_almost_equal(np.array(self.sec.Py),
- [2.48 , 4.382, 0.097],
+ [2.48, 4.382, 0.097],
decimal=3)
np.testing.assert_array_almost_equal(np.array(self.sec.Pz),
- [-0.549, 0.564, -0.015],
+ [-0.549, 0.564, -0.015],
decimal=3)
def test_secondary_types(self):
@@ -59,11 +59,45 @@ class TestTauPropagation(unittest.TestCase):
def test_secondary_positions(self):
np.testing.assert_array_almost_equal(np.array(self.sec.x),
- [10.0,10.0,10.0],
+ [10.0, 10.0, 10.0],
decimal=1)
np.testing.assert_array_almost_equal(np.array(self.sec.y),
- [10.0,10.0,10.0],
+ [10.0, 10.0, 10.0],
decimal=1)
np.testing.assert_array_almost_equal(np.array(self.sec.z),
[100., 100., 100.],
decimal=1)
+
+
+class TestMuonPropagation(unittest.TestCase):
+
+ def setUp(self):
+ prop = Propagator([13, -13],
+ CylindricalVolume().make_proposal_geometries())
+ self.position = np.array([200.0, 200.0, 100.0])
+ self.direction = np.array([-1., -1., 0])
+ self.energy = 100.0
+ self.sec = prop.propagate(13, self.energy, self.position,
+ self.direction)
+
+ def test_secondary_momenta(self):
+ np.testing.assert_array_almost_equal(np.array(self.sec.E), [77.102],
+ decimal=3)
+ np.testing.assert_array_almost_equal(np.array(self.sec.Px), [-54.519],
+ decimal=3)
+ np.testing.assert_array_almost_equal(np.array(self.sec.Py), [-54.519],
+ decimal=3)
+ np.testing.assert_array_almost_equal(np.array(self.sec.Pz),
+ [-0.],
+ decimal=3)
+
+ def test_secondary_types(self):
+ np.testing.assert_array_equal(np.array(self.sec.barcode), [13])
+
+ def test_secondary_positions(self):
+ np.testing.assert_array_almost_equal(np.array(self.sec.x), [141.7],
+ decimal=1)
+ np.testing.assert_array_almost_equal(np.array(self.sec.y), [141.7],
+ decimal=1)
+ np.testing.assert_array_almost_equal(np.array(self.sec.z), [100.],
+ decimal=1)
--
GitLab