Making z-grid relative to the mesh origin in from_domain. (#2710)

This commit is contained in:
Patrick Shriwise 2023-09-27 13:46:53 -05:00 committed by GitHub
parent d4b989ab7c
commit f7bb27e0d9
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 7 additions and 2 deletions

View file

@ -1439,6 +1439,10 @@ class CylindricalMesh(StructuredMesh):
num=dimension[2]+1
)
origin = (cached_bb.center[0], cached_bb.center[1], z_grid[0])
# make z-grid relative to the origin
z_grid -= origin[2]
mesh = cls(
r_grid=r_grid,
z_grid=z_grid,

View file

@ -30,7 +30,7 @@ def test_cylindrical_mesh_from_cell():
assert np.array_equal(mesh.dimension, (2, 4, 3))
assert np.array_equal(mesh.r_grid, [0., 25., 50.])
assert np.array_equal(mesh.phi_grid, [0., 0.5*np.pi, np.pi, 1.5*np.pi, 2.*np.pi])
assert np.array_equal(mesh.z_grid, [10., 20., 30., 40.])
assert np.array_equal(mesh.z_grid, [0., 10., 20., 30.])
assert np.array_equal(mesh.origin, [0., 0., 10.])
# Cell is not centralized on Z or X axis
@ -83,7 +83,8 @@ def test_cylindrical_mesh_from_region():
assert np.array_equal(mesh.dimension, (6, 2, 3))
assert np.array_equal(mesh.r_grid, [0., 1., 2., 3., 4., 5., 6.])
assert np.array_equal(mesh.phi_grid, [0., 0.5*np.pi, np.pi])
assert np.array_equal(mesh.z_grid, [-30., -10., 10., 30.])
assert np.array_equal(mesh.z_grid, [0.0, 20., 40., 60])
assert np.array_equal(mesh.origin, (0.0, 0.0, -30.))
def test_reg_mesh_from_universe():