crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 05:35:49 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

material and plot_xs tests for plotter.py

Browse source
This commit is contained in:
shimwell 2023-03-02 23:52:00 +00:00
parent 33f9ab7083
commit 7f16f46bed
1 changed files with 38 additions and 8 deletions
Download patch file Download diff file Expand all files Collapse all files

46
tests/unit_tests/test_plotter.py
View file

@ -1,22 +1,27 @@
import numpy as np
import openmc
import pytest
from matplotlib.figure import Figure
def test_calculate_cexs_elem_mat_sab():
"""Checks that sab cross sections are included in the
_calculate_cexs_elem_mat method and have the correct shape"""
@pytest.fixture(scope="module")
def test_mat():
mat_1 = openmc.Material()
mat_1.add_element("H", 4.0, "ao")
mat_1.add_element("O", 4.0, "ao")
mat_1.add_element("C", 4.0, "ao")
return mat_1
mat_1.add_s_alpha_beta("c_C6H6")
mat_1.set_density("g/cm3", 0.865)
def test_calculate_cexs_elem_mat_sab(test_mat):
"""Checks that sab cross sections are included in the
_calculate_cexs_elem_mat method and have the correct shape"""
test_mat.add_s_alpha_beta("c_C6H6")
test_mat.set_density("g/cm3", 0.865)
energy_grid, data = openmc.plotter._calculate_cexs_elem_mat(
mat_1,
test_mat,
["inelastic"],
sab_name="c_C6H6",
)
@ -28,7 +33,7 @@ def test_calculate_cexs_elem_mat_sab():
assert len(data[0]) == len(energy_grid)
@pytest.mark.parametrize("this, data_type", [("Li", "element"), ("Li6", "nuclide")])
@pytest.mark.parametrize("this,data_type", [("Li", "element"), ("Li6", "nuclide")])
def test_calculate_cexs_with_element(this, data_type):
# single type (reaction)
@ -55,3 +60,28 @@ def test_calculate_cexs_with_element(this, data_type):
assert len(data[0]) == len(energy_grid)
# reactions are both the same MT number 2 is elastic
assert np.array_equal(data[0], data[1])
def test_calculate_cexs_with_materials(test_mat):
energy_grid, data = openmc.plotter.calculate_cexs(
this=test_mat, types=[205], data_type="material"
)
assert isinstance(energy_grid, np.ndarray)
assert isinstance(data, np.ndarray)
assert len(energy_grid) > 1
assert len(data) == 1
assert len(data[0]) == len(energy_grid)
@pytest.mark.parametrize(("this,data_type"), [("Be", "element"), ("Be9", "nuclide")])
def test_plot_xs(this, data_type):
assert isinstance(
openmc.plotter.plot_xs(this, data_type=data_type, types=["total"]), Figure
)
def test_plot_xs_mat(test_mat):
assert isinstance(
openmc.plotter.plot_xs(test_mat, data_type="material", types=["total"]), Figure
)