File: t_Matrix_operators.py

package info (click to toggle)
openturns 1.7-3
  • links: PTS, VCS
  • area: main
  • in suites: stretch
  • size: 38,588 kB
  • ctags: 26,495
  • sloc: cpp: 144,032; python: 26,855; ansic: 7,868; sh: 419; makefile: 263; yacc: 123; lex: 44
file content (68 lines) | stat: -rwxr-xr-x 1,545 bytes parent folder | download
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
#! /usr/bin/env python

from __future__ import print_function
import openturns as ot

t_names = ['Matrix', 'SquareMatrix', 'TriangularMatrix',
           'SymmetricMatrix', 'CovarianceMatrix', 'CorrelationMatrix']
t_names.extend(
    ['ComplexMatrix', 'HermitianMatrix', 'TriangularComplexMatrix', 'SquareComplexMatrix'])

for i, iname in enumerate(t_names):

    # try conversion
    ref = ot.Matrix([[1.0, 0.0], [0.0, 0.5]])
    a = getattr(ot, iname)(ref)
    print('a=', a)

    # try scalar mul
    try:
        s = 5.
        ats = a * s
        print('a*s=', ats)
        sta = s * a
        print('s*a=', sta)
    except:
        print('no scalar mul for', iname)

    # try scalar div
    try:
        s = 5.
        ads = a / s
        print('a/s=', ads)
    except:
        print('no scalar div for', iname)

    # try vec mul
    try:
        x = ot.NumericalPoint([6, 7])
        ax = a * x
        print('a*x=', ax)
    except:
        print('no vec mul for', iname)

    try:
        a3 = a ** 3
        print('a**3=', a3)
    except:
        print('no pow for', iname)

    for j, jname in enumerate(t_names):
        b = getattr(ot, iname)(ref)
        try:
            ab = a * b
            print('a*b=', ab)
        except:
            print('no mul for', iname, jname)

        try:
            apb = a + b
            print('a+b=', apb)
        except:
            print('no add for', iname, jname)

        try:
            amb = a - b
            print('a-b=', amb)
        except:
            print('no sub for', jname)