Ë
    ¤eh¸N  ã                   óþ  — d dl Z d dlZd dlZd dlZd dlmZ d dlmZ	 d dl
mZ  ej                  ddg¬«      d„ «       Z ej                  dd	g¬«      d
„ «       Z G d„ d«      Zej                   j#                  de j$                  e j&                  g«      d„ «       Zd„ Zd„ Zej                   j#                  dej.                  ej0                  g«      ej                   j#                  d eg d¢«       eg d¢«      g«      d„ «       «       Zej                   j#                  d eg d¢«       ej4                  g d¢«      f eg d¢d¬«       ej4                  g d¢«      f eg d¢d¬«       ej4                  g d¢«      f eg d¢d¬«       ej4                  g d¢«      f eg d¢d¬«       ej4                  g d¢«      fg«      ej                   j#                  dej&                  ej6                  g«      d„ «       «       Zd„ Zd„ Zej                   j#                  deej4                  eg«      d„ «       Z ej                   j#                  dg d ¢«      ej                   j#                  d!ejB                  d"g«      d#„ «       «       Z"y)$é    N)ÚSparseDtype)ÚSparseArrayÚintegerÚblock)Úparamsc                 ó   — | j                   S )z!kind kwarg to pass to SparseArray©Úparam©Úrequests    úi/var/www/skyplay_api_hub/venv/lib/python3.12/site-packages/pandas/tests/arrays/sparse/test_arithmetics.pyÚkindr      s   € ð �=‰=Ðó    TFc                 ó   — | j                   S )z}
    Fixture returning True or False, determining whether to operate
    op(sparse, dense) instead of op(sparse, sparse)
    r	   r   s    r   Úmixr      s   € ð �=‰=Ðr   c                   ó¾  — e Zd Zd„ Zdefd„Zd„ Zd„ Zd„ Ze	j                  j                  dg d¢«      e	j                  j                  d	g d
¢«      d„ «       «       Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Zd„ Ze	j                  j                  d	ddej.                  g«      d„ «       Ze	j                  j                  d	ddej.                  g«      d„ «       Zd„ Zd„ Zd„ Zy)ÚTestSparseArrayArithmeticsc                 ó0   — t        j                  ||«       y ©N)ÚtmÚassert_numpy_array_equal)ÚselfÚaÚbs      r   Ú_assertz"TestSparseArrayArithmetics._assert   s   € ä
×#Ñ# A qÕ)r   r   c                 óò  — t        |t        j                  «      r' |t        j                  |«      |«      j
                  }nGt        |t        j                  «      r' ||t        j                  |«      «      j
                  }nt        ‚t        j                  dd¬«      5  |r |||«      j                  «       }n |||«      j                  «       }d d d «       | j                  |«       y # 1 sw Y   ŒxY w)NÚignore)ÚinvalidÚdivide)
Ú
isinstanceÚnpÚndarrayÚpdÚSeriesÚvaluesÚNotImplementedErrorÚerrstateÚto_denser   )	r   r   r   Úa_denseÚb_denser   ÚopÚexpectedÚresults	            r   Ú_check_numeric_opsz-TestSparseArrayArithmetics._check_numeric_ops    sº   € ô �gœrŸz™zÔ*Ùœ"Ÿ)™) GÓ,¨gÓ6×=Ñ=‰HÜ˜¤§¡Ô,Ù˜'¤2§9¡9¨WÓ#5Ó6×=Ñ=‰Hä%Ð%ä�[‰[ °(Ô;ñ 	-ÙÙ˜A˜w›×0Ñ0Ó2‘á˜A˜q›×*Ñ*Ó,�÷		-ð 	�‰�V˜XÕ&÷	-ð 	-ús   Â 2C-Ã-C6c                 óê   — t        |t        «      sJ ‚t        |j                  t        «      sJ ‚|j                  j                  t
        j                  k(  sJ ‚t        |j                  t        «      sJ ‚y r   )	r    r   Údtyper   Úsubtyper!   Úbool_Ú
