
    sg                     Z    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y)	    )	_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 y )#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         X/var/www/html/venv/lib/python3.12/site-packages/sympy/matrices/tests/test_sparsetools.pytest_doktocsrr*      sY   lL,?@AQ & 	'AQNOARb!R@AAiI67AQAb12AQ<.0Bq!f   Q<< /Aq6; ; ; ;Q<.!Q! ! ! !Q<K)B85 5 5 5Q<Iy,AGGGGQ<QGaV-E1vNNNN    c            
         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 y )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:   	I|aV4A	Yq!f5A	Y	Aq62A	<);aVDA
Q!Q1v.A!Aa&&&QAq)+ + + +Q<<1Q*, , , ,Q<<:J(KLLLLQ<<1RR8: : : :Q<<1q!.DEEEEr+   c            
      T   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J  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J  J t	        ddid?      t        g d%g d@g dAg      k(  sJ y )BNc                      t               S )Nr    r+   r)   <lambda>ztest_banded.<locals>.<lambda>.   s    fh r+   c                      t        d      S )Nr   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>/   s
    fQi r+   c                      t        dd      S Nr   r   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>0   s    fQl r+   c                      t        ddd      S )Nr   r   r   r;   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>1   s    fQ1o 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 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 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 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                  4    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&8q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                  8    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&d1gZ\&B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                  4    t        dt        d      fdd      S )Nr   r   rO   rU   r<   r+   r)   r=   ztest_banded.<locals>.<lambda>   s    v1d1g,1&=> r+   rE   )r   r   r   r    )
r   	TypeError
ValueErrorr1   r   r
   r	   r   r   r   )svertu2s      r)   test_bandedr^   -   s   
9&'
9'(
9*+
9-.
901
::;
::;
:56
:89
:89fQ2&555!QuQ{***aA,58+++1faVn%&&!Q.888l	239),-	 	     !QQ1+, 	  
 	A!q\" 	   !aV	
A	
A 	   !QA 	   )D1d)!$ 	   !aa\"	 	   :ABaT!WJqL)*      " 	## # # :CD	!Q!Q 	!Bf"q)* 	       aa\a() 	  
 :>?1a&q!VQF%;Es1v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 ; ; ' 'O(F$Wr+   