def generate_integration_tests():
def print_(f, msg):
f.write("\t" + msg + "\n")
integrators_names = {"gauss_kronrod": "gauss_kronrod()",
"simplex": "simplex()", "gauss_legendre": "gauss_legendre()",
"gauss_lobatto": "gauss_lobatto()",
"simplex_steps_1000": "simplex(steps=1000)",
"gauss_legendre_steps_500": "gauss_legendre(steps=500)",
"gauss_lobatto_max_iter_127": "gauss_lobatto(max_number_of_iterations = 255, abs_tolerance = 1.0e-7)",
"gauss_kronrod_check_rel_tolerance": "gauss_kronrod(check_abs_tolerance = False, check_rel_tolerance = True)",
"gauss_kronrod_check_abs_tolerance_check_rel_tolerance": "gauss_kronrod(check_abs_tolerance = True, check_rel_tolerance = True)"
}
fcts_names = ("lambda x:2", "lambda x:x*x", "lambda x:exp(-x*x)",
"lambda x: x*x*x if x > 1 else x-2")
integrators = {'gauss_legendre': gauss_legendre(), 'gauss_lobatto': gauss_lobatto(),
'simplex_steps_1000': simplex(steps=1000), 'simplex': simplex(), 'gauss_kronrod': gauss_kronrod(),
'gauss_legendre_steps_500': gauss_legendre(steps=500),
'gauss_kronrod_check_abs_tolerance_check_rel_tolerance': gauss_kronrod(check_abs_tolerance=True,
check_rel_tolerance=True),
'gauss_lobatto_max_iter_127': gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7),
'gauss_kronrod_check_rel_tolerance': gauss_kronrod(check_abs_tolerance=False,
check_rel_tolerance=True)}
fcts = (lambda x: 2, lambda x: x * x, lambda x: exp(-x * x), lambda x: x * x * x if x > 1 else x - 2)
'''
s = []
for int_name, i in integrators_names.items():
s.append("'" + int_name + "':" + i)
print("integrators = {" + ", ".join(s) + "}")
print("fcts = (" + ", ".join(fcts_names) + ")")
'''
integration_boundaries = ((-1.0, 2.0), (0.0, 1.0), (-100.0, 99.0))
file_name = r"/Users/MarcusSteinkamp/Documents/integrator_test_code.txt"
fi = open(file_name, "w")
c = 0
i_fct = 0
for n, i in integrators_names.items():
i_fct = 0
for f in fcts_names:
for l, u in integration_boundaries:
print_(fi, "def test_integrator_" + str(c).zfill(3) + "_" + n + "(self):")
print_(fi, "\tfct = " + f)
print_(fi, "\tintegrator = " + i)
print_(fi, "\tresult = integrator(fct, " + str(l) + ", " + str(u) + ")")
print_(fi, "\tbenchmark = " + "{:18.9f}".format(integrators[n](fcts[i_fct], l, u)))
print_(fi, "\tself.assertAlmostEqual(benchmark, result, 8)")
print_(fi, "")
c = c + 1
i_fct = i_fct + 1
def test_integrator_compare(self):
i0 = simplex(steps=5000)
i1 = gauss_kronrod()
# i2 = gauss_legendre(steps = 1000) # gauss_legendre calculates bad.
i3 = gauss_lobatto()
fct = lambda x: cos(x)
l = 0
u = PI / 2.0
r0 = i0(fct, l, u)
r1 = i1(fct, l, u)
# r2 = i2(fct, l, u)
r3 = i3(fct, l, u)
benchmark = 1.0
self.assertAlmostEqual(r0, benchmark, 6)
self.assertAlmostEqual(r1, benchmark, 6)
self.assertAlmostEqual(r3, benchmark, 6)
def test_integrator_093_gauss_lobatto_max_iter_127(self):
fct = lambda x: x * x * x if x > 1 else x - 2
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, -1.0, 2.0)
benchmark = -0.250000000002
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_092_gauss_lobatto_max_iter_127(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, -100.0, 99.0)
benchmark = 1.77245385091
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_091_gauss_lobatto_max_iter_127(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, 0.0, 1.0)
benchmark = 0.746824132812
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_090_gauss_lobatto_max_iter_127(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, -1.0, 2.0)
benchmark = 1.62890552357
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_089_gauss_lobatto_max_iter_127(self):
fct = lambda x: x * x
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, -100.0, 99.0)
benchmark = 656766.3333333333
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_085_gauss_lobatto_max_iter_127(self):
fct = lambda x: 2
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, 0.0, 1.0)
benchmark = 2.0
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_015_gauss_lobatto(self):
fct = lambda x: x * x
integrator = gauss_lobatto()
result = integrator(fct, -1.0, 2.0)
benchmark = 3.0
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_022_gauss_lobatto(self):
fct = lambda x: x * x * x if x > 1 else x - 2
integrator = gauss_lobatto()
result = integrator(fct, 0.0, 1.0)
benchmark = -1.5
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_020_gauss_lobatto(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto()
result = integrator(fct, -100.0, 99.0)
benchmark = 1.77245385091
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_019_gauss_lobatto(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto()
result = integrator(fct, 0.0, 1.0)
benchmark = 0.746824132812
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_018_gauss_lobatto(self):
fct = lambda x: exp(-x * x)
integrator = gauss_lobatto()
result = integrator(fct, -1.0, 2.0)
benchmark = 1.62890552357
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_017_gauss_lobatto(self):
fct = lambda x: x * x
integrator = gauss_lobatto()
result = integrator(fct, -100.0, 99.0)
benchmark = 656766.3333333333
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_016_gauss_lobatto(self):
fct = lambda x: x * x
integrator = gauss_lobatto()
result = integrator(fct, 0.0, 1.0)
benchmark = 0.333333333333
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_095_gauss_lobatto_max_iter_127(self):
fct = lambda x: x * x * x if x > 1 else x - 2
integrator = gauss_lobatto(max_number_of_iterations=255, abs_tolerance=1.0e-7)
result = integrator(fct, -100.0, 99.0)
benchmark = 24009698.499997977
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_023_gauss_lobatto(self):
fct = lambda x: x * x * x if x > 1 else x - 2
integrator = gauss_lobatto()
result = integrator(fct, -100.0, 99.0)
benchmark = 24009698.5
self.assertAlmostEqual(benchmark, result, 8)
def test_integrator_014_gauss_lobatto(self):
fct = lambda x: 2
integrator = gauss_lobatto()
result = integrator(fct, -100.0, 99.0)
benchmark = 398.0
self.assertAlmostEqual(benchmark, result, 8)
