
    _-Ph'	                     ^    d Z ddlZddlZddlmZ ddlmZmZ ddl	m
Z
  G d de          ZdS )zOTest 1, 2, 8 element meshes writing the vtu file.  Validate as well-formed xml.    N)TestCase)arrayuint32)	write_vtuc                       e Zd Zd Zd ZdS )TestWriteVtuc                    g } G d d          } |            }t          ddgddgddgg          |_        t          g dgt                    }d|i|_        d |_        d |_        |                    |           t          ddgddgddgddgg          |_        t          g dg dgt                    }d|i|_        d |_        d |_        |                    |           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g	          |_        t          g d
g dg dg dg dg dg dg dgt                    }d|i|_        d |_        d |_        |                    |           || _        d S )Nc                   <    e Zd Z ej                    ZdZdZdZdZ	dS ) TestWriteVtu.setUp.<locals>.MeshN)
__name__
__module____qualname__tempfilemktemp	file_nameVE2Vpdatacdata     X/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/vis/tests/test_vtu.pyMeshr      s1        '))IACEEEEr   r   g        g      ?)r            )r      r   )r   r   r   g       @)r      r   )r   r   r   )r   r   r   )r   r   r   )r         )r   r   r   )r      r   )r   r   r!   )	r   r   r   cellsr   r   appendVscases)selfr%   r   meshr   s        r   setUpzTestWriteVtu.setUp   s   	 	 	 	 	 	 	 	 tvv c
c
c
$ % % YYYK((X


T #ssss% & & YYYYY !') )X


T c
c
c
c
c
c
c
c
c
$ % % YYYYYYYYYYYYYYYYY  "() ) X


T


r   c                    | j         D ]}t          |j        |j        |j        |j        |j                   	 t          j        j	        
                                }t          |j        d          5 }|                    |           d d d            n# 1 swxY w Y   # t          $ r}t          d|             d }~ww xY wd S )N)r   r"   r   r   fnamerbz	problem: )r%   r   r   r"   r   r   r   xmlparsersexpatParserCreateopen	ParseFile	Exceptionprint)r&   r'   parserfexs        r   test_xmlzTestWriteVtu.test_xmlH   s   J 	 	Ddj Jdj N, , , ,*7799$.$// (1$$Q'''( ( ( ( ( ( ( ( ( ( ( ( ( ( (   &"&&'''	 	s;   8B0BBB	BB	B
C (B;;C N)r   r   r   r(   r7   r   r   r   r   r      s3        8 8 8t    r   r   )__doc__r   xml.parsers.expatr,   numpy.testingr   numpyr   r   	pyamg.visr   r   r   r   r   <module>r=      s    U U      " " " " " "              G G G G G8 G G G G Gr   