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
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
|
from numpy.testing import *
from numpy import *
class _test_ev:
def check_syev(self,level=1,sym='sy',suffix=''):
a = [[1,2,3],[2,2,3],[3,3,6]]
exact_w = [-0.6699243371851365,0.4876938861533345,9.182230451031804]
f = getattr(self.lapack,sym+'ev'+suffix)
w,v,info=f(a)
assert not info,`info`
assert_array_almost_equal(w,exact_w)
for i in range(3):
assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i])
def check_syevd(self): self.check_syev(suffix='d')
#def check_heev(self): self.check_syev(sym='he')
#def check_heevd(self): self.check_syev(sym='he',suffix='d')
## def check_heev_complex(self,level=1,suffix=''):
## a= [[1,2-2j,3+7j],[2+2j,2,3],[3-7j,3,5]]
## exact_w=[-6.305141710654834,2.797880950890922,11.50726075976392]
## f = getattr(self.lapack,'heev'+suffix)
## w,v,info=f(a)
## assert not info,`info`
## assert_array_almost_equal(w,exact_w)
## for i in range(3):
## assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i],self.decimal)
#def check_heevd_complex(self): self.check_heev_complex(suffix='d')
def check_syevr(self,level=1,sym='sy'):
a = [[1,2,3],[2,2,3],[3,3,6]]
exact_w = [-0.6699243371851365,0.4876938861533345,9.182230451031804]
f = getattr(self.lapack,sym+'evr')
w,v,info = f(a)
assert not info,`info`
assert_array_almost_equal(w,exact_w)
for i in range(3):
assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i])
## def check_heevr_complex(self,level=1):
## a= [[1,2-2j,3+7j],[2+2j,2,3],[3-7j,3,5]]
## exact_w=[-6.305141710654834,2.797880950890922,11.50726075976392]
## f = self.lapack.heevr
## w,v,info = f(a)
## assert not info,`info`
## assert_array_almost_equal(w,exact_w)
## for i in range(3):
## assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i])
## def check_heevr(self): self.check_syevr(sym='he')
def check_syevr_irange(self,level=1,sym='sy',irange=[0,2]):
a = [[1,2,3],[2,2,3],[3,3,6]]
exact_w = [-0.6699243371851365,0.4876938861533345,9.182230451031804]
f = getattr(self.lapack,sym+'evr')
w,v,info = f(a,irange=irange)
assert not info,`info`
rslice = slice(irange[0],irange[1]+1)
m = irange[1] - irange[0] + 1
assert_equal(len(w),m)
assert_array_almost_equal(w,exact_w[rslice])
for i in range(m):
assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i])
def check_syevr_irange_low(self): self.check_syevr_irange(irange=[0,1])
def check_syevr_irange_mid(self): self.check_syevr_irange(irange=[1,1])
def check_syevr_irange_high(self): self.check_syevr_irange(irange=[1,2])
## def check_heevr_irange(self): self.check_syevr_irange(sym='he')
## def check_heevr_irange_low(self): self.check_syevr_irange(sym='he',irange=[0,1])
## def check_heevr_irange_high(self): self.check_syevr_irange(sym='he',irange=[1,2])
def check_syevr_vrange(self,level=1,sym='sy',vrange=None):
a = [[1,2,3],[2,2,3],[3,3,6]]
exact_w = [-0.6699243371851365,0.4876938861533345,9.182230451031804]
if vrange is None:
vrange = [-1,10]
ew = [value for value in exact_w if vrange[0]<value<=vrange[1]]
f = getattr(self.lapack,sym+'evr')
w,v,info = f(a,vrange=vrange)
assert not info,`info`
assert_array_almost_equal(w,ew)
m = len(w)
for i in range(m):
assert_array_almost_equal(dot(a,v[:,i]),w[i]*v[:,i])
def check_syevr_vrange_low(self): self.check_syevr_vrange(vrange=[-1,1])
def check_syevr_vrange_mid(self): self.check_syevr_vrange(vrange=[0,1])
def check_syevr_vrange_high(self): self.check_syevr_vrange(vrange=[1,10])
## def check_heevr_vrange(self): self.check_syevr_vrange(sym='he')
## def check_heevr_vrange_low(self): self.check_syevr_vrange(sym='he',vrange=[-1,1])
## def check_heevr_vrange_high(self): self.check_syevr_vrange(sym='he',vrange=[1,10])
|
