Skip to content
Snippets Groups Projects

Draft: Resolve "Muon Propagation"

Closed Johannes Schumann requested to merge 8-muon-propagation into master
@@ -17,16 +17,18 @@ import uproot
import pytest
from os.path import abspath, join, dirname
from thepipe.logger import get_logger
from scipy.spatial.transform import Rotation
import proposal as pp
import awkward as ak
from km3buu.propagation import propagate_lepton
from km3buu.geometry import CanVolume
pp.RandomGenerator.get().set_seed(1234)
@pytest.mark.skip(reason="CI boost lib problem")
# @pytest.mark.skip(reason="CI boost lib problem")
class TestTauPropagation(unittest.TestCase):
def setUp(self):
data = ak.Array({
@@ -47,7 +49,10 @@ class TestTauPropagation(unittest.TestCase):
4.743161507744377, 23.096499191566885
]
})
self.sec = propagate_lepton(data, 15)
volume = CanVolume()
vtx_pos = np.zeros((5,3))
vtx_rot = [Rotation.from_euler("yz", [0,0]) for i in range(5)]
self.sec = propagate_lepton(data, 15, volume, vtx_pos, vtx_rot)
def test_secondary_momenta(self):
np.testing.assert_array_almost_equal(np.array(self.sec[0].E),
Loading