fill_valueÚbool)r   Úress     r   Ú_check_bool_resultz-TestSparseArrayArithmetics._check_bool_result2   sV   € Ü˜#œ{Ô+Ð+Ð+Ü˜#Ÿ)™)¤[Ô1Ð1Ð1Ø�y‰y× Ñ ¤B§H¡HÒ,Ð,Ð,Ü˜#Ÿ.™.¬$Ô/Ð/Ñ/r   c                 óÊ  — t        j                  d¬«      5  | j                  ||k(  «       | j                  ||k(  j	                  «       ||k(  «       | j                  ||k7  «       | j                  ||k7  j	                  «       ||k7  «       | j                  ||k\  «       | j                  ||k\  j	                  «       ||k\  «       | j                  ||k  «       | j                  ||k  j	                  «       ||k  «       | j                  ||kD  «       | j                  ||kD  j	                  «       ||kD  «       | j                  ||k  «       | j                  ||k  j	                  «       ||k  «       | j                  ||k(  «       | j                  ||k(  j	                  «       ||k(  «       | j                  ||k7  «       | j                  ||k7  j	                  «       ||k7  «       | j                  ||k\  «       | j                  ||k\  j	                  «       ||k\  «       | j                  ||k  «       | j                  ||k  j	                  «       ||k  «       | j                  ||kD  «       | j                  ||kD  j	                  «       ||kD  «       | j                  ||k  «       | j                  ||k  j	                  «       ||k  «       d d d «       y # 1 sw Y   y xY w)Nr   )r   )r!   r'   r6   r   r(   ©r   r   r   r)   r*   s        r   Ú_check_comparison_opsz0TestSparseArrayArithmetics._check_comparison_ops8   s„  € Ü�[‰[ Ô*ñ (	Fð
 ×#Ñ# A¨¡FÔ+Ø�L‰L˜!˜q™&×*Ñ*Ó,¨g¸Ñ.@ÔAà×#Ñ# A¨¡FÔ+Ø�L‰L˜!˜q™&×*Ñ*Ó,¨g¸Ñ.@ÔAà×#Ñ# A¨¡FÔ+Ø�L‰L˜!˜q™&×*Ñ*Ó,¨g¸Ñ.@ÔAà×#Ñ# A¨¡FÔ+Ø�L‰L˜!˜q™&×*Ñ*Ó,¨g¸Ñ.@ÔAà×#Ñ# A¨¡EÔ*Ø�L‰L˜!˜a™%×)Ñ)Ó+¨W°wÑ->Ô?à×#Ñ# A¨¡EÔ*Ø�L‰L˜!˜a™%×)Ñ)Ó+¨W°wÑ->Ô?ð ×#Ñ# A¨¡LÔ1Ø�L‰L˜!˜w™,×0Ñ0Ó2°G¸wÑ4FÔGà×#Ñ# A¨¡LÔ1Ø�L‰L˜!˜w™,×0Ñ0Ó2°G¸wÑ4FÔGà×#Ñ# A¨¡LÔ1Ø�L‰L˜!˜w™,×0Ñ0Ó2°G¸wÑ4FÔGà×#Ñ# A¨¡LÔ1Ø�L‰L˜!˜w™,×0Ñ0Ó2°G¸wÑ4FÔGà×#Ñ# A¨¡KÔ0Ø�L‰L˜!˜g™+×/Ñ/Ó1°7¸WÑ3DÔEà×#Ñ# A¨¡KÔ0Ø�L‰L˜!˜g™+×/Ñ/Ó1°7¸WÑ3DÔE÷Q(	F÷ (	Fñ (	Fús   —J9KËK"c                 óÔ  — | j                  ||z  «       | j                  ||z  j                  «       ||z  «       | j                  ||z  «       | j                  ||z  j                  «       ||z  «       | j                  ||z  «       | j                  ||z  j                  «       ||z  «       | j                  ||z  «       | j                  ||z  j                  «       ||z  «       y r   )r6   r   r(   r8   s        r   Ú_check_logical_opsz-TestSparseArrayArithmetics._check_logical_opsc   sÊ   € à×Ñ  A¡Ô&Ø�‰�a˜!‘e×%Ñ%Ó'¨°7Ñ):Ô;à×Ñ  A¡Ô&Ø�‰�a˜!‘e×%Ñ%Ó'¨°7Ñ):Ô;à×Ñ  G¡Ô,Ø�‰�a˜'‘k×+Ñ+Ó-¨w¸Ñ/@ÔAà×Ñ  G¡Ô,Ø�‰�a˜'‘k×+Ñ+Ó-¨w¸Ñ/@ÕAr   Úscalar)r   é   é   r3   )Nr   é   c                 óâ   — |}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        |||¬«      }	| j	                  |	|||||«       y )Nr=   r?   r   ©r   r3   ©r!   ÚarrayÚnanr   r.   )
