
    [ j                         S SK r S SKrS SKrS SKJr  S SKJr  S\S\4S jrS\S\S-  4S jr	S\S\4S jr
S	\S\S-  4S
 jrS\4S jrS r\S:X  a  \" 5         gg)    N)Counter)Pathnodeidreturnc                 Z    U R                  S5      S   n[        R                  " SSU5      $ )N::z\[.*\]$ )splitresub)r   names     r/root/GenerationalWealth/GenerationalWealth/venv/lib/python3.13/site-packages/transformers/utils/pytest_helpers.py_base_test_namer      s(    <<b!D66*b$''    c                 N    U R                  S5      n[        U5      S:  a  US   $ g )Nr      )r   len)r   partss     r   _class_namer      s(    LLE
5zQRyr   c                 *    U R                  S5      S   $ )Nr   r   )r   )r   s    r   
_file_pathr      s    <<a  r   	file_pathc                 b    [         R                  " SU 5      nU(       a  UR                  S5      $ g )Nz"test_modeling_([A-Za-z0-9_]+)\.py$   )r   searchgroup)r   ms     r   _modeling_keyr       s'    
		7CAwwqzr   report_pathc           	      x   [        U 5      nUR                  5       (       d  [        SUR                  5        35      e[        R
                  " UR                  5       5      nUR                  S/ 5      n[        S U 5       5      nU Vs/ s H  oUR                  S5      S;   d  M  UPM     nn[        S U 5       5      n[        S U 5       5      n[        S U 5       5      n	[        5       n
U H<  n[        [        UR                  S	S
5      5      5      nU(       d  M0  X==   S-  ss'   M>     UUUU	U
S.$ s  snf )NzReport file not found: testsc              3   D   #    U  H  oR                  S S5      v   M     g7f)outcomeunknownN)get.0ts     r   	<genexpr>summarize.<locals>.<genexpr>+   s     BEquuY	22Es    r%   )failederrorc              3   X   #    U  H   n[        UR                  S S5      5      v   M"     g7fr   r
   N)r   r'   r(   s     r   r+   r,   1   s#     PA
1552+> ? ?   (*c              3   j   #    U  H)  n[        UR                  S S5      5      =(       d    Sv   M+     g7f)r   r
   NO_CLASSN)r   r'   r(   s     r   r+   r,   4   s+      b[aVW+aeeHb.A"B"Pj"P[as   13c              3   X   #    U  H   n[        UR                  S S5      5      v   M"     g7fr0   )r   r'   r(   s     r   r+   r,   7   s%     #YRXQOAEE(B4G$H$HRXr1   r   r
   r   )outcomesfailures_per_filefailures_per_classfailures_per_testnamefailures_per_modeling_key)r   existsFileNotFoundErrorresolvejsonloads	read_textr'   r   r    r   )r!   pdatar#   r5   r*   r-   r6   r7   r8   r9   keys               r   	summarizerC   "   s(   [A88::"9!))+ GHH::akkm$DHHWb!E BEBBH JA%%	"26I"IaFJ  PPP ! b[a bb $#YRX#YY !(	JquuXr':;<3%*a/*  .0!6%> % Ks   	D7%D7c                     [         R                  " SS9n U R                  SSSS9  U R                  5       n [	        UR
                  5      nUS   n[        S5        [        UR                  5       5      n[        S	U 35        [        U5       H  n[        US
 SXF    35        M     SS[        4S jjnU" SUS   SS9  U" SUS   SS9  U" SUS   5        U" SUS   5        g ! [         a  n[        [        U5      5         S nAg S nAff = f)Nz%Summarize pytest JSON report failures)descriptionz--reportzreport.jsonz6Path to pytest JSON report file (default: report.json))defaulthelpr5   z=== Overall ===zTotal tests: z>10z: counterc                     [        SU  S35        U(       d  [        S5        g [        UR                  5       S S9 H2  u  p4U(       a  [        US SU U 35        M!  [        US SU 35        M4     g )Nz
=== z ===Nonec                     U S   U S   4$ )Nr   r    )xs    r   <lambda>.main.<locals>._print_counter.<locals>.<lambda>b   s    qtQqTlr   )rB   4dz  )printsorteditems)titlerH   labelrB   cnts        r   _print_countermain.<locals>._print_counter]   sl    ugT"#&Mw}}4JKHCR5'#/0R3%()	 Lr   zFailures per test classr7   zclass )rU   zFailures per test_modeling_xxxr9   zmodel zFailures per test filer6   zFailures per test name (base)r8   )r
   )argparseArgumentParseradd_argument
parse_argsrC   reportr;   rQ   strsumvaluesrR   r   )parserargssummaryer5   totalkrW   s           r   mainrg   I   s$   $$1XYF
M0h   DDKK(
 z"H	
!"E	M%
!"H3r(+'( 	*w 	* ,g6J.KS[\3W=X5Yaij+W5H-IJ2G<S4TU3  c!fs   C$ $
D.DD__main__)rY   r=   r   collectionsr   pathlibr   r^   r   r   r   r    rC   rg   __name__rL   r   r   <module>rl      s      	  (C (C ( d
 !s !s !S S4Z $3 $N"VJ zF r   