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
|
import numpy
from numpy import cos, sin, pi
from numpy.testing import *
set_package_path()
from scipy.integrate import quadrature, romberg, romb
restore_path()
class test_quadrature(ScipyTestCase):
def quad(self, x, a, b, args):
raise NotImplementedError
def check_quadrature(self):
# Typical function with two extra arguments:
def myfunc(x,n,z): # Bessel function integrand
return cos(n*x-z*sin(x))/pi
val, err = quadrature(myfunc,0,pi,(2,1.8))
table_val = 0.30614353532540296487
assert_almost_equal(val, table_val, decimal=7)
def check_romberg(self):
# Typical function with two extra arguments:
def myfunc(x, n, z): # Bessel function integrand
return cos(n*x-z*sin(x))/pi
val = romberg(myfunc,0,pi, args=(2, 1.8))
table_val = 0.30614353532540296487
assert_almost_equal(val, table_val, decimal=7)
def check_romb(self):
assert_equal(romb(numpy.arange(17)),128)
if __name__ == "__main__":
ScipyTest().run()
|
