From 6cd17cb087c28b5b732f0104131db4079871ba14 Mon Sep 17 00:00:00 2001 From: Madicken Munk Date: Tue, 16 Dec 2025 16:20:07 -0800 Subject: [PATCH 1/2] add maximum quadrature function and test --- rad/quadrature.py | 13 +++++++++++++ test/test_quadrature.py | 12 ++++++++++++ 2 files changed, 25 insertions(+) diff --git a/rad/quadrature.py b/rad/quadrature.py index bdd2318..7957166 100644 --- a/rad/quadrature.py +++ b/rad/quadrature.py @@ -33,3 +33,16 @@ def normalization(self, phi): phi /= np.sum(phi) return phi + + + def calculate_maximum_flux(self, psi): + """ + Calculalaing the maximum flux value + + :param phi: Vector of flux values + :return: maximum flux + """ + + max_psi = np.max(psi) + + return max_psi diff --git a/test/test_quadrature.py b/test/test_quadrature.py index ef7af97..a7c7b4a 100644 --- a/test/test_quadrature.py +++ b/test/test_quadrature.py @@ -29,6 +29,18 @@ def test_normalization(): # using all close cus these are floats! assert np.allclose(np.sum(x), 1.0) +def test_maximum(): + N = 10 + + x = np.random.random(10) + quad = rad.quadrature(4) + psi_m = quad.maximum(x) + + max_val = np.sort(x)[-1] + + # using all close cus these are floats! + assert np.allclose(psi_m, max_val) + if __name__ == "__main__": test_normalization() From c7a5d78e1112c76a2a30800c1b80e465c41ea67f Mon Sep 17 00:00:00 2001 From: Madicken Munk Date: Tue, 16 Dec 2025 16:41:28 -0800 Subject: [PATCH 2/2] update test function reference to correct max flux name --- test/test_quadrature.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/test/test_quadrature.py b/test/test_quadrature.py index a7c7b4a..b4965ea 100644 --- a/test/test_quadrature.py +++ b/test/test_quadrature.py @@ -34,7 +34,7 @@ def test_maximum(): x = np.random.random(10) quad = rad.quadrature(4) - psi_m = quad.maximum(x) + psi_m = quad.calculate_maximum_flux(x) max_val = np.sort(x)[-1]