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

Add test for evaluate zn_rad

Browse source
This commit is contained in:
Zhuoran Han 2018-08-13 10:35:16 -05:00
parent 7e54eb0e97
commit 1b127cd2e8
2 changed files with 17 additions and 2 deletions
Download patch file Download diff file Expand all files Collapse all files

2
src/math_functions.cpp
View file

@ -653,7 +653,7 @@ double evaluate_zernike_rad_c(int n, const double data[], double r) {
calc_zn_rad_c(n, r, rnr);
int index;
for (int i = 0; i <= n; i+ = 2) {
for (int i = 0; i <= n; i+= 2) {
index = int(i/2);
val += data[index] * rnr[index];
}

17
tests/unit_tests/test_math.py
View file

@ -161,7 +161,6 @@ def test_calc_zn():
def test_calc_zn_rad():
n = 10
rho = 0.5
phi = 0.5
# Reference solution from running the Fortran implementation
ref_vals = np.array([
@ -172,6 +171,22 @@ def test_calc_zn_rad():
assert np.allclose(ref_vals, test_vals)
def test_evaluate_legendre():
max_order = 10
rho = 0.5
raw_zn = np.array([
1.00000000e+00, -5.00000000e-01, -1.25000000e-01,
4.37500000e-01, -2.89062500e-01,-8.98437500e-02])
raw_coeff = np.ones(6)
norm_vec = np.divide(np.arange(0, max_order + 1, 2) + 1, np.pi * rho * rho)
real_coeff = np.multiply(raw_coeff,norm_vec)
ref_vals = np.sum(np.multiply(real_coeff,raw_zn))
test_vals = openmc.capi.math.evaluate_zernike_rad(real_coeff, rho)
assert np.allclose(ref_vals, test_vals)
def test_calc_norm_zn_rad():
max_order = 10