crp/OpenMC
1
0
Fork 0
mirror of https://github.com/openmc-dev/openmc.git synced 2026-07-28 14:15:42 -04:00
Code Issues Projects Releases Packages Wiki Activity Actions 5

bug fix in unit test

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

29
tests/unit_tests/test_polynomials.py
View file

@ -94,3 +94,32 @@ def test_zernike():
raw_zn1 = lib.calc_zn(zn_azimuthal.order, rho, theta[0])
raw_zn2 = lib.calc_zn(zn_azimuthal.order, rho, theta[1])
ref_vals = [np.sum(norm_coeff * raw_zn1), np.sum(norm_coeff * raw_zn2)]
test_vals = zn_azimuthal(rho * zn_azimuthal.radius, [j for j in theta])
assert np.allclose(ref_vals, test_vals)
rho = [0.2, 0.5]
theta = np.radians([30, 60])
#Reference solution from running the C API for calc_zn
raw_zn1 = lib.calc_zn(zn_azimuthal.order, rho[0], theta[0])
raw_zn2 = lib.calc_zn(zn_azimuthal.order, rho[1], theta[0])
raw_zn3 = lib.calc_zn(zn_azimuthal.order, rho[0], theta[1])
raw_zn4 = lib.calc_zn(zn_azimuthal.order, rho[1], theta[1])
ref_vals = [np.sum(norm_coeff * raw_zn1), np.sum(norm_coeff * raw_zn2),
np.sum(norm_coeff * raw_zn3), np.sum(norm_coeff * raw_zn4)]
test_vals = zn_azimuthal([i * zn_azimuthal.radius for i in rho], [j for j in theta])
test_vals = np.ravel(test_vals)
assert np.allclose(ref_vals, test_vals)