From 8791c7212e0bbca9abebf3f786b06ba8e82e151d Mon Sep 17 00:00:00 2001 From: Patrick Shriwise Date: Thu, 11 Aug 2022 07:48:58 -0500 Subject: [PATCH] Adding check for first vertex and centroid --- .../unstructured_mesh/test.py | 21 +++++++++++++++++++ 1 file changed, 21 insertions(+) diff --git a/tests/regression_tests/unstructured_mesh/test.py b/tests/regression_tests/unstructured_mesh/test.py index 090a47599..6146c1b1e 100644 --- a/tests/regression_tests/unstructured_mesh/test.py +++ b/tests/regression_tests/unstructured_mesh/test.py @@ -37,6 +37,27 @@ class UnstructuredMeshTest(PyAPITestHarness): def _compare_results(self): with openmc.StatePoint(self._sp_name) as sp: + # check some properties of the unstructured mesh + umesh = None + for m in sp.meshes.values(): + if isinstance(m, openmc.UnstructuredMesh): + umesh = m + assert umesh is not None + + # check that the first element centroid is correct + # this will depend on whether the tet mesh or hex mesh + # file is being used in this test + if umesh.element_types[0] == umesh._LINEAR_TET: + exp_vertex = (-10.0, -10.0, -10.0) + exp_centroid = (-8.75, -9.75, -9.25) + else: + exp_vertex = (-10.0, -10.0, 10.0) + exp_centroid = (-9.0, -9.0, 9.0) + + np.testing.assert_array_equal(umesh.vertices[0], exp_vertex) + np.testing.assert_array_equal(umesh.centroid(0), exp_centroid) + + # loop over the tallies and get data for tally in sp.tallies.values(): # find the regular and unstructured meshes