r   r   r   Úall_arithmetic_functionsr3   r<   r   r+   r%   r   s
             r   Útest_float_scalarz,TestSparseArrayArithmetics.test_float_scalarq   s^   € ð
 &ˆÜ—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ˜ T°jÔAˆØ×Ñ  6¨6°6¸3ÀÕCr   c                 óP  — t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        ||¬«      }| j	                  |d|d«       | j	                  |d|d«       | j	                  |d|d«       t        ||d¬«      }| j	                  |d|d«       | j	                  |d|d«       | j	                  |d|d«       t        ||d¬«      }| j	                  |d|d«       | j	                  |d|d«       | j	                  |d|d«       y )Nr=   r?   r   ©r   r>   rA   ©r!   rC   rD   r   r9   )r   r   r%   r   s       r   Útest_float_scalar_comparisonz7TestSparseArrayArithmetics.test_float_scalar_comparison{   s  € Ü—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆä˜ TÔ*ˆØ×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Ô3ä˜ T°aÔ8ˆØ×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Ô3ä˜ T°aÔ8ˆØ×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Ô3Ø×"Ñ" 1 a¨°Õ3r   c                 óÈ   — |}t        j                  g d¢«      }t        j                  g d¢«      }t        ||d¬«      }t        ||d¬«      }| j                  ||||||«       y )N©
ç        ç      ð?ç       @g      @rM   rM   rN   rO   rN   rM   ©
rM   rO   ç      @g      @rM   rM   rN   rQ   rO   rM   r   rA   )r!   rC   r   r.   ©	r   r   r   rE   r+   r%   Úrvaluesr   r   s	            r   Ú"test_float_same_index_without_nansz=TestSparseArrayArithmetics.test_float_same_index_without_nans�   sY   € à%ˆä—‘ÒLÓMˆÜ—(‘(ÒMÓNˆä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×Ñ  1 f¨g°s¸BÕ?r   c                 óŒ  — |}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        ||¬«      }t        ||¬«      }	| j	                  ||	||||«       y )Nr=   r?   r   r>   é   rH   rB   )
r   r   r   rE   r   r+   r%   rS   r   r   s
             r   Útest_float_same_index_with_nansz:TestSparseArrayArithmetics.test_float_same_index_with_nans˜   s—   € ð &ˆÜ—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(œBŸF™F A q¨!¬R¯V©V°Q¸¸1¸aÄÇÁÐHÓIˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ×Ñ  1 f¨g°s¸BÕ?r   c                 ó@  — t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        ||¬«      }t        ||¬«      }| j	                  ||||«       t        j                  g d¢«      }t        j                  g d¢«      }t        ||d¬	«      }t        ||d¬	«      }| j	                  ||||«       y )
