From a740ae6d30b17fc6bdc1cf49e05a97201153e351 Mon Sep 17 00:00:00 2001 From: rockfool Date: Thu, 30 Jan 2020 21:08:07 -0500 Subject: [PATCH] modify radius in unittest for zernikeradial --- tests/unit_tests/test_polynomials.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/unit_tests/test_polynomials.py b/tests/unit_tests/test_polynomials.py index f6ac2711fd..dfcdb1f58a 100644 --- a/tests/unit_tests/test_polynomials.py +++ b/tests/unit_tests/test_polynomials.py @@ -24,7 +24,7 @@ def test_zernike_radial(): ref_vals = np.sum(norm_coeff * raw_zn) - test_vals = zn_rad(rho) + test_vals = zn_rad(rho * zn_rad.radius) assert ref_vals == test_vals @@ -39,6 +39,6 @@ def test_zernike_radial(): ref_vals = [np.sum(norm_coeff * raw_zn1), np.sum(norm_coeff * raw_zn2)] - test_vals = zn_rad(rho) + test_vals = zn_rad([i * zn_rad.radius for i in rho]) assert np.allclose(ref_vals, test_vals)