Ë
    u™ãjK  ã                   ó@  — d dl Z d dlZd dlmZ e j                  j                  dddg«      d„ «       Zd„ Z	d„ Z
d„ Zd	„ Ze j                  j                  d
ddg«      d„ «       Zd„ Zd„ Ze j                  j                  dg d¢g d¢fg d¢g d¢fg«      d„ «       Zd„ Zd„ Zy)é    Nzn, frac)é   N)Ngš™™™™™É?c                 óÚ  — dgdz  dgdz  z   }t        j                  ||dœ«      }|j                  d«      j                  | |¬«      }dgdz  dgdz  z   }t        j                  ||dœ|j                  ¬«      }t        j                  ||«       |j                  d«      d   j                  | |¬«      }t        j                  |d|j                  ¬	«      }t        j                  ||«       y ©
Né   é
   r   ©ÚaÚbr	   ©ÚnÚfrac©Úindexr
   ©Únamer   ©	ÚpdÚ	DataFrameÚgroupbyÚsampler   ÚtmÚassert_frame_equalÚSeriesÚassert_series_equal)r   r   ÚvaluesÚdfÚresultÚexpecteds         ús/var/www/html/python-recuperauto-other/env/lib/python3.12/site-packages/pandas/tests/groupby/methods/test_sample.pyÚ)test_groupby_sample_balanced_groups_shaper       sÏ   € àˆS�2‰X˜˜˜b™Ñ €FÜ	�‰˜F¨Ñ0Ó	1€Bà�Z‰Z˜‹_×#Ñ# a¨dÐ#Ó3€FØˆS�1‰W˜�s˜Q‘wÑ€FÜ�|‰| &¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨1°4Ð(Ó8€FÜ�y‰y˜ c°·±Ô>€HÜ×Ñ˜6 8Õ,ó    c                  óÖ  — dgdz  dgdz  z   } t        j                  | | dœ«      }|j                  d«      j                  d¬«      }dgdz  dgdz  z   } t        j                  | | dœ|j                  ¬	«      }t        j                  ||«       |j                  d«      d
   j                  d¬«      }t        j                  | d
|j                  ¬«      }t        j                  ||«       y )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¨Ñ0Ó	1€Bà�Z‰Z˜‹_×#Ñ# aÐ#Ó(€FØˆS�1‰W˜�s˜Q‘wÑ€FÜ�|‰| &¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨1Ð(Ó-€FÜ�y‰y˜ c°·±Ô>€HÜ×Ñ˜6 8Õ,r!   c                  óÞ  — dgdz  dgdz  z   } t        j                  | | dœg d¢¬«      }|j                  d«      j                  d¬«      }dgdz  dgdz  z   } t        j                  | | dœ|j                  ¬«      }t        j                  ||«       |j                  d«      d	   j                  d¬«      }t        j                  | d	|j                  ¬
«      }t        j                  ||«       y )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¨Ñ0Ò8JÔ	K€Bà�Z‰Z˜‹_×#Ñ# aÐ#Ó(€FØˆS�1‰W˜�s˜Q‘wÑ€FÜ�|‰| &¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨1Ð(Ó-€FÜ�y‰y˜ c°·±Ô>€HÜ×Ñ˜6 8Õ,r!   c                  óŽ  — t        j                  ddgddgdœ«      } d}t        j                  t        |¬«      5  | j                  d«      j                  dd¬«       d d d «       t        j                  t        |¬«      5  | j                  d«      d	   j                  dd¬«       d d d «       y # 1 sw Y   ŒSxY w# 1 sw Y   y xY w)
Nr   r   r   z0Please enter a value for `frac` OR `n`, not both©Úmatchr	   g      ð?r   r
   ©r   r   ÚpytestÚraisesÚ
ValueErrorr   r   ©r   Úmsgs     r   Ú%test_groupby_sample_n_and_frac_raisesr4   2   sª   € Ü	�‰˜Q ˜F¨!¨Q¨Ñ0Ó	1€BØ
<€Cä	�‰”z¨Ô	-ñ .Ø
�
‰
�3‹×Ñ ¨ÐÔ-÷.ô 
�‰”z¨Ô	-ñ 3Ø
�
‰
�3‹˜Ñ×#Ñ# a¨cÐ#Ô2÷3ð 3÷.ð .ú÷3ð 3úó   º#B/Â &B;Â/B8Â;Cc                  óŽ  — t        j                  ddgddgdœ«      } d}t        j                  t        |¬«      5  | j                  d«      j                  dd¬	«       d d d «       t        j                  t        |¬«      5  | j                  d«      d
   j                  dd¬	«       d d d «       y # 1 sw Y   ŒSxY w# 1 sw Y   y xY w)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:   =   sª   € Ü	�‰˜Q ˜F¨!¨Q¨Ñ0Ó	1€BØ
V€Cä	�‰”z¨Ô	-ñ 8Ø
�
‰
�3‹×Ñ C°ÐÔ7÷8ô 
�‰”z¨Ô	-ñ =Ø
�
‰
�3‹˜Ñ×#Ñ#¨°eÐ#Ô<÷=ð =÷8ð 8ú÷=ð =úr5   r   éÿÿÿÿr7   c                 óš  — t        j                  ddgddgdœ«      }| dk  rd}nd}t        j                  t        |¬«      5  |j                  d«      j                  | ¬	«       d d d «       t        j                  t        |¬«      5  |j                  d«      d
   j                  | ¬	«       d d d «       y # 1 sw Y   ŒRxY w# 1 sw Y   y xY w)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=   H   s²   € ä	�‰˜Q ˜F¨!¨Q¨Ñ0Ó	1€Bàˆ1‚uØM‰à4ˆä	�‰”z¨Ô	-ñ $Ø
