
    _-Ph.                     J    d Z ddlZddlmZ ddlmZmZ  G d de          ZdS )zTest diffusion example.    N)advection_2d)TestCaseassert_array_almost_equalc                       e Zd Zd ZdS )TestAvection2Dc                 :   t          j        g dg dg dg dg          }t          j        g d          }t          dt           j        dz            \  }}t	          ||                                d	
           t	          ||d	
           t          j        g dg dg dg dg          }t          j        g d          }t          dt           j        dz            \  }}t	          ||                                d	
           t	          ||d	
           d S )N)'e?        'er
   )r   r	   r
   r   )r
   r
   r	   r
   )r
   r
   r   r	   )'e?r
   r	   r   )   r      )theta   )decimal)(s=?r
   +Pzr
   )      r   r
   r   )r
   r
   r   r
   )r
   r
   r   r   )g      ?r
   r   g+Pz?r   )nparrayr   pir   todense)selfA_refrhs_refArhss        b/var/www/html/test/jupyter/venv/lib/python3.11/site-packages/pyamg/gallery/tests/test_advection.pytest_simple_finite_differencez,TestAvection2D.test_simple_finite_difference	   sP   DDDMMMDDDLLLN O O (CCCDDfBE!G4443!%a@@@@!'3:::: AAAHHHAAAHHHJ K K (;;;<<fBE!G4443!%a@@@@!'3::::::    N)__name__
__module____qualname__r    r    r   r   r      s#        ; ; ; ; ;r    r   )	__doc__numpyr   pyamg.galleryr   numpy.testingr   r   r   r$   r    r   <module>r)      su          & & & & & & = = = = = = = =; ; ; ; ;X ; ; ; ; ;r    