Coverage for gpaw/test/linalg/test_lapack.py: 100%

1import numpy as np 

2 

3import pytest 

4from gpaw.lrtddft.apmb import sqrt_matrix 

5 

6# check sqrt of a matrix 

7 

8 

9def test_linalg_lapack(): 

10 A = [[20, 4], [4, 1]] 

11 a = [[4.4, 0.8], [0.8, 0.6]] 

12 A = np.array(A, float) 

13 print('A=', A) 

14 a = np.array(a) 

15 b = sqrt_matrix(A) 

16 print('sqrt(A)=', b) 

17 assert ((a - b)**2).sum() == pytest.approx(0, abs=1.0e-12)