Ë
    âQ(h™  ã                   óÌ  — d dl Z d dlZd dlmZ d dlmc mZ d dl	m
Z
mZ 	 d dlZe j                  j                  edu d¬«      ZdZ e j"                  d«       e j"                  de j                  j%                  e¬«      g¬«      fZej(                  ej*                  ej,                  ej.                  ej0                  ej2                  gZ e j6                  e¬	«      d
„ «       Z e j6                  e¬	«      d„ «       Ze j6                  d„ «       Ze j6                  d„ «       Zd„ Z d„ Z!d„ Z"d„ Z#d„ Z$d„ Z%d„ Z&d„ Z'd„ Z(d„ Z)d„ Z*d„ Z+d„ Z,d„ Z-d„ Z.d„ Z/d„ Z0d„ Z1y# e$ r dZY �Œ=w xY w) é    N)Úassert_allcloseÚassert_equalzpydata/sparse not installed)Úreasonz>pydata/sparse (0.15.1) does not implement necessary operationsÚCOOÚDOK)Úmarks)Úparamsc                 ó6   — t        t        | j                  «      S ©N)ÚgetattrÚsparseÚparam©Úrequests    új/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/scipy/sparse/linalg/tests/test_pydata_sparse.pyÚ
sparse_clsr   "   s   € ä”6˜7Ÿ=™=Ó)Ð)ó    c                 ó   — | j                   S r   )r   r   s    r   Úsp_sparse_clsr   '   s   € à�=‰=Ðr   c                 ó    — t         j                  j                  d«       t         j                  j                  dd«      } ||«       | |«      fS ©NiÒ  é	   )ÚnpÚrandomÚseedÚrand)r   r   ÚA_denses      r   Úsame_matrixr   ,   s;   € ä‡I�I‡N�N�4ÔÜ�i‰i�n‰n˜Q Ó"€GÙ˜Ó!¡:¨gÓ#6Ð6Ð6r   c                 óö   — t         j                  j                  d«       t         j                  j                  dd«      }||j                  z  } | |«      }t         j                  j                  d«      }|||fS r   )r   r   r   r   ÚT)r   r   ÚA_sparseÚbs       r   Úmatricesr#   3   s]   € ä‡I�I‡N�N�4ÔÜ�i‰i�n‰n˜Q Ó"€GØ˜Ÿ	™	Ñ!€GÙ˜'Ó"€HÜ
�	‰	�‰�qÓ€AØ�H˜aÐÐr   c                 óª   — | \  }}}t        j                  ||d¬«      \  }}|dk(  sJ ‚t        |t        j                  «      sJ ‚t        ||z  |«       y )NgVçž¯Ò<©Úatolr   )ÚsplinÚgmresÚ
isinstancer   Úndarrayr   )r#   r   r!   r"   ÚxÚinfos         r   Útest_isolve_gmresr-   =   sT   € ð $Ñ€GˆX�qÜ�k‰k˜( A¨EÔ2�G€A€tØ�1Š9Ðˆ9Ü�aœŸ™Ô$Ð$Ð$Ü�H˜q‘L !Õ$r   c                 ó�   — | \  }}}t        j                  ||«      }t        j                  ||«      }t        |d   |d   d¬«       y )Nr   çü©ñÒMbP?r%   )r'   Úlsmrr   ©r#   r   r!   r"   Úres0Úress         r   Ú	test_lsmrr4   G   óB   € Ø#Ñ€GˆX�qÜ�:‰:�g˜qÓ!€DÜ
�*‰*�X˜qÓ
!€CÜ�C˜‘F˜D ™G¨$Ö/r   c                  óØ   — t        j                  t        j                  d«      t        j                  d«      t        j                  d«      ¬«      d   } t        | j                  d«       y )N)é
   é   r7   r8   )ÚAr"   Úx0r   )r8   )r'   r0   r   ÚonesÚzerosr   Úshape)r+   s    r   Útest_lsmr_output_shaper>   O   s@   € Ü�
