added results file to tests

fusion-energy · Jun 7, 2023 · a38b078 · a38b078
1 parent 78055e2
commit a38b078
Show file tree

Hide file tree

Showing 4 changed files with 724 additions and 15 deletions.
diff --git a/src/openmc_depletion_plotter/app.py b/src/openmc_depletion_plotter/app.py
@@ -89,7 +89,7 @@ def main():
 
         activity_or_atoms = st.sidebar.selectbox(
             label="Plot",
-            options=("activity", "number of atoms"),
+            options=("activity", "number of atoms", "decay heat"),
             index=0,
             key="activity_or_atoms",
             help="",
@@ -201,6 +201,21 @@ def main():
                 include_total=include_total,
                 path=materials_file.name,
             )
+        elif activity_or_atoms == "decay heat":
+            plot = plot_decay_heat_vs_time(
+                excluded_material=material_to_exclude,
+                time_units=time_units,
+                show_top=show_top,
+                x_scale=x_scale,
+                y_scale=y_scale,
+                include_total=include_total,
+                # x_axis_title=None,
+                plotting_backend=backend,
+                # units="W/g", # TODO add drop down option
+                # threshold=None,
+                title="Decay heat of nuclides in material",
+                material_index=material_index,
+            )
         elif activity_or_atoms == "number of atoms":
             plot = results.plot_atoms_vs_time(
                 # todo allow no materials to be excluded