�
‰
�3‹×Ñ ÐÔ#÷$ô 
�‰”z¨Ô	-ñ )Ø
�
‰
�3‹˜Ñ×#Ñ# aÐ#Ô(÷)ð )÷$ð $ú÷)ð )ús   Á"B5Â%CÂ5B>ÃC
c                  óÚ  — dgdz  dgdz  z   } t        j                  | | dœ«      }|j                  d«      j                  dd¬«      }dgd	z  dgd	z  z   } t        j                  | | dœ|j                  ¬
«      }t        j                  ||«       |j                  d«      d   j                  dd¬«      }t        j                  | d|j                  ¬«      }t        j                  ||«       y )Nr   r   r   r   r	   g       @Tr8   r#   r   r
   r   r   r&   s       r   Útest_groupby_sample_oversampler?   X   sÏ   € ØˆS�2‰X˜˜˜b™Ñ €FÜ	�‰˜F¨Ñ0Ó	1€Bà�Z‰Z˜‹_×#Ñ#¨°dÐ#Ó;€FØˆS�2‰X˜˜˜b™Ñ €FÜ�|‰| &¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨c¸4Ð(Ó@€FÜ�y‰y˜ c°·±Ô>€HÜ×Ñ˜6 8Õ,r!   c                  óÌ  — dgdz  dgdz  z   } t        j                  | | dœ«      }|j                  d«      j                  d d ¬«      }t        j                  ddgddgdœ|j                  ¬«      }t        j                  ||«       |j                  d«      d   j                  d d ¬«      }t        j                  ddgd|j                  ¬	«      }t        j                  ||«       y r   r   r&   s       r   Ú%test_groupby_sample_without_n_or_fracrA   f   sÅ   € ØˆS�2‰X˜˜˜b™Ñ €FÜ	�‰˜F¨Ñ0Ó	1€Bà�Z‰Z˜‹_×#Ñ# d°Ð#Ó6€FÜ�|‰| 1 a &°°1¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨4°dÐ(Ó;€FÜ�y‰y˜!˜Q˜ c°·±Ô>€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        j                  ||dœt        j                  | «      ¬«      }|j                  d«      j	                  ddg d¢¬«      }t        j                  ||dœt        j                  |«      ¬«      }t        j                  ||«       |j                  d«      d	   j	                  ddg d¢¬«      }t        j                  |d	t        j                  |«      ¬
«      }t        j                  ||«       y )Nr   r   r   r   r	   T)r   r   r   r   )r   r9   Úweightsr
   r   )	r   r   ÚIndexr   r   r   r   r   r   )r   Úexpected_indexr   r   r   r   s         r   Ú test_groupby_sample_with_weightsrL   s   sÓ   € ð ˆS�1‰W˜�s˜Q‘wÑ€FÜ	�‰˜F¨Ñ0¼¿¹À»Ô	H€Bà�Z‰Z˜‹_×#Ñ# a°º|Ð#ÓL€FÜ�|‰| &¨vÑ6¼b¿h¹hÀ~Ó>VÔW€HÜ×Ñ˜& (Ô+à�Z‰Z˜‹_˜SÑ!×(Ñ(¨1°dÂLÐ(ÓQ€FÜ�y‰y˜ c´·±¸.Ó1IÔJ€HÜ×Ñ˜6 8Õ,r!   c                  ó  — dgdz  dgdz  z   } t        j                  | | | dœ«      }|j                  d«      ddg   j                  d d ¬«      }t        j                  ddgddgd	œ|j                  ¬
«      }t        j                  ||«       y )Nr   r   r   )r	   r
   Úcr	   r
   rN   r   )r
   rN   r   )r   r   r   r   r   r   r   r&   s       r   Ú#test_groupby_sample_with_selectionsrO   …   s„   € àˆS�2‰X˜˜˜b™Ñ €FÜ	�‰˜F¨°fÑ=Ó	>€Bà�Z‰Z˜‹_˜c 3˜ZÑ(×/Ñ/°$¸TÐ/ÓB€FÜ�|‰| 1 a &°°1¨vÑ6¸f¿l¹lÔK€HÜ×Ñ˜& (Õ+r!   c                  ó¦   — t        j                  g g dœ«      } | j                  d«      }|j                  «       }| }t	        j
                  ||«       y )Nr   r	   )r   r   r   r   r   r   )r   Ú
groupby_dfr   r   s       r   Ú%test_groupby_sample_with_empty_inputsrR   �   sE   € ä	�‰˜B RÑ(Ó	)€BØ—‘˜C“€Jà×ÑÓ €FØ€HÜ×Ñ˜& (Õ+r!   )r/   Úpandasr   Úpandas._testingÚ_testingr   ÚmarkÚparametrizer    r'   r*   r4   r:   r=   r?   rA   rL   rO   rR   © r!   r   ú<module>rY      sÂ   ðÛ ã Ý ð ‡�×Ñ˜ Y°Ð$<Ó=ñ-ó >ð-ò-ò-ò3ò=ð ‡�×Ñ˜˜r 3˜iÓ(ñ)ó )ð)ò-ò
-ð ‡�×ÑØÚÒ0Ð1²LÂ,Ð3OÐPóñ-ó	ð-ò,ó,r!   