diff --git a/tests/unit_tests/test_plotter.py b/tests/unit_tests/test_plotter.py index a3f1ff6662..15526351a4 100644 --- a/tests/unit_tests/test_plotter.py +++ b/tests/unit_tests/test_plotter.py @@ -1,5 +1,6 @@ -import openmc import numpy as np +import openmc +import pytest def test_calculate_cexs_elem_mat_sab(): @@ -25,3 +26,18 @@ def test_calculate_cexs_elem_mat_sab(): assert len(energy_grid) > 1 assert len(data) == 1 assert len(data[0]) == len(energy_grid) + + +@pytest.mark.parametrize("this, data_type", [("Li", "element"), ("Li6", "nuclide")]) +def test_calculate_cexs_with_element(this, data_type): + + # single type (reaction) + energy_grid, data = openmc.plotter.calculate_cexs( + this=this, data_type=data_type, types=[205] + ) + + 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)