§
    tŠtj#  ã                   óZ  — d dl Z d dlmZmZmZ d dlmZ e j         	                    dddg¦  «        d„ ¦   «         Z
d„ Zd„ Zd	„ Zd
„ Ze j         	                    dddg¦  «        d„ ¦   «         Zd„ Zd„ Ze j         	                    dg d¢g d¢fg d¢g d¢fg¦  «        d„ ¦   «         Zd„ Zd„ ZdS )é    N)Ú	DataFrameÚIndexÚSerieszn, frac)é   N)Ngš™™™™™É?c                 óÆ  — dgdz  dgdz  z   }t          ||dœ¦  «        }|                     d¦  «                             | |¬¦  «        }dgdz  dgdz  z   }t          ||dœ|j        ¬¦  «        }t	          j        ||¦  «         |                     d¦  «        d                              | |¬¦  «        }t          |d|j        ¬	¦  «        }t	          j        ||¦  «         d S ©
Né   é
   r   ©ÚaÚbr   ©ÚnÚfrac©Úindexr   ©Únamer   ©r   ÚgroupbyÚsampler   ÚtmÚassert_frame_equalr   Úassert_series_equal)r   r   ÚvaluesÚdfÚresultÚexpecteds         úf/var/www/html/CA-Chatbot/venv/lib/python3.11/site-packages/pandas/tests/groupby/methods/test_sample.pyÚ)test_groupby_sample_balanced_groups_shaper       sê   € àˆS�2‰X˜˜˜b™Ñ €FÝ	˜ fÐ-Ð-Ñ	.Ô	.€Bà�ZŠZ˜‰_Œ_×#Ò# a¨dÐ#Ñ3Ô3€FØˆS�1‰W˜�s˜Q‘wÑ€FÝ˜v¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨1°4Ð(Ñ8Ô8€FÝ�f 3¨f¬lÐ;Ñ;Ô;€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,ó    c                  óÂ  — dgdz  dgdz  z   } t          | | dœ¦  «        }|                     d¦  «                             d¬¦  «        }dgdz  dgdz  z   } t          | | dœ|j        ¬	¦  «        }t	          j        ||¦  «         |                     d¦  «        d
                              d¬¦  «        }t          | d
|j        ¬¦  «        }t	          j        ||¦  «         d S )Nr	   r
   r   é   r   r   é   ©r   r   r   r   r   ©r   r   r   r   s       r   Ú+test_groupby_sample_unbalanced_groups_shaper'      sæ   € ØˆS�2‰X˜˜˜b™Ñ €FÝ	˜ fÐ-Ð-Ñ	.Ô	.€Bà�ZŠZ˜‰_Œ_×#Ò# aÐ#Ñ(Ô(€FØˆS�1‰W˜�s˜Q‘wÑ€FÝ˜v¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨1Ð(Ñ-Ô-€FÝ�f 3¨f¬lÐ;Ñ;Ô;€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,r!   c                  óÊ  — dgdz  dgdz  z   } t          | | dœg d¢¬¦  «        }|                     d¦  «                             d¬¦  «        }dgdz  dgdz  z   } t          | | dœ|j        ¬¦  «        }t	          j        ||¦  «         |                     d¦  «        d	                              d¬¦  «        }t          | d	|j        ¬
