
    _-Ph"                     V    d Z ddlZddlmZ ddlmZ ddlmZm	Z	  G d de          Z
dS )zTest tentative interpolation.    N)
csr_matrix)fit_candidates)TestCaseassert_almost_equalc                       e Zd Zd Zd ZdS )TestFitCandidatesc           
      )   g | _         | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d	          t          j        d
          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        d
          fd          t          j        d                              dd          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d	          t          j        d
          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        d
          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        t          j        d          dt          j        d          z  f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          dt          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        t          j        d
          t          j        d
          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        g d          fd          t          j        d          f           | j                             t          t          j        d          t          j        g d          t          j        g d          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        g d          fd          t          j        t          j        d          t          j        d          t          j        d          dz  f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        g d          fd          t          j        t          j        d          t          j        d          t          j        d          dz  f          j	        f           | j                             t          t          j        d          t          j        g d          t          j        g d          fd          t          j        t          j        d          t          j        d          f          j	        f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d"t          j        d!          z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d#t          j        d!          z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d$t          j        d!          z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d#t          j        d                              dd          z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d$t          j        d                              dd          z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          t          j        d%gd&gd'gd(gg          f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d"t          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          d$t          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d           t          j        d          fd!          t          j        d%d$gd&d)gd'd*gd(d+gg          f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d$t          j        d                              dd          z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d#t          j        d                              dd          z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d$t          j        d                              dd          z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d"t          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d#t          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          d$t          j        t          j        d          t          j        d          f          j	        z  f           | j                             t          t          j        d          t          j        g d          t          j        d          fd          t          j        d%d$gd&d)gd'd*gd(d+gg          f           d S ),N   )r   r   r      r      )r
      )shape)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   g#B;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   y      ?        y              @y      ?      @y            @y              @y      @       y      "@       y      @        y      @      ?y      @       @)
casesappendr   nponesarrayarangereshapevstackT)selfs    f/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/aggregation/tests/test_tentative.pysetUpzTestFitCandidates.setUp   sT   
 	


BH___$=$=ry||L#% % %&(gfoo7 	8 	8 	8 	


BH___$=$=ry||L#% % %&(gfoo7 	8 	8 	8 	


BH-H-H-H$I$I	"'#% % %&(gfoo7 	8 	8 	8 	


BH-H-H-H$I$I	"'#% % %&(ill&:&:1a&@&@B 	C 	C 	C
 	


BH\\\$:$:BIaLLI#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P 	


BH-H-H-H$I$I	"'#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P 	


BH-H-H-H$I$I	"'#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P
 	


BH\\\$:$:BIaLLI#% % %Irwqzz529Q<<#7899;= 	> 	> 	> 	


BH\\\$:$:BIaLLI#% % %BIrwqzz29Q<<899;;= 	> 	> 	> 	


BHYYY$7$71F#% % %&(gfoo7 	8 	8 	8 	


BHYYY$7$71F#% % %&(gfoo7 	8 	8 	8 	


BH___$=$=ry||L#% % %&(gg&6&68 	9 	9 	9
 	


BHYYY$7$71F#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P 	


BHYYY$7$71F#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P 	


BH___$=$=ry||L#% % %Irwr{{BIbMM23357 	8 	8 	8 	


BH\\\$:$:!3!3!3446#% % %&(gfoo7 	8 	8 	8 	


BH\\\$:$:!3!3!3446#% % %&(iRYq\\0J&K&K&MO 	P 	P 	P 	


BH\\\$:$:!3!3!3446#% % % Irwqzz29Q<<1qABBD	F 	G 	G 	G
 	


BH-?-?-?$@$@!?!?!?@@B#% % % Irwqzz29Q<<11BCDDF	H 	I 	I 	I
 	


BH-?-?-?$@$@!?!?!?@@B#% % % Irwqzz29Q<<0113	5 	6 	6 	6 	


BH\\\$:$:BIaLLI#% % %'-&@B 	C 	C 	C 	


BH\\\$:$:BIaLLI#% % %'-&@B 	C 	C 	C 	


BH\\\$:$:BIaLLI#% % %'-&@B 	C 	C 	C 	


BH\\\$:$:BIaLLI#% % %'-11E1Ea1K1K&KM 	N 	N 	N 	


BH\\\$:$:BIaLLI#% % %'-11E1Ea1K1K&KM 	N 	N 	N 	


BH\\\$:$:BIaLLI#% % %Hwi&F8fX>??A 	B 	B 	B 	


BH\\\$:$:BIaLLI#% % %ry"'!**bill!;<<>>@ 	A 	A 	A
 	


BH\\\$:$:BIaLLI#% % %ry"'!**bill!;<<>>@ 	A 	A 	A 	


BH\\\$:$:BIaLLI#% % %Hw'&&)9v&(8: ; ;< 	= 	= 	= 	


BH\\\$:$:BIaLLI#% % %'-11E1Ea1K1K&KM 	N 	N 	N 	


BH\\\$:$:BIaLLI#% % %'-11E1Ea1K1K&KM 	N 	N 	N 	


BH\\\$:$:BIaLLI#% % %'-11E1Ea1K1K&KM 	N 	N 	N 	


BH\\\$:$:BIaLLI#% % %ry"'!**bill!;<<>>@ 	A 	A 	A 	


BH\\\$:$:BIaLLI#% % %ry"'!**bill!;<<>>@ 	A 	A 	A 	


BH\\\$:$:BIaLLI#% % %ry"'!**bill!;<<>>@ 	A 	A 	A 	


BH\\\$:$:BIaLLI#% % %Hw'&&)9v&(8: ; ;< 	= 	= 	= 	= 	=    c                     d }| j         D ]d\  }} |||           t          ||          \  }}t          |||z             t          ||j                                        |z  z  |           ed S )Nc                 ~    d|t          j        t          j        | j                  dk              d         d d f<   d S )Nr   )r   wherediffindptr)AggOp
candidatess     r   mask_candidatez8TestFitCandidates.test_all_cases.<locals>.mask_candidate   s;    EFJrx 5 5 :;;A>ABBBr   )r   r   r   r   	conjugate)r   r'   r%   fine_candidatesQcoarse_candidatess         r   test_all_casesz TestFitCandidates.test_all_cases   s    	G 	G 	G '+j 	Z 	Z"E?N5/222#1%#I#I A   5F1FGGGQS]]__%F GYYYY	Z 	Zr   N)__name__
__module____qualname__r   r,    r   r   r   r   
   s;        `= `= `=DZ Z Z Z Zr   r   )__doc__numpyr   scipy.sparser   pyamg.aggregation.aggregationr   numpy.testingr   r   r   r0   r   r   <module>r6      s    # #     # # # # # # 8 8 8 8 8 8 7 7 7 7 7 7 7 7oZ oZ oZ oZ oZ oZ oZ oZ oZ oZr   