Skip to content
Snippets Groups Projects
Commit fd12b279 authored by Johannes Schumann's avatar Johannes Schumann
Browse files

Add tests for seed option in jobcard

parent 2a4a7fa0
No related branches found
No related tags found
1 merge request!13Random Generator Seed
Pipeline #22415 canceled
...@@ -12,8 +12,11 @@ __status__ = "Development" ...@@ -12,8 +12,11 @@ __status__ = "Development"
import unittest import unittest
import numpy as np import numpy as np
from itertools import combinations
from km3buu.jobcard import * from km3buu.jobcard import *
from tempfile import TemporaryFile from km3buu.ctrl import run_jobcard
from km3buu.output import GiBUUOutput
from tempfile import TemporaryFile, TemporaryDirectory
class TestJobcard(unittest.TestCase): class TestJobcard(unittest.TestCase):
...@@ -127,3 +130,22 @@ class TestNeutrinoSingleEnergyJobcard(unittest.TestCase): ...@@ -127,3 +130,22 @@ class TestNeutrinoSingleEnergyJobcard(unittest.TestCase):
def test_photon_propagation_flag(self): def test_photon_propagation_flag(self):
self.assertEqual(self.test_jobcard["insertion"]["propagateNoPhoton"], self.assertEqual(self.test_jobcard["insertion"]["propagateNoPhoton"],
not self.photon_propagation_flag) not self.photon_propagation_flag)
class TestJobcardSeed(unittest.TestCase):
def setUp(self):
jc = generate_neutrino_jobcard(100,
"CC",
"electron", (1.0, 2.0), (1, 1),
do_decay=False,
photon_propagation=False,
seed=1234)
self.dfs = []
for i in range(2):
output_dir = TemporaryDirectory()
run_jobcard(jc, output_dir.name, container=True)
self.dfs.append(GiBUUOutput(output_dir.name).df)
def test_output(self):
for a, b in combinations(self.dfs,2):
assert all((a == b).all(1))
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment