From d3b018e8564de791acea85bccba09a1d8e30399e Mon Sep 17 00:00:00 2001
From: Kayla Clements <clemekay@oregonstate.edu>
Date: Sat, 27 Apr 2024 15:32:34 -0700
Subject: [PATCH] Working on uq tests

---
 mcdc/main.py                                  |  6 +++---
 test/verification/analytic/run.py             | 19 +++++++++++--------
 .../analytic/variance_deconv/reference.py     |  4 ++++
 3 files changed, 18 insertions(+), 11 deletions(-)
 create mode 100644 test/verification/analytic/variance_deconv/reference.py

diff --git a/mcdc/main.py b/mcdc/main.py
index 102f3793..4d3155bc 100644
--- a/mcdc/main.py
+++ b/mcdc/main.py
@@ -859,6 +859,8 @@ def dictlist_to_h5group(dictlist, input_group, name):
 
 def dict_to_h5group(dict_, group):
     for k, v in dict_.items():
+        if k == 'uq':
+            x = 1
         if type(v) == dict:
             dict_to_h5group(dict_[k], group.create_group(k))
         elif v is None:
@@ -886,9 +888,7 @@ def generate_hdf5(mcdc):
                 dictlist_to_h5group(input_deck.sources, input_group, "source")
                 dict_to_h5group(input_deck.tally, input_group.create_group("tally"))
                 dict_to_h5group(input_deck.setting, input_group.create_group("setting"))
-                dict_to_h5group(
-                    input_deck.technique, input_group.create_group("technique")
-                )
+                dict_to_h5group(input_deck.technique, input_group.create_group("technique"))
 
             # Tally
             T = mcdc["tally"]
diff --git a/test/verification/analytic/run.py b/test/verification/analytic/run.py
index ccc96f23..e8925de1 100644
--- a/test/verification/analytic/run.py
+++ b/test/verification/analytic/run.py
@@ -46,11 +46,14 @@ def run(N_hist, name, N_batch=1):
 
 for name in task.keys():
     os.chdir(name)
-    N_min = task[name]["N_lim"][0]
-    N_max = task[name]["N_lim"][1]
-    N = task[name]["N"]
-    for N_hist in np.logspace(N_min, N_max, N):
-        N_hist = int(N_hist)
-        print(name, N_hist)
-        run(N_hist, name)
-    os.chdir(r"..")
+    if "N_batch" in task[name]:
+            x = 1
+    else:
+        N_min = task[name]["N_lim"][0]
+        N_max = task[name]["N_lim"][1]
+        N = task[name]["N"]
+        for N_hist in np.logspace(N_min, N_max, N):
+            N_hist = int(N_hist)
+            print(name, N_hist)
+            run(N_hist, name)
+        os.chdir(r"..")
diff --git a/test/verification/analytic/variance_deconv/reference.py b/test/verification/analytic/variance_deconv/reference.py
new file mode 100644
index 00000000..ebe31cfd
--- /dev/null
+++ b/test/verification/analytic/variance_deconv/reference.py
@@ -0,0 +1,4 @@
+import numpy as np
+
+# Parameters
+