From cc338c98d5a9d7ce707eb57aaacb6a681dfdd9b0 Mon Sep 17 00:00:00 2001 From: Paul Romano Date: Thu, 26 May 2022 09:42:03 -0500 Subject: [PATCH] Add mask_components in Plots -> XML round trip unit test --- tests/unit_tests/test_plots.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tests/unit_tests/test_plots.py b/tests/unit_tests/test_plots.py index f61b52604a..2db396cd5b 100644 --- a/tests/unit_tests/test_plots.py +++ b/tests/unit_tests/test_plots.py @@ -93,6 +93,7 @@ def test_plots(run_in_tmpdir): p1 = openmc.Plot(name='plot1') p1.origin = (5., 5., 5.) p1.colors = {10: (255, 100, 0)} + p1.mask_components = [2, 4, 6] p2 = openmc.Plot(name='plot2') p2.origin = (-3., -3., -3.) plots = openmc.Plots([p1, p2]) @@ -109,4 +110,5 @@ def test_plots(run_in_tmpdir): assert len(plots) assert plots[0].origin == p1.origin assert plots[0].colors == p1.colors + assert plots[0].mask_components == p1.mask_components assert plots[1].origin == p2.origin