
    _-Ph                     n    d Z ddlZddlZddlmZ ddlmZmZmZ ddl	m
Z
 ddlmZ  G d de
          ZdS )	z.Test C/F splitting and simple aggregation vis.    N)
csr_matrix)arrayonesuint32)TestCase)vis_aggregate_groupsc                   &    e Zd Zd Zd Zd Zd ZdS )TestVisc                 6    t          j                    | _        d S N)tempfilemktemp	file_nameselfs    X/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/vis/tests/test_vis.pysetUpzTestVis.setUp   s    !**    c                 8    t          j        | j                   d S r   )osremover   r   s    r   tearDownzTestVis.tearDown   s    
	$.!!!!!r   c                    t          ddgddgddgddgddgddgddgddgddgddgddgddgg          }t          g dg dg dg dg d	g d
g dg dg dg dg dg dgt                    }t          g d          }t          g d          }t          dt                                                    }t	          |||ffd          }t          |||d| j                   d 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,   dtype)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&   shapetriVE2VAggOp	mesh_typefnamer   r   r   ravelr   r   r   r   r4   r5   rowcoldatar6   s          r   test_1zTestVis.test_1   sm   C::::::::::::     YYYYYYYYYYYYYYYYYZZZZ [[ZZ! )/0 0 0 :::;;88899G6***0022D3*-W===qc%#'>	3 	3 	3 	3 	3 	3r   c                 *   t          ddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgddgg          }t          g dg dg d	g d
g dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dg dgt                    }t          g d          }t          g d          }t          dt                                                    }t	          |||ffd !          }t          |||d"| j        #           d S )$Nr   r   r   r   g      @g      @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)   rA   )r)      rA   )r)   r+   rB   )r+      rB   )r+   r*   rC   )r*      rC   )r*   r,   rD   )r,      rD   r-   )r   r"   r&   r    r   r$   r(   r'   r)   r+   r*   r,   r/   rA   rB   rC   rD   rE   )r   r"   r    r   r"   r"   r    r   r   r"   r    r   r   r   r   r&   r   r   )r"      )rF   r$   r0   r2   r3   r9   r;   s          r   test_2zTestVis.test_29   s   C::::::::::::::::::#    & YYYYYYYYYYYYYYYYYYYZZZZ [[ [[ZZ [[ZZ [[ [[!\\!\\!\\)#( +1)2 2 2,  5 5 5 6 6 0 0 0 1 1G6***0022D3*-W===qc%#'>	3 	3 	3 	3 	3 	3r   N)__name__
__module____qualname__r   r   r?   rG    r   r   r
   r
      sQ        + + +" " ""3 "3 "3H33 33 33 33 33r   r
   )__doc__r   r   scipy.sparser   numpyr   r   r   numpy.testingr   	pyamg.visr   r
   rK   r   r   <module>rQ      s    4 4  				 # # # # # # % % % % % % % % % % " " " " " " * * * * * *^3 ^3 ^3 ^3 ^3h ^3 ^3 ^3 ^3 ^3r   