summaryrefslogtreecommitdiff
path: root/numpy/linalg/tests/test_regression.py
blob: 4e2295539b9a529fd3e0dd79aa9e32dd829c7df2 (plain)
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
""" Test functions for linalg module
"""

from numpy.testing import *
set_package_path()
from numpy import linalg, arange, float64
restore_path()

rlevel = 1

class TestRegression(NumpyTestCase):
    def test_eig_build(self, level = rlevel):
        """Ticket #652"""
        rva = [1.03221168e+02 +0.j, 
               -1.91843603e+01 +0.j,
               -6.04004526e-01+15.84422474j, 
               -6.04004526e-01-15.84422474j,
               -1.13692929e+01 +0.j,
               -6.57612485e-01+10.41755503j,
               -6.57612485e-01-10.41755503j,
               1.82126812e+01 +0.j,
               1.06011014e+01 +0.j ,
               7.80732773e+00 +0.j ,
               -7.65390898e-01 +0.j,
               1.51971555e-15 +0.j ,
               -1.51308713e-15 +0.j]        
        a = arange(13*13, dtype = float64)
        a.shape = (13,13)
        a = a%17
        va, ve = linalg.eig(a)
        assert_array_almost_equal(va, rva)

if __name__ == '__main__':
    NumpyTest().run()