‰
”R—W‘W˜WÓ%¬¯©°"«¼"¿'¹'À!»*ÔEÀaÑH€AÜ�—‘˜$Õr   c                 ó�   — | \  }}}t        j                  ||«      }t        j                  ||«      }t        |d   |d   d¬«       y )Nr   gñhãˆµøä>r%   )r'   Úlsqrr   r1   s         r   Ú	test_lsqrrA   T   r5   r   c                 óP  — | \  }}}t        j                  |dz  «      }|j                  |«      }t        j                  |d|¬«      \  }}t        j                  |d|¬«      \  }}	t        ||«       t        |	|«       ||fD ]¢  }
t        j                  ||d|¬«      \  }}t        j                  ||
d|¬«      \  }}	t        ||«       t        |	|«       t        j                  ||d|¬«      \  }}t        j                  ||
d|¬«      \  }}	t        ||«       t        |	|«       Œ¤ y )Né   é   ©ÚkÚv0)ÚMrF   rG   )r   ÚdiagÚ	__class__r'   Úeigsr   Úeigsh)r#   r   r!   rG   ÚM_denseÚM_sparseÚw_denseÚv_denseÚwÚvrH   s              r   Ú	test_eigsrS   [   s  € Ø$Ñ€GˆX�rä�g‰g�b˜!‘e‹n€GØ×!Ñ! 'Ó*€Hä—z‘z '¨Q°2Ô6Ñ€GˆWÜ�:‰:�h !¨Ô+�D€A€qä�A�wÔÜ�A�wÔà˜Ð ò $ˆÜ Ÿ:™: g°¸AÀ"ÔEÑˆ�Ü�z‰z˜( a¨1°Ô4‰ˆˆ1ä˜˜7Ô#Ü˜˜7Ô#ä Ÿ;™; w°'¸QÀ2ÔFÑˆ�Ü�{‰{˜8 q¨A°"Ô5‰ˆˆ1ä˜˜7Ô#Ü˜˜7Õ#ñ$r   c                 ó`  — | \  }}}t        j                  |d|¬«      \  }}}t        j                  |d|¬«      \  }}}	t        ||«       t        t        j                  |«      t        j                  |«      «       t        t        j                  |	«      t        j                  |«      «       y )NrC   rE   )r'   Úsvdsr   r   Úabs)
r#   r   r!   rG   Úu0Ús0Úvt0ÚuÚsÚvts
             r   Ú	test_svdsr]   u   s|   € Ø$Ñ€GˆX�rä—*‘*˜W¨¨bÔ1�K€BˆˆCÜ�z‰z˜( a¨BÔ/�H€A€qˆ"ä�A�rÔÜ”B—F‘F˜1“IœrŸv™v b›zÔ*Ü”B—F‘F˜2“J¤§¡ s£Õ,r   c                 ó¶   — | \  }}}|d d …d f   }t        j                  ||«      \  }}t        j                  ||«      \  }}t        ||«       t        ||«       y r   )r'   Úlobpcgr   )	r#   r   r!   r+   ÚXrO   rP   rQ   rR   s	            r   Útest_lobpcgra   €   sX   € Ø#Ñ€GˆX�qØ	Š!ˆDˆ&‰	€Aä—|‘| G¨QÓ/Ñ€GˆWÜ�<‰<˜ !Ó$�D€A€qä�A�wÔÜ�A�wÕr   c                 óò  — | \  }}}t         j                  j                  t        |«      d«      }t	        j
                  t        j                  |«      |«      }t	        j
                  ||«      }t        |t         j                  «      sJ ‚t        ||«       t	        j
                  t        j                  |«      |«      }t	        j
                  ||d¬«      }t        |t         j                  «      sJ ‚t        ||«       t	        j
                  t        j                  |«      |«      }t	        j
                  ||«      }t        |t         j                  «      sJ ‚t        ||«       t	        j
                  t        j                  |«      t        j                  |«      «      }t	        j
                  ||«      }t        |t        |«      «      sJ ‚t        |j                  «       |j                  «       «       y )NrD   T)Úuse_umfpack)r   r   r   Úlenr'   ÚspsolveÚspÚ	csc_arrayr)   r*   r   ÚtypeÚtodense)r#   r   r!   r"   Úb2r:   r+   s          r   Útest_spsolverk   ‹   s\  € Ø#Ñ€GˆX�qÜ	�‰�‰œ˜A› Ó	"€Bä	�‰”r—|‘| GÓ,¨aÓ	0€BÜ�‰�h Ó"€AÜ�aœŸ™Ô$Ð$Ð$Ü�A�rÔä	�‰”r—|‘| GÓ,¨aÓ	0€BÜ�‰�h ¨tÔ4€AÜ�aœŸ™Ô$Ð$Ð$Ü�A�rÔä	�‰”r—|‘| GÓ,¨bÓ	1€BÜ�‰�h Ó#€AÜ�aœŸ™Ô$Ð$Ð$Ü�A�rÔä	�‰”r—|‘| GÓ,Ü—|‘| GÓ,ó
