o
    NrfL  ã                   @  sR  d dl mZ d dlmZmZmZmZmZmZ d dl	m
Z
mZ dd„ Zdd„ Zdd	„ Zd
d„ ZdZee
ddfedfeƒZee
ddfedfeƒZee
edfedffe
edfdfeƒZee
edfedffe
e
ddfdfeƒZeeg d¢fe
e
ddfdfeƒZdd„ ZeedfedƒZdd„ Zdd„ ZeeeeeegZeeŽ Zdd„ Zdd„ Zdd „ Z d!S )"é    )Úannotations)ÚVARÚRewriteRuleÚRuleSetÚ	TraverserÚargsÚhead)ÚaddÚincc                 C  s   | d S )Né   © ©Úxr   r   ú`/var/www/html/software/conda/envs/catlas/lib/python3.10/site-packages/dask/tests/test_rewrite.pyÚdouble   s   r   c                   C  s`   t tdfƒtks
J ‚t tddfƒtksJ ‚t ttdftdffƒtks$J ‚t g d¢ƒtks.J ‚d S ©Né   r   ©r   r   é   )r   r
   r	   Úlistr   r   r   r   Ú	test_head   s   r   c                   C  sV   t tdfƒdks
J ‚t tddfƒdksJ ‚t dƒdksJ ‚t g d¢ƒg d¢ks)J ‚d S )Nr   )r   r   )r   r   r   r   )r   r
   r	   r   r   r   r   Ú	test_args   s   r   c                  C  s¨   t tdfttdfdff} t| ƒ}| ¡ }|jt ksJ ‚| ¡  |jtks&J ‚|jt ks-J ‚| ¡  |jtks8J ‚| ¡  |jtksCJ ‚t|ƒt tdttddgksRJ ‚d S )Nr   r   )	r	   r
   r   r   ÚcopyÚcurrentÚnextÚskipr   )ÚtermÚtÚt2r   r   r   Útest_traverser   s   "r   ©ÚaÚbÚcr!   r   r   r"   ©r#   r"   r!   r#   c                 C  s   | d }t |tƒr|S t|fS )Nr   )Ú
isinstancer   )Úsdr   r   r   r   Ú	repl_list7   s   
r'   r   r   c                   C  s¢   t jdksJ ‚t jdgksJ ‚tjdksJ ‚tjddgksJ ‚tjdks&J ‚tjddgks/J ‚tjdks6J ‚tjddgks?J ‚tjdksFJ ‚tjg d¢ksOJ ‚d S )N)r!   r!   ©r!   r"   r"   r    r$   )Úrule1ÚvarsZ_varlistÚrule2Úrule3Úrule4Úrule5r   r   r   r   Útest_RewriteRuleB   s   r/   c                   C  s<   t  ddi¡tdfksJ ‚t dg d¢i¡g d¢ksJ ‚d S )Nr!   r   r   r   )r)   Úsubsr
   Úrule6r   r   r   r   Útest_RewriteRuleSubsP   s   "r2   c                  C  s¬   t tti dgfdi dgfig ftttti ddgfig fig fig fig ftti dgfig fttttti dgfig fig fig fig fig f} tj| ksMJ ‚tjtksTJ ‚d S )Nr   r   r   r   é   é   )r	   r   r
   r   ÚsumÚrsZ_netÚrules)Únetr   r   r   Útest_RuleSetZ   s   $þû*÷ôr9   c                  C  sˆ  t ddf} tt | ¡ƒ}t|ƒdksJ ‚|d tddifks J ‚t ddf} tt | ¡ƒ}t|ƒdks4J ‚|d tddifks@J ‚|d tddifksLJ ‚t dgdgf} tt | ¡ƒ}t|ƒdksbJ ‚|d tddgifksoJ ‚t tdftdff} tt | ¡ƒ}t|ƒdks‡J ‚|d tddifks“J ‚|d t	dddœfks J ‚|d tdtdfifks®J ‚t ddf} tt | ¡ƒ}t|ƒdksÂJ ‚d S )Nr   r   r   r!   r   r(   )
r	   r   r6   Ziter_matchesÚlenr)   r+   r
   r,   r-   )r   Úmatchesr   r   r   Útest_matchesm   s,   


r<   c                  C  s  t tddftddftddfgf} t | ¡}|tttdftdfftdffks'J ‚t |¡}|tttdfdftdffks<J ‚ttttddftddffttddftddfffdf} t | ¡ttttddffffkshJ ‚tg d¢f} t | ¡g d¢ksyJ ‚tttg d¢ff} t | ¡| ks‹J ‚d S r   )r5   r	   r6   Zrewriter
   r   r   Úmap)r   Znew_termr   r   r   Útest_rewrite‹   s    
$
 ,ý$r>   N)!Ú
__future__r   Zdask.rewriter   r   r   r   r   r   Zdask.utils_testr	   r
   r   r   r   r   r*   r)   r+   r,   r-   r5   r.   r'   r   r1   r/   r2   r7   r6   r9   r<   r>   r   r   r   r   Ú<module>   s,     $& 