o
    "ѹg%                     @   s  d Z ddlZddlmZmZmZ ddlmZmZ ddlm	Z	 ddl
mZ ddlmZmZmZ d&dd	Zd
d Zd&ddZd"ddZdd Zdd Zdd ZddddZedddd ZG dd  d eZed!eg d"gddd#fed$gd"gd%difed%gd&gd$difed%gd&d%digd$difed$d%gd&gi feg d&d%digddd#fed%gd'd(dd#gd$d(ifeg d'd$digd$difeg d)gd$d(ifg	d*d+ Zd,d- Zed!eg d"gddg i d.feg d/gdd(dd0gi d.feg d1d2d(igdd3g d2d(id.fed4gd5d2d(igdd(d2d(id6fgd7d8 Zeg d"d%d(igdd(d#feg d"d%d(igdd(dd9fgZed!eg d:d2d(igdd(gd2d(id;feg d<gdd0gi d;fge d=d> Zd?d@ Z edAed	fedfedfgdBdC Z!dDdE Z"dFdG Z#dHdI Z$dJdKdLdMdNZ%dOe&dPe&dQdfdRdSZ'dTdU Z(dVdW Z)edXe*dYedZd[gfe*d\edZgd[gfe*d]eg gfgd^d_ Z+d`da Z,dbdc Z-eddee.e/degi dffeg dgd%dhidifgdjdk Z0dldm Z1edndo Z2dpdq Z3drds Z4dtdu Z5dS )vz
Test the func_inspect module.
    N)filter_argsget_func_nameget_func_code)_clean_win_charsformat_signature)Memory)
with_numpy)fixtureparametrizeraisesc                 C      d S N xyr   r   a/Users/admin/.pyenv/versions/3.10.0/lib/python3.10/site-packages/joblib/test/test_func_inspect.pyf      r   c                 C   r   r   r   r   r   r   r   g   r   r   c                 O   r   r   r   )r   r   argskwargsr   r   r   h   r   r      c                 C   r   r   r   r   r   r   r   i    r   r   c                 K   r   r   r   )r   r   r   r   r   r   j$   r   r   c                  O   r   r   r   )r   r   r   r   r   k(   r   r   c                C   r   r   r   r   r   r   r   m1,   r   r      )zc                C   r   r   r   r   r   r    r   r   r   m20   r   r"   module)Zscopec                 C   s&   |  d}t|j}|jdd }|S )NZjoblib_test_func_inspectc                 S      | S r   r   r   r   r   r   cached_func_inner<   s   z&cached_func.<locals>.cached_func_inner)mktempr   Zstrpathcache)Ztmpdir_factoryZcachedirZmemr%   r   r   r   cached_func4   s
   


r(   c                   @   s   e Zd Zdd ZdS )Klassc                 C   s   |S r   r   )selfr   r   r   r   r   E   r   zKlass.fN)__name__
__module____qualname__r   r   r   r   r   r)   C   s    r)   zfunc,args,filtered_argsr   r   r   r   )r   r      )r/   c                 C      t | g|R  |ksJ d S r   r   funcr   Zfiltered_argsr   r   r   test_filter_argsL   s   r4   c                  C   s&   t  } t| jg dd| dksJ d S )Nr.   r   )r   r*   r)   r   r   )objr   r   r   test_filter_args_methodZ   s    r7   )r   r   ***)r   r/   r      r:   )r      eer;   r8   )r   r/   r;   r   r   r9   c                 C   r0   r   r1   r2   r   r   r   test_filter_varargs_   s   
r>   r!   r   r/   r8   r9   r   r:   c                 C   r0   r   r1   r2   r   r   r   test_filter_kwargsr   s   rB   c                  C   sp   t tg dddiddddidksJ ttd} t | g ddgi dks'J t | dgddgi dks6J d S )	Nr?   r<   r/   r   r=   r.   r@   r   )r   r   	functoolspartialr   ffr   r   r   test_filter_args_2|   s   "rG   zfunc,funcnamec                 C   s   t | d |ks
J d S )Nr   r   )r3   funcnamer   r   r   test_func_name   s   rJ   c                 C   s   t | d dks
J d S )Nr   r%   rH   )r(   r   r   r   test_func_name_on_inner_func   s   rK   c                  C   s@   dd } dd }|  \}}| \}}||ksJ ||ksJ d S )Nc                  S      dd } t | S )Nc                   S   r   r   r   r   r   r   r   
inner_func   r   zEtest_func_name_collision_on_inner_func.<locals>.f.<locals>.inner_funcrH   rM   r   r   r   r         z1test_func_name_collision_on_inner_func.<locals>.fc                  S   rL   )Nc                   S   r   r   r   r   r   r   r   rM      r   zEtest_func_name_collision_on_inner_func.<locals>.g.<locals>.inner_funcrH   rN   r   r   r   r      rO   z1test_func_name_collision_on_inner_func.<locals>.gr   )r   r   r#   nameZother_moduleZ
other_namer   r   r   &test_func_name_collision_on_inner_func   s   