.€Bä�‰�h Ó)€AÜ�aœ˜h›Ô(Ð(Ð(Ü�A—I‘I“K §¡£Õ.r   c                 ód  — | \  }}}t        |«      }t        |«      }t        j                  |«      }t	        |j
                  |«      sJ ‚t	        |j                  |«      sJ ‚t        j                  t        j                  |«      |j                  t        j                  |«      ff«      }t        j                  t        j                  |«      t        j                  |«      |j                  ff«      }|j                  |«      }	|j                  |«      }
|	j                  |j
                  z  |j                  z  |
j                  z  }t!        |j#                  «       |j#                  «       «       |j%                  |j#                  «       «      }t!        |t        j&                  |«      d¬«       y )Ng»½×Ùß|Û=r%   )rd   rh   r'   Úsplur)   ÚLÚUrf   rg   r   r;   Úperm_rÚarangeÚperm_cÚfrom_scipy_sparser    r   ri   ÚsolveÚeye)r#   r   r!   r"   Únr   ÚluÚ	_Pr_scipyÚ	_Pc_scipyÚPrÚPcÚA2Úzs                r   Ú	test_splur~   ¥   s5  € Ø#Ñ€GˆX�qÜˆA‹€AÜ�h“€Jä	�‰�HÓ	€Bä�b—d‘d˜JÔ'Ð'Ð'Ü�b—d‘d˜JÔ'Ð'Ð'ä—‘œbŸg™g a›j¨2¯9©9´b·i±iÀ³lÐ*CÐDÓE€IÜ—‘œbŸg™g a›j¬2¯9©9°Q«<¸¿¹Ð*CÐDÓE€IØ	×	%Ñ	% iÓ	0€BØ	×	%Ñ	% iÓ	0€BØ	�‰�—‘‰�r—t‘tÑ	˜bŸd™dÑ	"€Bä�B—J‘J“L (×"2Ñ"2Ó"4Ô5à
�‰�×!Ñ!Ó#Ó$€AÜ�A”r—v‘v˜a“y uÖ-r   c                 óB  — | \  }}}t        |«      }t        j                  |«      }t        |j                  |«      sJ ‚t        |j
                  |«      sJ ‚|j                  |j                  «       «      }t        |t        j                  t        |«      «      d¬«       y )Nr/   r%   )rh   r'   Úspilur)   rn   ro   rt   ri   r   r   ru   rd   )r#   r   r!   r"   r   rw   r}   s          r   Ú
