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..b4965ea 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.calculate_maximum_flux(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()