
    _-Ph                     z    d Z ddlZddlmZmZmZ ddlmZ ddl	m
Z
mZ  G d de          Z G d d	e          ZdS )
zTest Laplacian example.    N)TestCaseassert_equalassert_almost_equal)eigvals)poissongauge_laplacianc                       e Zd Zd ZdS )TestPoissonc                 b   g }|                     dt          j        dgg          f           |                     dt          j        ddgddgg          f           |                     dt          j        g dg dg dg d	g          f           |                     d
t          j        dgg          f           |                     dt          j        ddgddgg          f           |                     dt          j        ddgddgg          f           |                     dt          j        g dg dg dg          f           |                     dt          j        g dg dg dg dg          f           |                     dt          j        g dg dg dg dg          f           |                     dt          j        g dg dg dg d g d!g d"g d#g d$g          f           |D ]4\  }}t          t	          |                                          |           5d S )%N)      )r   )   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   )r   r   )r   r   )r      )r   r   r   )r   r   r   )r   r   r   )r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   )   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   r   )r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )appendnparrayr   r   toarray)selfcasesgridexpecteds       b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/gallery/tests/test_laplacian.pytest_poissonzTestPoisson.test_poisson	   s    	dBHqcUOO,---dBHq"g')1g&/ 0 0 1 	2 	2 	2dBHmmm&4nn&4nn&3mm&5 6 6 7 	8 	8 	8 	fbhuoo.///fbhB)+Q(1 2 2 3 	4 	4 	4fbhB)+Q(1 2 2 3 	4 	4 	4fbh


(3(2

(4 5 5 6 	7 	7 	7 	fbh(6(6(6(8 9 9 : 	; 	; 	;
 	i>>>+9>>+9>>+9>>+; "< "< = 	> 	> 	> 	i+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+F+H "I "I J 	K 	K 	K $ 	< 	<ND(..00(;;;;	< 	<    N)__name__
__module____qualname__r    r   r   r
   r
      s#        (< (< (< (< (<r   r
   c                       e Zd Zd ZdS )TestGaugeLaplacianc           
      &   g }d}d}t          |d|          }|                    ||f           d}d}t          |d|          }|                    ||f           d}t          |d|          }|                    ||f           t          |d|          }|                    ||f           d	}t          |d|          }|                    ||f           t          |d
|          }|                    ||f           |D ]\  }}||j                                        z
  }t	          |j        t          j        g                      t          |	                                          }|dk    r0t          t          t          j        |                    d           t          t          t          j        t          j        |                              d           t          t          j        |                    dk    sJ 
d S )Ng           g      ?)spacingbetag)\(?r      g?gzG?g333333&@)r   r   T	conjugater   datar   r   r   r   r   minabsimagreal)r   r   r&   nptsAdiffes          r   test_gaugelaplacianz&TestGaugeLaplacian.test_gaugelaplacian5   s   D#D999aYD#D999aYD#D999aYD#D999aYD#D999aYD$T:::aY 	- 	-GAtqs}}&DBHRLL111 		$$As{{#Cq		NNC8888 $Crwqzz(:(:$;$;SAAA271::,,,,,	- 	-r   N)r   r   r   r3   r    r   r   r"   r"   4   s#        &- &- &- &- &-r   r"   )__doc__numpyr   numpy.testingr   r   r   scipy.linalgr   pyamg.gallery.laplacianr   r   r
   r"   r    r   r   <module>r9      s          E E E E E E E E E E             < < < < < < < <)< )< )< )< )<( )< )< )<X'- '- '- '- '- '- '- '- '- '-r   