¦  «        }t	          j        ||¦  «         d S )Nr	   é   r   r   )r	   r   r   r   r   r   r   r   r%   r   r   r   r&   s       r   Ú,test_groupby_sample_index_value_spans_groupsr*   (   sò   € ØˆS�1‰W˜�s˜Q‘wÑ€FÝ	˜ fÐ-Ð-Ð5GÐ5GÐ5GÐ	HÑ	HÔ	H€Bà�ZŠZ˜‰_Œ_×#Ò# aÐ#Ñ(Ô(€FØˆS�1‰W˜�s˜Q‘wÑ€FÝ˜v¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨1Ð(Ñ-Ô-€FÝ�f 3¨f¬lÐ;Ñ;Ô;€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,r!   c                  ó¸  — t          ddgddgdœ¦  «        } d}t          j        t          |¬¦  «        5  |                      d¦  «                             dd¬¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          |¬¦  «        5  |                      d¦  «        d	                              dd¬¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )
Nr	   r   r   z0Please enter a value for `frac` OR `n`, not both©Úmatchr   g      ð?r   r   ©r   ÚpytestÚraisesÚ
ValueErrorr   r   ©r   Úmsgs     r   Ú%test_groupby_sample_n_and_frac_raisesr4   6   sT  € Ý	˜!˜Q˜ q¨! fÐ-Ð-Ñ	.Ô	.€BØ
<€Cå	Œ•z¨Ð	-Ñ	-Ô	-ð .ð .Ø
�
Š
�3‰Œ×Ò ¨ÐÑ-Ô-Ð-ð.ð .ð .ñ .ô .ð .ð .ð .ð .ð .ð .øøøð .ð .ð .ð .õ 
Œ•z¨Ð	-Ñ	-Ô	-ð 3ð 3Ø
�
Š
�3‰Œ˜Ô×#Ò# a¨cÐ#Ñ2Ô2Ð2ð3ð 3ð 3ñ 3ô 3ð 3ð 3ð 3ð 3ð 3ð 3ð 3øøøð 3ð 3ð 3ð 3ð 3ð 3ó#   ´+A+Á+A/Á2A/Â1CÃCÃCc                  ó¸  — t          ddgddgdœ¦  «        } d}t          j        t          |¬¦  «        5  |                      d¦  «                             dd¬	¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          |¬¦  «        5  |                      d¦  «        d
                              dd¬	¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr	   r   r   zJReplace has to be set to `True` when upsampling the population `frac` > 1.r,   r   ç      ø?F©r   Úreplacer   r.   r2   s     r   Ú:test_groupby_sample_frac_gt_one_without_replacement_raisesr:   A   sT  € Ý	˜!˜Q˜ q¨! fÐ-Ð-Ñ	.Ô	.€BØ
V€Cå	Œ•z¨Ð	-Ñ	-Ô	-ð 8ð 8Ø
�
Š
�3‰Œ×Ò C°ÐÑ7Ô7Ð7ð8ð 8ð 8ñ 8ô 8ð 8ð 8ð 8ð 8ð 8ð 8øøøð 8ð 8ð 8ð 8õ 
Œ•z¨Ð	-Ñ	-Ô	-ð =ð =Ø
�
Š
�3‰Œ˜Ô×#Ò#¨°eÐ#Ñ<Ô<Ð<ð=ð =ð =ñ =ô =ð =ð =ð =ð =ð =ð =ð =øøøð =ð =ð =ð =ð =ð =r5   r   éÿÿÿÿr7   c                 óÆ  — t          ddgddgdœ¦  «        }| dk     rd}nd}t          j        t          |¬¦  «        5  |                     d¦  «                             | ¬	¦  «         d d d ¦  «         n# 1 swxY w Y   t          j        t          |¬¦  «        5  |                     d¦  «        d
                              | ¬	¦  «         d d d ¦  «         d S # 1 swxY w Y   d S )Nr	   r   r   r   z=A negative number of rows requested. Please provide `n` >= 0.z$Only integers accepted as `n` valuesr,   r   r%   r   r.   )r   r   r3   s      r   Ú$test_groupby_sample_invalid_n_raisesr=   L   s`  € å	˜!˜Q˜ q¨! fÐ-Ð-Ñ	.Ô	.€Bàˆ1‚u€uØMˆˆà4ˆå	Œ•z¨Ð	-Ñ	-Ô	-ð $ð $Ø
�
Š
�3‰Œ×Ò ÐÑ#Ô#Ð#ð$ð $ð $ñ $ô $ð $ð $ð $ð $ð $ð $øøøð $ð $ð $ð $õ 
Œ•z¨Ð	-Ñ	-Ô	-ð )ð )Ø
�
Š
�3‰Œ˜Ô×#Ò# aÐ#Ñ(Ô(Ð(ð)ð )ð )ñ )ô )ð )ð )ð )ð )ð )ð )ð )øøøð )ð )ð )ð )ð )ð )s#   ½*A3Á3A7Á:A7Â0CÃCÃCc                  óÆ  — dgdz  dgdz  z   } t          | | dœ¦  «        }|                     d¦  «                             dd¬¦  «        }dgd	z  dgd	z  z   } t          | | dœ|j        ¬
¦  «        }t	          j        ||¦  «         |                     d¦  «        d                              dd¬¦  «        }t          | d|j        ¬¦  «        }t	          j        ||¦  «         d S )Nr	   r
   r   r   r   g       @Tr8   r#   r   r   r   r   r&   s       r   Útest_groupby_sample_oversampler?   \   sê   € ØˆS�2‰X˜˜˜b™Ñ €FÝ	˜ fÐ-Ð-Ñ	.Ô	.€Bà�ZŠZ˜‰_Œ_×#Ò#¨°dÐ#Ñ;Ô;€FØˆS�2‰X˜˜˜b™Ñ €FÝ˜v¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨c¸4Ð(Ñ@Ô@€FÝ�f 3¨f¬lÐ;Ñ;Ô;€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,r!   c                  ó¸  — dgdz  dgdz  z   } t          | | dœ¦  «        }|                     d¦  «                             d d ¬¦  «        }t          ddgddgdœ|j        ¬¦  «        }t	          j        ||¦  «         |                     d¦  «        d                              d d ¬¦  «        }t          ddgd|j        ¬	¦  «        }t	          j        ||¦  «         d S r   r   r&   s       r   Ú%test_groupby_sample_without_n_or_fracrA   j   sà   € ØˆS�2‰X˜˜˜b™Ñ €FÝ	˜ fÐ-Ð-Ñ	.Ô	.€Bà�ZŠZ˜‰_Œ_×#Ò# d°Ð#Ñ6Ô6€FÝ  1˜v¨Q°¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨4°dÐ(Ñ;Ô;€FÝ�q˜!�f 3¨f¬lÐ;Ñ;Ô;€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,r!   zindex, expected_index)ÚwÚxÚyÚz)rB   rB   rD   rD   )r)   é   r$   é   )r)   r)   r$   r$   c                 óö  — dgdz  dgdz  z   }t          ||dœt          | ¦  «        ¬¦  «        }|                     d¦  «                             ddg d¢¬¦  «        }t          ||dœt          |¦  «        ¬¦  «        }t	          j        ||¦  «         |                     d¦  «        d	                              ddg d¢¬¦  «        }t          |d	t          |¦  «        ¬