Nr=   r?   r   r>   rV   rH   rL   rP   rA   rI   ©r   r   r%   rS   r   r   s         r   Ú test_float_same_index_comparisonz;TestSparseArrayArithmetics.test_float_same_index_comparison¤   sÚ   € ä—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(œBŸF™F A q¨!¬R¯V©V°Q¸¸1¸aÄÇÁÐHÓIˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ×"Ñ" 1 a¨°Ô9ä—‘ÒLÓMˆÜ—(‘(ÒMÓNˆä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×"Ñ" 1 a¨°Õ9r   c                 óî  — |}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  dt         j                  ddt         j                  ddddt         j                  g
«      }t        ||¬«      }t        ||¬«      }| j	                  ||||||«       | j	                  ||dz  ||dz  ||«       t        ||d¬«      }t        ||¬«      }| j	                  ||||||«       t        ||d¬«      }t        ||d¬«      }| j	                  ||||||«       t        ||d¬«      }t        ||d¬«      }| j	                  ||||||«       y ©Nr=   r?   r   r>   é   rH   rA   rB   rR   s	            r   Útest_float_arrayz+TestSparseArrayArithmetics.test_float_array´   sO  € Ø%ˆä—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(˜AœrŸv™v q¨!¬R¯V©V°Q¸¸1¸aÄÇÁÐHÓIˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ×Ñ  1 f¨g°s¸BÔ?Ø×Ñ  1 q¡5¨&°'¸A±+¸sÀBÔGä˜ T°aÔ8ˆÜ˜ dÔ+ˆØ×Ñ  1 f¨g°s¸BÔ?ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×Ñ  1 f¨g°s¸BÔ?ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×Ñ  1 f¨g°s¸BÕ?r   c                 óî  — |}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  dt         j                  ddt         j                  ddddt         j                  g
«      }t        |d¬«      }t        |d¬«      }| j	                  ||||||«       | j	                  ||dz  ||dz  ||«       t        |dd¬	«      }t        |d¬«      }| j	                  ||||||«       t        |dd¬	«      }t        |dd¬	«      }| j	                  ||||||«       t        |dd¬	«      }t        |dd¬	«      }| j	                  ||||||«       y )
Nr=   r?   r   r>   r]   r   rH   r   rA   rB   )r   r   rE   r+   r%   rS   r   r   s           r   Útest_float_array_different_kindz:TestSparseArrayArithmetics.test_float_array_different_kindË   sO  € Ø%ˆä—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(˜AœrŸv™v q¨!¬R¯V©V°Q¸¸1¸aÄÇÁÐHÓIˆä˜ YÔ/ˆÜ˜ gÔ.ˆØ×Ñ  1 f¨g°s¸BÔ?Ø×Ñ  1 q¡5¨&°'¸A±+¸sÀBÔGä˜ Y¸1Ô=ˆÜ˜ gÔ.ˆØ×Ñ  1 f¨g°s¸BÔ?ä˜ Y¸1Ô=ˆÜ˜ g¸!Ô<ˆØ×Ñ  1 f¨g°s¸BÔ?ä˜ Y¸1Ô=ˆÜ˜ g¸!Ô<ˆØ×Ñ  1 f¨g°s¸BÕ?r   c                 óÖ  — t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  dt         j                  ddt         j                  ddddt         j                  g
«      }t        ||¬«      }t        ||¬«      }| j	                  ||||«       | j	                  ||dz  ||dz  «       t        ||d¬«      }t        ||¬«      }| j	                  ||||«       t        ||d¬«      }t        ||d¬«      }| j	                  ||||«       t        ||d¬«      }t        ||d¬«      }| j	                  ||||«       y r\   rI   rY   s         r   Útest_float_array_comparisonz6TestSparseArrayArithmetics.test_float_array_comparisonâ   s6  € Ü—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(˜AœrŸv™v q¨!¬R¯V©V°Q¸¸1¸aÄÇÁÐHÓIˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ×"Ñ" 1 a¨°Ô9Ø×"Ñ" 1 a¨!¡e¨V°W¸q±[ÔAä˜ T°aÔ8ˆÜ˜ dÔ+ˆØ×"Ñ" 1 a¨°Ô9ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×"Ñ" 1 a¨°Ô9ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ×"Ñ" 1 a¨°Õ9r   c                 ó  — |}t         j                  }t        j                  g d¢|¬«      }t        j                  g d¢|¬«      }t        |||¬«      }|j                  t        |«      k(  sJ ‚t        |||¬«      }	|	j                  t        |«      k(  sJ ‚| j                  ||	||||«       | j                  ||	dz  ||dz  ||«       t        |d||¬«      }|j                  t        |«      k(  sJ ‚t        |||¬«      }	|	j                  t        |«      k(  sJ ‚| j                  ||	||||«       t        |d||¬«      }|j                  t        |«      k(  sJ ‚t        |d||¬«      }	|	j                  t        |«      k(  sJ ‚| j                  ||	||||«       t        |d||¬«      }|j                  t        |d¬«      k(  sJ ‚t        |d	||¬«      }	|	j                  t        |d	¬«      k(  sJ ‚| j                  ||	||||«       y )
N©
r   r=   r?   r   r   r   r=   r?   r=   r   ©r0   ©
r?   r   r?   r>   r   r   r=   r]   r?   r   ©r0   r   r   )r3   r0   r   r=   ©r3   r?   )r!   Úint64rC   r   r0   r   r.   )
r   r   r   rE   r+   r0   r%   rS   r   r   s
             r   Útest_int_arrayz)TestSparseArrayArithmetics.test_int_array÷   sê  € Ø%ˆô —‘ˆä—‘Ò8ÀÔFˆÜ—(‘(Ò9ÀÔGˆä˜ e°$Ô7ˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,Ü˜ u°4Ô8ˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,à×Ñ  1 f¨g°s¸BÔ?Ø×Ñ  1 q¡5¨&°'¸A±+¸sÀBÔGä˜¨1°EÀÔEˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,Ü˜ u°4Ô8ˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,à×Ñ  1 f¨g°s¸BÔ?ä˜¨1°EÀÔEˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,Ü˜¨A°UÀÔFˆØ�w‰wœ+ eÓ,Ò,Ð,Ð,Ø×Ñ  1 f¨g°s¸BÔ?ä˜¨1°EÀÔEˆØ�w‰wœ+ e¸Ô:Ò:Ð:Ð:Ü˜¨A°UÀÔFˆØ�w‰wœ+ e¸Ô:Ò:Ð:Ð:Ø×Ñ  1 f¨g°s¸BÕ?r   c                 ó*  — d}t        j                  g d¢|¬«      }t        j                  g d¢|¬«      }t        |||¬«      }t        |||¬«      }| j                  ||||«       | j                  ||dz  ||dz  «       t        |||d¬«      }t        |||¬«      }| j                  ||||«       t        |||d¬«      }t        |||d¬«      }| j                  ||||«       t        |||d¬«      }t        |||d	¬«      }| j                  ||||«       y )
Nri   rd   re   rf   rg   r   )r0   r   r3   r=   r?   )r!   rC   r   r9   )r   r   r0   r%   rS   r   r   s          r   Útest_int_array_comparisonz4TestSparseArrayArithmetics.test_int_array_comparison  s  € Øˆô —‘Ò8ÀÔFˆÜ—(‘(Ò9ÀÔGˆä˜ e°$Ô7ˆÜ˜ u°4Ô8ˆØ×"Ñ" 1 a¨°Ô9Ø×"Ñ" 1 a¨!¡e¨V°W¸q±[ÔAä˜ e°$À1ÔEˆÜ˜ u°4Ô8ˆØ×"Ñ" 1 a¨°Ô9ä˜ e°$À1ÔEˆÜ˜ u°4ÀAÔFˆØ×"Ñ" 1 a¨°Ô9ä˜ e°$À1ÔEˆÜ˜ u°4ÀAÔFˆØ×"Ñ" 1 a¨°Õ9r   TFc                 ó<  — t        j                  g d¢t         j                  ¬«      }t        j                  g d¢t         j                  ¬«      }t        ||t         j                  |¬«      }t        ||t         j                  |¬«      }| j	                  ||||«       y )N)TFTTre   ©r   r0   r3   ©r!   rC   r2   r   r;   ©r   r   r3   r%   rS   r   r   s          r   Útest_bool_same_indexz/TestSparseArrayArithmetics.test_bool_same_index3  sj   € ô —‘Ò3¼2¿8¹8ÔDˆÜ—(‘(Ò4¼B¿H¹HÔEˆä˜ T´·±ÀjÔQˆÜ˜ d´"·(±(ÀzÔRˆØ×Ñ  1 f¨gÕ6r   c                 ó<  — t        j                  g d¢t         j                  ¬«      }t        j                  g d¢t         j                  ¬«      }t        ||t         j                  |¬«      }t        ||t         j                  |¬«      }| j	                  ||||«       y )N)TFTFTTre   )TFFTFTrn   ro   rp   s          r   Útest_bool_array_logicalz2TestSparseArrayArithmetics.test_bool_array_logical>  sj   € ô —‘Ò@ÌÏÉÔQˆÜ—(‘(ÒBÌ"Ï(É(ÔSˆä˜ T´·±ÀjÔQˆÜ˜ d´"·(±(ÀzÔRˆØ×Ñ  1 f¨gÕ6r   c                 óf  — |}d}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  g d¢|¬«      }t        ||¬«      }	t        ||¬«      }
|
j                  t        |«      k(  sJ ‚| j                  |	|
||||«       | j                  |	|
dz  ||dz  ||«       t        ||d¬«      }	t        ||¬«      }
|
j                  t        |«      k(  sJ ‚| j                  |	|
||||«       t        ||d¬«      }	t        ||d¬«      }
|
j                  t        |«      k(  sJ ‚| j                  |	|
||||«       t        ||d¬«      }	t        ||d¬«      }
|
j                  t        |d¬	«      k(  sJ ‚| j                  |	|
||||«       y ©
Nri   r=   r?   r   rf   re   rH   rA   rh   )r!   rC   rD   r   r0   r   r.   )r   r   r   rE   r   r+   Úrdtyper%   rS   r   r   s              r   Útest_mixed_array_float_intz5TestSparseArrayArithmetics.test_mixed_array_float_intI  s”  € Ø%ˆØˆÜ—‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(Ò9ÀÔHˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-à×Ñ  1 f¨g°s¸BÔ?Ø×Ñ  1 q¡5¨&°'¸A±+¸sÀBÔGä˜ T°aÔ8ˆÜ˜ dÔ+ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-Ø×Ñ  1 f¨g°s¸BÔ?ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-Ø×Ñ  1 f¨g°s¸BÔ?ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ�w‰wœ+ f¸Ô;Ò;Ð;Ð;Ø×Ñ  1 f¨g°s¸BÕ?r   c                 óN  — d}t        j                  t         j                  dddt         j                  ddddt         j                  g
«      }t        j                  g d¢|¬«      }t        ||¬«      }t        ||¬«      }|j                  t        |«      k(  sJ ‚| j                  ||||«       | j                  ||dz  ||dz  «       t        ||d¬«      }t        ||¬«      }|j                  t        |«      k(  sJ ‚| j                  ||||«       t        ||d¬«      }t        ||d¬«      }|j                  t        |«      k(  sJ ‚| j                  ||||«       t        ||d¬«      }t        ||d¬«      }|j                  t        |d¬	«      k(  sJ ‚| j                  ||||«       y ru   )r!   rC   rD   r   r0   r   r9   )r   r   rv   r%   rS   r   r   s          r   Útest_mixed_array_comparisonz6TestSparseArrayArithmetics.test_mixed_array_comparisone  s}  € Øˆô —‘œ2Ÿ6™6 1 a¨¬B¯F©F°A°q¸!¸QÄÇÁÐGÓHˆÜ—(‘(Ò9ÀÔHˆä˜ TÔ*ˆÜ˜ dÔ+ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-à×"Ñ" 1 a¨°Ô9Ø×"Ñ" 1 a¨!¡e¨V°W¸q±[ÔAä˜ T°aÔ8ˆÜ˜ dÔ+ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-Ø×"Ñ" 1 a¨°Ô9ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ�w‰wœ+ fÓ-Ò-Ð-Ð-Ø×"Ñ" 1 a¨°Ô9ä˜ T°aÔ8ˆÜ˜ d°qÔ9ˆØ�w‰wœ+ f¸Ô;Ò;Ð;Ð;Ø×"Ñ" 1 a¨°Õ9r   c                 ó"  — t        g d¢«      }t        g d¢«      }||z  }t        j                  j                  j                  j                  dt        j                  g d¢d¬«      «      }t        g d¢|¬«      }t        j                  ||«       y )	N)TTFF)TFTFrV   ©r   r=   r?   Úint32re   )FTT)Úsparse_index)
r   r#   ÚcoreÚarraysÚsparseÚIntIndexr!   rC   r   Úassert_sp_array_equal)r   ÚsÚtr-   Úsp_indexr,   s         r   Útest_xorz#TestSparseArrayArithmetics.test_xor‚  si   € ÜÒ2Ó3ˆÜÒ2Ó3ˆØ�Q‘ˆÜ—7‘7—>‘>×(Ñ(×1Ñ1°!´R·X±XºiÈwÔ5WÓXˆÜÒ2ÀÔJˆÜ
× Ñ  ¨Õ2r   N)Ú__name__Ú
__module__Ú__qualname__r   r4   r.   r6   r9   r;   ÚpytestÚmarkÚparametrizerF   rJ   rT   rW   rZ   r^   r`   rb   rj   rl   r!   rD   rq   rs   rw   ry   r†   © r   r   r   r      s
  „ ò*ð'¸dó 'ò$0ò)FòVBð ‡[�[×Ñ˜X¢yÓ1Ø‡[�[×Ñ˜\ª<Ó8ñDó 9ó 2ðDò4ò$	@ò
@ò:ò @ò.@ò.:ò*"@òH:ð0 ‡[�[×Ñ˜\¨D°%¸¿¹Ð+@ÓAñ7ó Bð7ð ‡[�[×Ñ˜\¨D°%¸¿¹Ð+@ÓAñ7ó Bð7ò@ò8:ó:3r   r   r+   c                 óŒ   — t        ddgd¬«      } | |ddg«      } | |t        ddg«      «      }t        j                  ||«       y )Nr   r=   rh   )r   r   r‚   )r+   Úarrr-   r,   s       r   Útest_with_listr�   ‹  sF   € ä
�q˜!�f¨Ô
+€CÙ��a˜�V‹_€FÙ�#”{ A q 6Ó*Ó+€HÜ×Ñ˜V XÕ.r   c                  óŽ   — t        ddgd¬«      } t        j                  ddgddgg«      }| j                  |«      }|t        u sJ ‚y )Nr   r=   rh   r?   r>   rV   )r   r#   Ú	DataFrameÚ__add__ÚNotImplemented)r�   Údfr-   s      r   Útest_with_dataframer–   “  sH   € ä
�q˜!�f¨Ô
+€CÜ	�‰˜˜1�v  1˜vÐ&Ó	'€BØ�[‰[˜‹_€FØ”^Ñ#Ð#Ñ#r   c                  óˆ   — t        ddgd¬«      } | t        j                  d«      z  }| dz  }t        j                  ||«       y )Nr   r=   rh   r?   ©r   r!   rC   r   r‚   )r�   r-   r,   s      r   Útest_with_zerodim_ndarrayr™   ›  s=   € ä
�q˜!�f¨Ô
+€Cà”2—8‘8˜A“;Ñ€FØ�Q‰w€HÜ×Ñ˜V XÕ.r   Úufuncr�   )r   r   éÿÿÿÿr=   )NNr›   r=   c                 ó°   —  | |«      } | |j                   «      }t         | t        j                  |«      «      |¬«      }t	        j
                  ||«       y )Nrh   )r3   r   r!   Úasarrayr   r‚   )rš   r�   r-   r3   r,   s        r   Útest_ufuncsrž   ¤  sD   € ñ
 �3‹Z€FÙ�s—~‘~Ó&€JÜ™5¤§¡¨C£Ó1¸jÔI€HÜ×Ñ˜V XÕ.r   za, b)r   r   r   r{   r=   rh   c                 óê   —  | ||«      } | t        j                  |«      t        j                  |«      «      }t        |t        «      sJ ‚t	        j
                  t        j                  |«      |«       y r   )r!   r�   r    r   r   r   )rš   r   r   r-   r,   s        r   Útest_binary_ufuncsr    ¯  sT   € ñ �1�a‹[€FÙ”R—Z‘Z “]¤B§J¡J¨q£MÓ2€HÜ�fœkÔ*Ð*Ð*Ü×Ñ¤§
¡
¨6Ó 2°HÕ=r   c                  ó°   — t        g d¢«      } t        j                  g d¢«      }|| z  }t        j                  g d¢«      }t        j                  ||«       y )N©r   r?   r   r   ©r   r=   r?   r>   ©r   r>   r?   r>   )r   r!   rC   r   r   ©Úsparrayr"   r,   s      r   Útest_ndarray_inplacer§   Â  sB   € Üš,Ó'€GÜ�h‰h’|Ó$€GØˆwÑ€GÜ�x‰xšÓ%€HÜ×Ñ ¨Õ2r   c                  ó    — t        g d¢«      } t        j                  g d¢«      }| |z  } t        g d¢d¬«      }t        j                  | |«       y )Nr¢   r£   r¤   r   rh   r˜   r¥   s      r   Útest_sparray_inplacer©   Ê  s@   € Üš,Ó'€GÜ�h‰h’|Ó$€GØˆwÑ€GÜš<°AÔ6€HÜ×Ñ˜W hÕ/r   Úconsc                 óœ   — t        ddg«      } | g d¢«      }t        j                  t        d¬«      5  ||z   d d d «       y # 1 sw Y   y xY w)NT)TTTzoperands have mismatched length)Úmatch)r   rŠ   ÚraisesÚ
ValueError)rª   ÚleftÚrights      r   Útest_mismatched_length_cmp_opr±   Ò  sF   € ä˜˜d�|Ó$€DÙÒ#Ó$€EÜ	�‰”zÐ)JÔ	Kñ ØˆuŠ÷÷ ñ ús   ³AÁA)ÚaddÚsubÚmulÚtruedivÚfloordivÚpowr3   r>   c                 óÂ  — t        t        | «      } t        j                  j	                  d«      j                  d«      }t        j                  j	                  d«      j                  d«      }||d d d…<   ||d d d…<   t        ||¬«      }t        ||¬«      }t        j                  d¬«      5   | ||«      }t         | |j                  «       |j                  «       «      |j                  ¬«      }t        |t        «      sJ ‚t        j                  |j                  «       |j                  «       «        | ||j                  «       «      }t        |t        «      sJ ‚t        j                  ||«        | |j                  «       |«      }	t        |	t        «      sJ ‚t        j                  ||	«        | |d«      }
t        |
t        «      sJ ‚	  | |j                  «       d«      } | |j                  d«      }t        j                  |
j                  |«       t        j                  |
j                  «       |«       d d d «       y # t        $ r Y Œw xY w# 1 sw Y   y xY w)Nr?   é   r>   rh   r   )ÚallrV   )ÚgetattrÚoperatorr!   ÚrandomÚdefault_rngÚstandard_normalr   r'   r(   r3   r    r   Úassert_almost_equalr‚   r®   )r+   r3   Údata1Údata2ÚfirstÚsecondr5   ÚexpÚres2Úres3Úres4Úexp_fvs               r   Útest_binary_operatorsrÊ   Ú  sõ  € ô 
”˜2Ó	€BÜ�I‰I×!Ñ! !Ó$×4Ñ4°RÓ8€EÜ�I‰I×!Ñ! !Ó$×4Ñ4°RÓ8€Eà€E‰#ˆAˆ#�JØ€E‰#ˆAˆ#�Jä˜¨*Ô5€EÜ˜¨:Ô6€Fä	�‰˜Ô	"ñ 9Ù�˜ÓˆÜÙˆu�~‰~Ó §¡Ó!2Ó3À×@PÑ@Pô
ˆô ˜#œ{Ô+Ð+Ð+Ü
×Ñ˜sŸ|™|›~¨s¯|©|«~Ô>á�%˜Ÿ™Ó*Ó+ˆÜ˜$¤Ô,Ð,Ð,Ü
× Ñ   dÔ+á�%—.‘.Ó" FÓ+ˆÜ˜$¤Ô,Ð,Ð,Ü
× Ñ   dÔ+á�%˜‹|ˆÜ˜$¤Ô,Ð,Ð,ð	9Ù�U—^‘^Ó% qÓ)ˆCÙ˜×(Ñ(¨!Ó,ˆFô ×"Ñ" 4§?¡?°FÔ;Ü×"Ñ" 4§=¡=£?°CÔ8÷79ð 9øô. ò 	Ùð	ú÷/9ð 9ús2   Â-D!IÇ*IÇ9AIÉ	IÉIÉIÉIÉI)#r¼   Únumpyr!   rŠ   Úpandasr#   r   Úpandas._testingÚ_testingr   Úpandas.core.arrays.sparser   Úfixturer   r   r   r‹   rŒ   Úeqr²   r�   r–   r™   ÚabsrÅ   rž   rC   Úgreaterr    r§   r©   Úlistr±   rD   rÊ   r�   r   r   ú<module>rÕ      s^  ðÛ ã Û ã Ý Ý Ý 1ð €‡�˜	 7Ð+Ô,ñó -ðð
 €‡�˜˜e�}Ô%ñó &ð÷m3ñ m3ð` ‡�×Ñ˜ §¡¨X¯\©\Ð:Ó;ñ/ó <ð/ò$ò/ð ‡�×Ñ˜ 2§6¡6¨2¯6©6Ð"2Ó3Ø‡�×ÑØ	‰KšÓ&©Ò4GÓ(HÐIóñ/óó 4ð/ð ‡�×ÑØ
á	’YÓ	  §¡ª)Ó!4Ð5Ù	’Y¨1Ô	-¨x¨r¯x©xº	Ó/BÐCÙ	’Y¨1Ô	-¨x¨r¯x©xº	Ó/BÐCÙ	’Y¨1Ô	-¨x¨r¯x©xº	Ó/BÐCÙ	’Y¨1Ô	-¨x¨r¯x©xº	Ó/BÐCðó	ð ‡�×Ñ˜ 2§6¡6¨2¯:©:Ð"6Ó7ñ>ó 8ó	ð>ò3ò0ð ‡�×Ñ˜ $¨¯©°+Ð!>Ó?ñó @ðð ‡�×Ñ˜ÒRÓSØ‡�×Ñ˜¨¯©° {Ó3ñ&9ó 4ó Tñ&9r   