From fd12b2797e99504411aa8502a8cc51204b104f17 Mon Sep 17 00:00:00 2001
From: Johannes Schumann <johannes.schumann@fau.de>
Date: Fri, 1 Oct 2021 02:05:57 +0200
Subject: [PATCH] Add tests for seed option in jobcard
---
km3buu/tests/test_jobcard.py | 24 +++++++++++++++++++++++-
1 file changed, 23 insertions(+), 1 deletion(-)
diff --git a/km3buu/tests/test_jobcard.py b/km3buu/tests/test_jobcard.py
index 66b1162..c08fe45 100644
--- a/km3buu/tests/test_jobcard.py
+++ b/km3buu/tests/test_jobcard.py
@@ -12,8 +12,11 @@ __status__ = "Development"
import unittest
import numpy as np
+from itertools import combinations
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):
@@ -127,3 +130,22 @@ class TestNeutrinoSingleEnergyJobcard(unittest.TestCase):
def test_photon_propagation_flag(self):
self.assertEqual(self.test_jobcard["insertion"]["propagateNoPhoton"],
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))
--
GitLab