From 5787616ae41593bac4d100da162da5a8f98a1cea Mon Sep 17 00:00:00 2001
From: RemDelaporteMathurin <rdelaportemathurin@gmail.com>
Date: Fri, 7 Oct 2022 07:18:00 +0000
Subject: [PATCH] added two tests to check centre was written/read to/from xml

---
 tests/unit_tests/test_cylindrical_mesh.py | 30 +++++++++++++++++++++++
 tests/unit_tests/test_spherical_mesh.py   | 30 +++++++++++++++++++++++
 2 files changed, 60 insertions(+)
 create mode 100644 tests/unit_tests/test_cylindrical_mesh.py
 create mode 100644 tests/unit_tests/test_spherical_mesh.py

diff --git a/tests/unit_tests/test_cylindrical_mesh.py b/tests/unit_tests/test_cylindrical_mesh.py
new file mode 100644
index 0000000000..89322d7045
--- /dev/null
+++ b/tests/unit_tests/test_cylindrical_mesh.py
@@ -0,0 +1,30 @@
+import openmc
+import numpy as np
+
+def test_centre_read_write_to_xml():
+    """Tests that the centre attribute can be written and read back to XML
+    """
+    # build
+    mesh = openmc.CylindricalMesh()
+    mesh.phi_grid = [1, 2, 3]
+    mesh.z_grid = [1, 2, 3]
+    mesh.r_grid = [1, 2, 3]
+
+    mesh.centre = [0.1, 0.2, 0.3]
+
+    tally = openmc.Tally()
+
+    mesh_filter = openmc.MeshFilter(mesh)
+    tally.filters.append(mesh_filter)
+
+    tally.scores.append("heating")
+
+    tallies = openmc.Tallies([tally])
+
+    tallies.export_to_xml()
+
+    # read back
+    new_tallies = openmc.Tallies.from_xml()
+    new_tally = new_tallies[0]
+    new_mesh = new_tally.filters[0].mesh
+    assert np.allclose(new_mesh.centre, mesh.centre)
\ No newline at end of file
diff --git a/tests/unit_tests/test_spherical_mesh.py b/tests/unit_tests/test_spherical_mesh.py
new file mode 100644
index 0000000000..cdeb188ece
--- /dev/null
+++ b/tests/unit_tests/test_spherical_mesh.py
@@ -0,0 +1,30 @@
+import openmc
+import numpy as np
+
+def test_centre_read_write_to_xml():
+    """Tests that the centre attribute can be written and read back to XML
+    """
+    # build
+    mesh = openmc.SphericalMesh()
+    mesh.phi_grid = [1, 2, 3]
+    mesh.theta_grid = [1, 2, 3]
+    mesh.r_grid = [1, 2, 3]
+
+    mesh.centre = [0.1, 0.2, 0.3]
+
+    tally = openmc.Tally()
+
+    mesh_filter = openmc.MeshFilter(mesh)
+    tally.filters.append(mesh_filter)
+
+    tally.scores.append("heating")
+
+    tallies = openmc.Tallies([tally])
+
+    tallies.export_to_xml()
+
+    # read back
+    new_tallies = openmc.Tallies.from_xml()
+    new_tally = new_tallies[0]
+    new_mesh = new_tally.filters[0].mesh
+    assert np.allclose(new_mesh.centre, mesh.centre)
\ No newline at end of file