crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-27 13:45:36 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions

add unit test for azimuthal zernike

Browse source
This commit is contained in:
rockfool 2020-02-24 15:10:41 -05:00
parent d210239752
commit 8d83b391b6
1 changed files with 52 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

52
tests/unit_tests/test_polynomials.py
View file

@ -42,3 +42,55 @@ def test_zernike_radial():
test_vals = zn_rad([i * zn_rad.radius for i in rho])
assert np.allclose(ref_vals, test_vals)
def test_zernike():
import openmc.lib as lib
coeff = np.asarray([1.1e-1, -3.2e2, 5.3, 7.4, -9.5, 0.005])
zn_azimuthal = openmc.Zernike(coeff)
assert zn_azimuthal.order == 2
assert zn_azimuthal.radius == 1
coeff = np.asarray([1.5, -3.6, 9.7e-1, -6.8e-1, 0.11, 0.33e2, 0.002, 13.75, \
3.1, -7.3, 7.8e-1, -1.1e-1, 2.56, 5.25e3, 0.123])
zn_azimuthal = openmc.Zernike(coeff, 0.392)
assert zn_azimuthal.order == 4
assert zn_azimuthal.radius == 0.392
norm_vec = np.array([1, 4, 4, 6, 3, 6, 8, 8, 8, 8,
10, 10, 5, 10, 10]) / (np.pi * 0.392 ** 2)
norm_coeff = norm_vec * coeff
rho = 0.5
theta = np.radians(45)
# Reference solution from running the C API for calc_zn
raw_zn = lib.calc_zn(zn_azimuthal.order, rho, theta)
ref_vals = np.sum(norm_coeff * raw_zn)
test_vals = zn_azimuthal(rho * zn_azimuthal.radius, theta)
assert ref_vals == test_vals
rho = [0.2, 0.5]
theta = np.radians(30)
#Reference solution from running the C API for calc_zn
raw_zn1 = lib.calc_zn(zn_azimuthal.order, rho[0], theta)
raw_zn2 = lib.calc_zn(zn_azimuthal.order, rho[1], theta)
ref_vals = [np.sum(norm_coeff * raw_zn1), np.sum(norm_coeff * raw_zn2)]
test_vals = zn_azimuthal([i * zn_azimuthal.radius for i in rho], theta)
assert np.allclose(ref_vals, test_vals)
rho = 0.2
theta = np.radians([30, 60])
#Reference solution from running the C API for calc_zn
raw_zn1 = lib.calc_zn(zn_azimuthal.order, rho, theta[0])
raw_zn2 = lib.calc_zn(zn_azimuthal.order, rho, theta[1])