rQ   c                  C   s   t djd dksJ tdjdd  dksJ dd } t | dd	d d
ks(J t| d tddks6J d| _t | dd	d d
ksEJ t| d tddksSJ d S )Nalowerr   )NrS   c                 S   r$   r   r   r   r   r   r   <lambda>       z*test_func_inspect_errors.<locals>.<lambda>F)Zwin_charactersz<lambda>z.pycz.py__main__)r   rT   r   __file__replacer,   rE   r   r   r   test_func_inspect_errors   s    rZ   kw1kw2r[   r\   c                C   r   r   r   rR   br[   r\   r   r   r   func_with_kwonly_args   r   r`   rR   r_   returnc                 C   r   r   r   rR   r_   r   r   r   func_with_signature   r   rc   c                  C   s   t tg dddddddddksJ tt} t tg dd	di W d    n1 s+w   Y  | d
 t tdd	gdddddddksHJ t tdgdddiksUJ d S )Nr?   r   r:   r]   r   r/   r^   )r   r/   r   r\   z?Keyword-only parameter 'kw1' was passed as positional parameterr_   )rR   r[   rR   )r   r`   r   
ValueErrormatchrc   )excinfor   r   r   test_filter_args_edge_cases   s"   

rg   c                  C   s0   t  } t  }t| jg dt|jg dksJ dS )z Make sure that calling the same method on two different instances
        of the same class does resolv to different signatures.
    r.   Nr5   rb   r   r   r   test_bound_methods   s   $rh   zexception,regex,func,argsz1ignore_lst must be a list of parameters to ignorebarr   z-Ignore list: argument \'(.*)\' is not definedzWrong number of argumentsc                 C   sF   t | }t|g|R   W d   n1 sw   Y  || dS )za Make sure that filter_args returns decent error messages, for the
        sake of the user.
    N)r   r   re   )	exceptionregexr3   r   rf   r   r   r   test_filter_args_error_msg   s   
rl   c                  C   s*   ddi} t tg g |  | ddiksJ dS )zNone-regression test against 0.12.0 changes.

    https://github.com/joblib/joblib/pull/75

    Make sure filter args doesn't mutate the kwargs dict that gets passed in.
    r   r   N)r   r   )r   r   r   r   #test_filter_args_no_kwargs_mutation   s   rm   c                  C   s&   d} t | }dD ]}||vsJ qd S )NzC:\foo\bar\main.py)\:<>!)r   )stringZmangled_stringcharr   r   r   test_clean_win_chars   s
   ru   zfunc,args,kwargs,sgn_expected   zg([0, 1, 2, 3, 4]))r   r/   rA   Tzk(1, 2, (3, 4), y=True)c                 C   s*   t | g|R i |\}}||ksJ d S r   )r   )r3   r   r   Zsgn_expectedpathZ
sgn_resultr   r   r   test_format_signature   s   rx   c                     s   d} d}d|   t t \}}t||k sJ d} fddt|D }t tg|R  \}}t||| k s8J dd t|D }t tfi |\}}t||| k sUJ t tg|R i |\}}t||d	 | k snJ d S )
Ni  i  rR   rv   c                    s   g | ]} qS r   r   .0_argr   r   
<listcomp>  s    z8test_format_signature_long_arguments.<locals>.<listcomp>c                 S   s   i | ]	\}}t ||qS r   )str)rz   r   r}   r   r   r   
<dictcomp>  s    z8test_format_signature_long_arguments.<locals>.<dictcomp>r/   )r   r   lenrange	enumerate)Zshortening_thresholdZshortening_targetr{   	signatureZnb_argsr   r   r   r|   r   $test_format_signature_long_arguments  s   r   c                   C   s   dS )z6 Test the format signature formatting with numpy.
    Nr   r   r   r   r   test_format_signature_numpy   rV   r   c                  C   sB   ddl m}  t| \}}}|dksJ d|v sJ d|v sJ d S )Nr   big5_frv   zdef big5_f():Z"test_func_inspect_special_encodingZ.joblib.test.test_func_inspect_special_encodingr   r   )r   	func_codeZsource_file
first_liner   r   r   test_special_source_encoding&  s
   r   c                  C   s   ddl m}  t| d S )Nr   r   r   r   r   r   r   	_get_code.  s   r   c                     sH   ddl m} m  | dd fddtdD }tt|dks"J d S )	Nr   )Paralleldelayedr/   )Zn_jobsc                 3   s    | ]} t  V  qd S r   )r   ry   r   r   r   	<genexpr>5  s    z-test_func_code_consistency.<locals>.<genexpr>rv   r   )Zjoblib.parallelr   r   r   r   set)r   codesr   r   r   test_func_code_consistency3  s    r   )6__doc__rC   Zjoblib.func_inspectr   r   r   r   r   Zjoblib.memoryr   Zjoblib.test.commonr   Zjoblib.testingr	   r
   r   r   r   r   r   r   r   r   r"   r(   objectr)   r4   r7   r>   Ztest_filter_kwargs_extra_paramsrB   rG   rJ   rK   rQ   rZ   r`   intrc   rg   rh   rd   rl   rm   ru   listr   rx   r   r   r   r   r   r   r   r   r   <module>   s    



	


	


	

	

