
    g                     \    d dl mZmZmZ d dlmZmZmZmZ d dl	m
Z
 d dlmZ d Zd Zd ZdS )	    )	_doktocsr	_csrtodokbanded)Matrixeyeoneszeros)SparseMatrix)raisesc                     t          g dg dg dg          } t          ddg d          }t          ddg d          }t          ddd	d
d	d          }t          g dg dg dg          }t          d
ddd	d          }t          |           g dg dg dddggk    sJ t          |          g dg dg dddggk    sJ t          |          g dg dg dddggk    sJ t          |          g dg dg dddggk    sJ t          |          g dg d g d!ddggk    sJ t          |          dd	gddgg d"d
dggk    sJ d S )#N)      r   r   )r      	   r   )r   r      r   r      )
      r   r   r   r   r      r   (   r   r   r   r   2   <   F   r   r   r   r   r   r   P   )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   )abcdefs         a/var/www/html/ai-engine/env/lib/python3.11/site-packages/sympy/matrices/tests/test_sparsetools.pytest_doktocsrr*      s   lllLLL,,,?@@AQ & & & 	' 	'AQNNNOOARb!R@@AAAiiiIII677AQAb1122AQ<<...0B0B0Bq!f    Q<<<<<   ///Aq6; ; ; ; ;Q<<...!Q! ! ! ! !Q<<KKK)))B85 5 5 5 5Q<<IIIyyy,,,AGGGGGQ<<QGaV-E-E-E1vNNNNNNN    c            
      T   g dg dg dddgg} g dg dg ddd	gg}g d
g dg dddgg}g dg dg dddgg}ddgddgg dddgg}t          |           }t          |t                    sJ |t          dddd	dd          k    sJ t          |          t          dd	dddd          k    sJ t          |          t          g dg dg          k    sJ t          |          t          ddddddd          k    sJ t          |          t          ddddd          k    s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   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/   r0   )r   
isinstancer
   )hgijkms         r)   test_csrtodokr8      s   	III|||aV4A	YYYq!f5A	YYY			Aq62A			<<<);););aVDA
Q!Q1v.A!Aa&&&&&QAq))+ + + + + +Q<<<1Q**, , , , , ,Q<<<:J:J:J(KLLLLLLQ<<<1RR88: : : : : :Q<<<1q!.D.DEEEEEEEEr+   c            
      8	   t          t          d            t          t          d            t          t          d            t          t          d            t          t          d            t          t          d            t          t          d            t          t          d            t          t          d	            t          t          d
            t          t	          ddi           t
                    sJ t	          ddi           t          dd          k    sJ t	          ddd          t          d          k    sJ t	          dt          ddg          i          t          ddg          k    sJ t	          g dg dd          t	          ddd          cxk    rt          g dg dg dg dg          k    sn J t	          dddddd          t          g dg dg dg          k    sJ d } t	          d| | d          t          g dg d g d!g d"g d#g          k    sJ t	          dddi          t          ddgddgg          k    sJ t	          ddddi          t          g d$g d%g          k    sJ t          g d          }t	          d|id&          t          g d$g d'g d(g d)g d*g          k    sJ t	          ddt          d          i          t          g d+g d+g d,g d,g          k    sJ t          t          d-            t	          dt          d          fdz  d          t          g d.g d/g d0g d1g d2g d3g          k    sJ t          t          d4            t          ddgddgg          }t	          d5|fdz  d          t          g d6g d7g d8g d9g d:g d;g          k    sJ t	          dt          d          fdd          t          g d<g d=g d=g          k    sJ t          t          d>            t	          ddid&          t	          ddid?          cxk    rt          d          k    sn J t	          ddid?          t          g d%g d@g dAg          k    sJ d S )BNc                      t                      S )Nr    r+   r)   <lambda>ztest_banded.<locals>.<lambda>.   s
    fhh r+   c                       t          d          S )Nr   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>/   s    fQii r+   c                  "    t          dd          S Nr   r   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>0   s    fQll r+   c                  $    t          ddd          S )Nr   r   r   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>1   s    fQ1oo r+   c                  &    t          dddd          S )Nr   r   r   r   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>2   s    fQ1a00 r+   c                  (    t          ddid          S )Nr   r   r   r   rowsr;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>3       vq&k::: r+   c                  (    t          ddid          S )Nr   rD   r   colsr;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>4   rG   r+   c                  &    t          dddi          S )Nr   r   rD   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>5   s    va!V55 r+   c                  (    t          ddddi          S )Nr   r   r   rD   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>6       vaQK88 r+   c                  (    t          ddddi          S )Nr   r   r   rD   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>7   rM   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   )rP   r   r   )r   r   r   r   r-   c                     d| z   dz  S r@   r<   )r&   s    r)   r=   ztest_banded.<locals>.<lambda>I   s    1q51* 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   rI   )r   r   r   )r   r   r   )r   r   r   )r   r   r   )r   r   r   r   )r   r   r   r   c                  D    t          dt          d          dd          S )Nr   rO   r   rE   r   r   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>g   s!    v!Q&8&8qAAA 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   c                  H    t          dt          d          fdz  d          S )Nr   r   r   r   r   r   r   rO   rU   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>p   s"    v&d1ggZ\&B&BCC r+   rW   )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   c                  D    t          dt          d          fdd          S )Nr   r   rO   rU   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>   s     v1d1gg,1&=&=>> r+   rE   )r   r   r   r    )
r   	TypeError
ValueErrorr1   r   r
   r	   r   r   r   )svertu2s      r)   test_bandedr^   -   sb   
9&&'''
9''(((
9**+++
9--...
900111
:::;;;
:::;;;
:55666
:88999
:88999fQ2&&55555!QuQ{{****aA,,588++++1faVnn%&&&!Q..8888lll			22339),,--   	 	 	      !QQ1++,, 	 	   
 	A!q\\"" 	 	    !aV	
A	
A 	 	    !QA				 	 	    )))D1d)!$$$										 	 	    !aa\""	 	 	    :AABBBaT!WWJqL))**                  " 	# 	## # # # :CCDDD	!Q!Q 	!	!Bf"q))** 	  	         aa\a(())						 	 	   
 :>>???1a&q!!!VQF%;%;%;EEEEs1vvEEEEEE1a&q!!!V						- & &      r+   N)sympy.matrices.sparsetoolsr   r   r   sympy.matrices.denser   r   r   r	   sympy.matricesr
   sympy.testing.pytestr   r*   r8   r^   r<   r+   r)   <module>rc      s    C C C C C C C C C C ; ; ; ; ; ; ; ; ; ; ; ; ' ' ' ' ' ' ' ' ' ' ' 'O O O(F F F$W W W W Wr+   