diff --git a/km3buu/jobcard.py b/km3buu/jobcard.py index afda81c54a4d29c800d6501d20d8f7a6665855e2..18a44e47042be52e9d15e6fe5e08467991fa01b6 100644 --- a/km3buu/jobcard.py +++ b/km3buu/jobcard.py @@ -150,7 +150,7 @@ def generate_neutrino_jobcard(events, jc["nl_neutrino_energyflux"]["eflux_min"] = energy[0] jc["nl_neutrino_energyflux"]["eflux_max"] = energy[1] else: - jc["nl_sigmanc"]["enu"] = energy + jc["nl_sigmamc"]["enu"] = energy # DECAY if do_decay: for i in DECAYED_HADRONS: diff --git a/km3buu/tests/test_jobcard.py b/km3buu/tests/test_jobcard.py index ff5c04d69b6b7688ca64ff32657e1f2aa93faa4e..66b1162161db75a41876da0375604cdee621821f 100644 --- a/km3buu/tests/test_jobcard.py +++ b/km3buu/tests/test_jobcard.py @@ -45,7 +45,7 @@ class TestJobcard(unittest.TestCase): assert ctnt.find(expected_line) == -1 -class TestNeutrinoJobcard(unittest.TestCase): +class TestNeutrinoEnergyRangeJobcard(unittest.TestCase): def setUp(self): self.test_fluxfile = TemporaryFile() self.test_Z = np.random.randint(1, 100) @@ -93,3 +93,37 @@ class TestNeutrinoJobcard(unittest.TestCase): def test_photon_propagation_flag(self): self.assertEqual(self.test_jobcard["insertion"]["propagateNoPhoton"], not self.photon_propagation_flag) + + +class TestNeutrinoSingleEnergyJobcard(unittest.TestCase): + def setUp(self): + self.test_fluxfile = TemporaryFile() + self.test_Z = np.random.randint(1, 100) + self.test_A = np.random.randint(self.test_Z, 100) + self.test_energy = np.random.uniform(0.0, 100.0) + self.photon_propagation_flag = np.random.choice([True, False]) + self.do_decay = np.random.choice([True, False]) + self.test_jobcard = generate_neutrino_jobcard( + 1000, + "CC", + "electron", + self.test_energy, (self.test_Z, self.test_A), + do_decay=self.do_decay, + photon_propagation=self.photon_propagation_flag, + fluxfile=self.test_fluxfile.name, + input_path="/test") + + def test_input_path(self): + self.assertEqual("/test", self.test_jobcard["input"]["path_to_input"]) + + def test_target(self): + self.assertEqual(self.test_Z, self.test_jobcard["target"]["target_Z"]) + self.assertEqual(self.test_A, self.test_jobcard["target"]["target_A"]) + + def test_energy(self): + self.assertAlmostEqual(self.test_energy, + self.test_jobcard["nl_sigmamc"]["enu"]) + + def test_photon_propagation_flag(self): + self.assertEqual(self.test_jobcard["insertion"]["propagateNoPhoton"], + not self.photon_propagation_flag)