
    Cd                    \   d dl mZ d dlZd dlZd dlZd dlZd dlZd dlZd dlm	Z	 d dl
Z
d dlZ
d Zd Zd Zd Zd Zd	 Z ej                    d
             Zd Z ej        d          d             Z ej        d          d             Zd Z ej                    d             Zd ZdS )    )annotationsN)	CliRunnerc                     t                      } |                     t          j        j                  }|j        dk    sJ |j                            d          sJ d S )N   zConfig key not specifiedr   invokedaskcli
config_get	exit_codeoutput
startswithrunnerresults     3lib/python3.11/site-packages/dask/tests/test_cli.pytest_config_getr      sX    [[F]]48.//Fq    =##$>???????    c                     t                      } |                     t          j        j        dg          }|j        dk    sJ |j                            d          sJ d S )Narrayr   zbackend:r   r   s     r   test_config_get_valuer      s[    [[F]]48.	::Fq    =##J///////r   c                     t                      } |                     t          j        j        dg          }|j        dk    sJ |j                            d          sJ d S )Nbad_keyr   zSection not foundr   r   s     r   test_config_get_bad_valuer      s\    [[F]]48.<<Fq    =##$78888888r   c                     t                      } |                     t          j        j                  }|j        dk    sJ |j                            d          sJ d S )Nr   zarray:)r   r   r	   r
   config_listr   r   r   r   s     r   test_config_listr   %   sW    [[F]]48/00Fq    =##H-------r   c                     t                      } |                     t          j        j        dg          }|j        dk    sJ |j        dt          j         dk    sJ d S )Nz	--versionr   zcli, version 
)r   r   r	   r
   r   r   __version__r   s     r   test_versionr!   ,   s`    [[F]]48<+77Fq    =@D,<@@@@@@@@@r   c                    t                      } |                     t          j        j                  }|j        dk    sJ t          j        |j                  }|d         d	                    d t          j        d d         D                       k    sJ |d         t          j        k    sJ |d         t          j                    j        k    sJ 	 ddlm} n# t"          $ r d }Y nw xY w|d	         |k    sJ d S )
Nr   Python.c              3  4   K   | ]}t          |          V  d S N)str).0xs     r   	<genexpr>z%test_info_versions.<locals>.<genexpr>=   s(      &L&L!s1vv&L&L&L&L&L&Lr      r	   Platform)r    distributed)r   r   r	   r
   versionsr   jsonloadsr   joinsysversion_infor    platformunamesystemr-   ImportError)r   r   tabledistributed_versions       r   test_info_versionsr:   3   s   [[F]]48,--Fq    
 Jv}%%E?chh&L&Ls7G7K&L&L&LLLLLLL=D,,,,, 0 0 77777#BBBBBBB # # #"# #6666666s   C C)(C)c                     d S r&    r<   r   r   	dummy_clir=   I       Dr   c                     d S r&   r<   r<   r   r   bad_commandr@   N   s    Dr   good)namec                     d S r&   r<   r<   r   r   good_commandrD   R   r>   r   c                     d S r&   r<   r<   r   r   good_command_2rF   W   r>   r   c                 n   ddl m}  t          j        ddd          }t          j        ddd          }t	          j        t          d	
          5   | t          |           d d d            n# 1 swxY w Y    | t          |           dt          j        v sJ t          j        d         t          u sJ d S )Nr   _register_command_epbadzdask.tests.test_cli:bad_commanddask_clirB   valuegrouprA    dask.tests.test_cli:good_commandzmust be instances ofmatch)
dask.clirI   importlib_metadata
EntryPointpytestwarnsUserWarningr=   commandsrD   )rI   bad_epgood_eps      r   test_register_command_epr[   \   s   ------*/  F !+0  G 
k)?	@	@	@ 0 0Y///0 0 0 0 0 0 0 0 0 0 0 0 0 0 0 G,,,Y'''''f%555555s   A..A25A2c                     d S r&   r<   r<   r   r   dummy_cli_2r]   s   r>   r   c                    ddl m}  t          j        ddd          }t          j        ddd          } | t          |           t          j        t          d	
          5   | t          |           d d d            d S # 1 swxY w Y   d S )Nr   rH   onerO   rK   rL   twoz"dask.tests.test_cli:good_command_2z'While registering the command with namerP   )rR   rI   rS   rT   r]   rU   rV   rW   )rI   r_   r`   s      r   $test_repeated_name_registration_warnra   x   s    ------

'0  C 
'2  C c***	k)R	S	S	S / /[#.../ / / / / / / / / / / / / / / / / /s   !B  BB)
__future__r   r/   r4   r2   clickrS   rU   click.testingr   r	   rR   r   r   r   r   r!   r:   rN   r=   r@   commandrD   rF   r[   r]   ra   r<   r   r   <module>rf      s   " " " " " "   



       # # # # # #  @ @ @0 0 09 9 9. . .A A A7 7 7, 	 	 		 	 	 F	 	 	 F	 	 	6 6 6. 	 	 	/ / / / /r   