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
|
#!/usr/bin/env python
#
# Created by: Pearu Peterson, September 2002
#
from numpy.testing import *
from numpy import ones
from scipy.linalg import flapack, clapack
class TestFlapackSimple(TestCase):
def test_gebal(self):
a = [[1,2,3],[4,5,6],[7,8,9]]
a1 = [[1,0,0,3e-4],
[4,0,0,2e-3],
[7,1,0,0],
[0,1,0,0]]
for p in 'sdzc':
f = getattr(flapack,p+'gebal',None)
if f is None: continue
ba,lo,hi,pivscale,info = f(a)
assert not info,`info`
assert_array_almost_equal(ba,a)
assert_equal((lo,hi),(0,len(a[0])-1))
assert_array_almost_equal(pivscale,ones(len(a)))
ba,lo,hi,pivscale,info = f(a1,permute=1,scale=1)
assert not info,`info`
#print a1
#print ba,lo,hi,pivscale
def test_gehrd(self):
a = [[-149, -50,-154],
[ 537, 180, 546],
[ -27, -9, -25]]
for p in 'd':
f = getattr(flapack,p+'gehrd',None)
if f is None: continue
ht,tau,info = f(a)
assert not info,`info`
class TestLapack(TestCase):
def test_flapack(self):
if hasattr(flapack,'empty_module'):
print """
****************************************************************
WARNING: flapack module is empty
-----------
See scipy/INSTALL.txt for troubleshooting.
****************************************************************
"""
def test_clapack(self):
if hasattr(clapack,'empty_module'):
print """
****************************************************************
WARNING: clapack module is empty
-----------
See scipy/INSTALL.txt for troubleshooting.
Notes:
* If atlas library is not found by numpy/distutils/system_info.py,
then scipy uses flapack instead of clapack.
****************************************************************
"""
if __name__ == "__main__":
run_module_suite()
|