test_spilur�   »   s|   € Ø#Ñ€GˆX�qÜ�h“€Jä	�‰�XÓ	€Bä�b—d‘d˜JÔ'Ð'Ð'Ü�b—d‘d˜JÔ'Ð'Ð'à
�‰�×!Ñ!Ó#Ó$€AÜ�A”r—v‘vœc !›f“~¨DÖ1r   c                 ó„   — | \  }}}t        j                  |«      }t        j                  ||«      }t	        ||z  |«       y r   )r   Útrilr'   Úspsolve_triangularr   )r#   r   r!   r"   r+   s        r   Útest_spsolve_triangularr…   È   s<   € Ø#Ñ€GˆX�qÜ�{‰{˜8Ó$€Hä× Ñ  ¨1Ó-€AÜ�H˜q‘L !Õ$r   c                 ó|   — | \  }}}t        j                  |«      }t        j                  |«      }t        ||«       y r   )r'   Ú
onenormestr   )r#   r   r!   r"   Úest0Úests         r   Útest_onenormestrŠ   Ð   s8   € Ø#Ñ€GˆX�qÜ×Ñ˜GÓ$€DÜ
×
Ñ
˜8Ó
$€CÜ�C˜Õr   c                 ó¢   — | \  }}}t        j                  t        j                  |«      «      }t        j                  |«      }t	        ||«       y r   )r'   Únormrf   Ú	csr_arrayr   )r#   r   r!   r"   Únorm0rŒ   s         r   Ú	test_normr�   ×   s=   € Ø#Ñ€GˆX�qÜ�J‰J”r—|‘| GÓ,Ó-€EÜ�:‰:�hÓ€DÜ�D˜%Õ r   c                 óÚ   — | \  }}}t        j                  t        j                  |«      «      }t        j                  |«      }t	        |j                  «       |j                  «       «       y r   )r'   Úinvrf   rg   r   ri   ©r#   r   r!   r"   r:   r+   s         r   Útest_invr“   Þ   sI   € Ø#Ñ€GˆX�qÜ	�‰”2—<‘< Ó(Ó	)€BÜ�	‰	�(Ó€AÜ�A—I‘I“K §¡£Õ.r   c                 óÚ   — | \  }}}t        j                  t        j                  |«      «      }t        j                  |«      }t	        |j                  «       |j                  «       «       y r   )r'   Úexpmrf   rg   r   ri   r’   s         r   Ú	test_expmr–   å   sI   € Ø#Ñ€GˆX�qÜ	�‰”B—L‘L Ó)Ó	*€BÜ�
‰
�8Ó€AÜ�A—I‘I“K §¡£Õ.r   c                 ó  — | \  }}}t        j                  ||«      }t        j                  ||«      }t        ||«       t        j                  ||«      }t        j                  ||«      }t        |j                  «       |«       y r   )r'   Úexpm_multiplyr   ri   r’   s         r   Útest_expm_multiplyr™   ì   sp   € Ø#Ñ€GˆX�qÜ	×	Ñ	˜W aÓ	(€BÜ×Ñ˜H aÓ(€AÜ�A�rÔä	×	Ñ	˜W gÓ	.€BÜ×Ñ˜H hÓ/€AÜ�A—I‘I“K Õ$r   c                 ó”   — | \  }}t        j                  |«      j                  |j                  «      }||k(  j	                  «       sJ ‚y r   )rf   Ú
coo_matrixÚasformatÚformatÚall©r   Ú	sp_sparseÚ	pd_sparses      r   Útest_eqr¢   ÷   sB   € Ø&Ñ€Iˆyä—‘˜iÓ(×1Ñ1°)×2BÑ2BÓC€IØ˜Ñ"×'Ñ'Ô)Ð)Ñ)r   c                 ó”   — | \  }}t        j                  |«      j                  |j                  «      }||k7  j	                  «       rJ ‚y r   )rf   r›   rœ   r�   ÚanyrŸ   s      r   Útest_ner¥   þ   sE   € Ø&Ñ€Iˆyä—‘˜iÓ(×1Ñ1°)×2BÑ2BÓC€IØ˜YÑ&×+Ñ+Ô-Ð-Ð-Ð-r   )2ÚpytestÚnumpyr   Úscipy.sparser   rf   Úscipy.sparse.linalgÚlinalgr'   Únumpy.testingr   r   Ú	ExceptionÚmarkÚskipifÚ
pytestmarkÚmsgr   ÚxfailÚsparse_paramsÚ	bsr_arrayr�   Ú	coo_arrayrg   Ú	dia_arrayÚ	dok_arrayÚscipy_sparse_classesÚfixturer   r   r   r#   r-   r4   r>   rA   rS   r]   ra   rk   r~   r�   r…   rŠ   r�   r“   r–   r™   r¢   r¥   © r   r   ú<module>rº      s�  ðÛ ã Ý ß #Ð #ç 7ðÛð �[‰[×Ñ ¨$ Ø'Dð  ó F€
ð G€ð �—‘˜eÓ$Ø�—‘˜e¨F¯K©K×,=Ñ,=ÀSÐ,=Ó,IÐ+JÔKðM€ð ‡L�LØ‡L�LØ‡L�LØ‡L�LØ‡L�LØ‡L�LðÐ ð €‡��}Ô%ñ*ó &ð*ð €‡�Ð+Ô,ñó -ðð ‡�ñ7ó ð7ð ‡�ñ ó ð ò%ò0ò ò
0ò$ò4-ò ò/ò4.ò,
2ò%òò!ò/ò/ò%ò*ó.øðg ò ØƒFðús   ¡E ÅE#Å"E#