1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
|
"""
Copyright (C) 2000, 2001, 2002, 2003 RiskMap srl
This file is part of QuantLib, a free-software/open-source library
for financial quantitative analysts and developers - http://quantlib.org/
QuantLib is free software: you can redistribute it and/or modify it
under the terms of the QuantLib license. You should have received a
copy of the license along with this program; if not, please email
<quantlib-dev@lists.sf.net>. The license is also available online at
<https://www.quantlib.org/license.shtml>.
This program is distributed in the hope that it will be useful, but WITHOUT
ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
FOR A PARTICULAR PURPOSE. See the license for more details.
"""
import QuantLib as ql
import unittest
import math
class IntegralTest(unittest.TestCase):
def Gauss(self, x):
return math.exp(-x * x / 2.0) / math.sqrt(2 * math.pi)
def singleTest(self, I):
tolerance = 1e-4
cases = [
["f(x) = 1", lambda x: 1, 0.0, 1.0, 1.0],
["f(x) = x", lambda x: x, 0.0, 1.0, 0.5],
["f(x) = x^2", lambda x: x * x, 0.0, 1.0, 1.0 / 3.0],
["f(x) = sin(x)", math.sin, 0.0, math.pi, 2.0],
["f(x) = cos(x)", math.cos, 0.0, math.pi, 0.0],
["f(x) = Gauss(x)", self.Gauss, -10.0, 10.0, 1.0],
]
for tag, f, a, b, expected in cases:
calculated = I(f, a, b)
if not (abs(calculated - expected) <= tolerance):
self.fail(
"""
integrating %(tag)s
calculated: %(calculated)f
expected : %(expected)f
"""
% locals()
)
def testSegment(self):
"Testing segment integration"
self.singleTest(ql.SegmentIntegral(10000))
def testTrapezoid(self):
"Testing trapezoid integration"
self.singleTest(ql.TrapezoidIntegralDefault(1.0e-4, 1000))
def testSimpson(self):
"Testing Simpson integration"
self.singleTest(ql.SimpsonIntegral(1.0e-4, 1000))
def testKronrod(self):
"Testing Gauss-Kronrod integration"
self.singleTest(ql.GaussKronrodAdaptive(1.0e-4))
if __name__ == "__main__":
print("testing QuantLib", ql.__version__)
unittest.main(verbosity=2)
|
