1import cmsisdsp as dsp
2import numpy as np
3from numpy.testing import assert_allclose
4from scipy.stats import entropy,tstd, tvar
5from scipy.special import logsumexp
6from scipy.linalg import cholesky,ldl,solve_triangular
7from scipy import signal
8
9
10import colorama
11from colorama import init,Fore, Back, Style
12
13init()
14
15def printTitle(s):
16    print("\n" + Fore.GREEN + Style.BRIGHT +  s + Style.RESET_ALL)
17
18def printSubTitle(s):
19    print("\n" + Style.BRIGHT + s + Style.RESET_ALL)
20
21
22print("Cholesky")
23
24a=np.array([[4,12,-16],[12,37,-43],[-16,-43,98]])
25ref=cholesky(a,lower=True)
26print(ref)
27
28status,res=dsp.arm_mat_cholesky_f32(a)
29print(res)
30assert_allclose(ref,res,1e-6,1e-6)
31
32status,res=dsp.arm_mat_cholesky_f64(a)
33print(res)
34assert_allclose(ref,res,1e-10,1e-10)