¦  «        }t	          j        ||¦  «         d S )Nr	   r   r   r   r   T)r	   r   r	   r   )r   r9   Úweightsr   r   )r   r   r   r   r   r   r   r   )r   Úexpected_indexr   r   r   r   s         r   Ú test_groupby_sample_with_weightsrK   w   sù   € ð ˆS�1‰W˜�s˜Q‘wÑ€FÝ	˜ fÐ-Ð-µU¸5±\´\Ð	BÑ	BÔ	B€Bà�ZŠZ˜‰_Œ_×#Ò# a°¸|¸|¸|Ð#ÑLÔL€FÝ˜v¨FÐ3Ð3½5ÀÑ;PÔ;PÐQÑQÔQ€HÝÔ˜& (Ñ+Ô+Ð+à�ZŠZ˜‰_Œ_˜SÔ!×(Ò(¨1°dÀLÀLÀLÐ(ÑQÔQ€FÝ�f 3­e°NÑ.CÔ.CÐDÑDÔD€HÝÔ˜6 8Ñ,Ô,Ð,Ð,Ð,r!   c                  ó  — dgdz  dgdz  z   } t          | | | dœ¦  «        }|                     d¦  «        ddg                              d d ¬¦  «        }t          ddgddgd	œ|j        ¬
¦  «        }t	          j        ||¦  «         d S )Nr	   r
   r   )r   r   Úcr   r   rM   r   )r   rM   r   )r   r   r   r   r   r   r&   s       r   Ú#test_groupby_sample_with_selectionsrN   ‰   s—   € àˆS�2‰X˜˜˜b™Ñ €FÝ	˜ f°6Ð:Ð:Ñ	;Ô	;€Bà�ZŠZ˜‰_Œ_˜c 3˜ZÔ(×/Ò/°$¸TÐ/ÑBÔB€FÝ  1˜v¨Q°¨FÐ3Ð3¸6¼<ÐHÑHÔH€HÝÔ˜& (Ñ+Ô+Ð+Ð+Ð+r!   c                  óª   — t          g g dœ¦  «        } |                      d¦  «        }|                     ¦   «         }| }t          j        ||¦  «         d S )Nr   r   )r   r   r   r   r   )r   Ú
groupby_dfr   r   s       r   Ú%test_groupby_sample_with_empty_inputsrQ   “   sU   € å	˜ "Ð%Ð%Ñ	&Ô	&€BØ—’˜C‘”€Jà×ÒÑ Ô €FØ€HÝÔ˜& (Ñ+Ô+Ð+Ð+Ð+r!   )r/   Úpandasr   r   r   Úpandas._testingÚ_testingr   ÚmarkÚparametrizer    r'   r*   r4   r:   r=   r?   rA   rK   rN   rQ   © r!   r   ú<module>rX      s•  ðØ €€€ðð ð ð ð ð ð ð ð ð ð
 Ð Ð Ð Ð Ð ð „×Ò˜ Y°Ð$<Ñ=Ô=ð-ð -ñ >Ô=ð-ð-ð -ð -ð-ð -ð -ð3ð 3ð 3ð=ð =ð =ð „×Ò˜˜r 3˜iÑ(Ô(ð)ð )ñ )Ô(ð)ð-ð -ð -ð
-ð 
-ð 
-ð „×ÒØØÐÐÐ0Ð0Ð0Ð1°L°L°LÀ,À,À,Ð3OÐPñô ð-ð -ñ	ô ð-ð,ð ,ð ,ð,ð ,ð ,ð ,ð ,r!   