a
    kh                     @   s`   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d Zdd Zd	d
 ZdS )    )	_doktocsr	_csrtodokbanded)Matrixeyeoneszeros)SparseMatrix)raisesc                  C   st  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sJ t|g dg dg dddggksJ t|g dg dg dddggksJ 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sFJ t|dd	gddgg d"d
dggkspJ 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f r'   S/var/www/auris/lib/python3.9/site-packages/sympy/matrices/tests/test_sparsetools.pytest_doktocsr   s(    


*r)   c               	   C   sF  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sJ |tdddd	ddksJ t |tdd	ddddksJ t |tg dg dgksJ t |tdddddddks"J t |tdddddksBJ 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,   r-   )r   
isinstancer	   )hgijkmr'   r'   r(   test_csrtodok   s$    



 r5   c               	   C   s  t tdd  t tdd  t tdd  t tdd  t tdd  t tdd  t tdd  t td	d  t td
d  t tdd  ttddi tsJ tddi tddksJ tdddtdksJ tdtddgitddgksJ tg dg ddtddd  krJtg dg dg dg dgksPn J tddddddtg dg dg dgksJ dd } td| | dtg d g d!g d"g d#g d$gksJ tddditddgddggksJ tdddditg d%g d&gksJ tg d}td|idd'tg d%g d(g d)g d*g d+gksbJ tddtditg d,g d,g d-g d-gksJ t td.d  tdtdfd dtg d/g d0g d1g d2g d3g d4gksJ t td5d  tddgddgg}td6|fd dtg d7g d8g d9g d:g d;g d<gksZJ tdtdfddtg d=g d>g d>gksJ t td?d  tddidd'tddidd@  krt	dksn J tddidd@tg d&g dAg dBgksJ d S )CNc                   S   s   t  S )Nr   r'   r'   r'   r(   <lambda>.       ztest_banded.<locals>.<lambda>c                   S   s   t dS )Nr   r6   r'   r'   r'   r(   r7   /   r8   c                   S   s
   t ddS Nr   r   r6   r'   r'   r'   r(   r7   0   r8   c                   S   s   t dddS )Nr   r   r   r6   r'   r'   r'   r(   r7   1   r8   c                   S   s   t ddddS )Nr   r   r   r   r6   r'   r'   r'   r(   r7   2   r8   c                   S   s   t ddiddS )Nr   r   r   r   rowsr6   r'   r'   r'   r(   r7   3   r8   c                   S   s   t ddiddS )Nr   r:   r   colsr6   r'   r'   r'   r(   r7   4   r8   c                   S   s   t dddiS )Nr   r   r:   r6   r'   r'   r'   r(   r7   5   r8   c                   S   s   t ddddiS )Nr   r   r   r:   r6   r'   r'   r'   r(   r7   6   r8   c                   S   s   t ddddiS )Nr   r   r   r:   r6   r'   r'   r'   r(   r7   7   r8   r   r   r   r   r   r   )r   r   r   r   )r   r   r   )r   r    )r   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                 S   s   d|  d S r9   r'   )r$   r'   r'   r(   r7   I   r8   r   r,   )r   r   r   r   r   )r   r   r   r   r   )r   r   r   r   r   )r   r   r      r   )r   r   r   r      )r   r   r   )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                   S   s   t dtddddS )Nr   r?   r   r;   r   r   r'   r'   r'   r(   r7   g   r8   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )r   r   r   r   r   r   r   r   )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                   S   s   t dtdfd dS )Nr   r   r   r   r   r   r   r?   rC   r'   r'   r'   r(   r7   p   r8   rD   )r   r   r   r   r   r   r   )r   r   r   r   r   r   r   )r   r   r   r   r   r   r   )r   r   r   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                   S   s   t dtdfddS )Nr   r   r?   rC   r'   r'   r'   r(   r7      r8   r;   )r   r   r   r   )
r
   	TypeError
ValueErrorr.   r   r	   r   r   r   r   )sZvertu2r'   r'   r(   test_banded-   s    $









:rI   N)Zsympy.matrices.sparsetoolsr   r   r   Zsympy.matrices.denser   r   r   r   Zsympy.matricesr	   Zsympy.testing.pytestr
   r)   r5   rI   r'   r'   r'   r(   <module>   s   