
    mi
                     @   d dl mZ d dlmZ d dlZd dlZd dlZddlmZ ddl	m
Z
mZ  ed      Zej                  de	       ej                  d
ed       ej                  ded       ej                  d       ej                  de	       ddgZddZd Zd Zd Zy)    )ArgumentParser)PathN   )argparse_main)XPget_xp	test_mainz--a)typez--b*   )r
   defaultz--num_workers
   z--cat_az--cat_bnum_workerszcat_*c                     t        t        t        | |d      d        }t        j                  j                  d      dk(  rd|j                  _        |S )NT)parserexcludedirshareduse_underscorec                     t               } t        t        d      j                               }t        | j                  dz  j                               }| j
                  j                  r8|j                  |      sJ |       t        j                  |      sCJ t               |j                  |      rJ |       t        j                  |      rJ t               | j                  j                  ddi       | S )N.codeloss皙?)r   strr   resolvefolderdoragit_save
startswith__file__linkpush_metrics)xpcwdr   s      N/var/www/stems/demucs_env/lib/python3.12/site-packages/dora/tests/test_main.pymainzget_main.<locals>.main   s     X$s)##%&BII&//1277>>$',,'&&t,6h6,~~d+0S0+**40:(:0
fc]+	    _DORA_GIT_SAVE1)r   r   EXCLUDEosenvirongetr   r   )dora_dirr   r&   s      r%   get_mainr/      sN    wHd,, 
zz~~&'3.!		Kr'   c                     t        t        j                        }	 |t        j                  dd   |        |t        _        S # |t        _        w xY w)N   )listsysargv)r&   r4   old_argvs      r%   callr6   3   s7    CHH~Hv8s   A   Ac                    | dz  }t        | dz  |      }t        | dz        }t        | dz  |      }|j                  dg      }|j                  |       t        j                  t
              5  |j                  |j                         d d d        |j                  |j                        }|j                  |j                  k(  sJ |j                  |j                  k(  sJ y # 1 sw Y   [xY w)Nr   abcz--a=5)	r/   r   init_xppytestraisesRuntimeErrorget_xp_from_sigsigr4   )tmpdirr   main_amain_bmain_cr#   xp2s          r%   test_sharedrF   <   s    hFfslF+Ffsl#FfslF+F	y	!B
NN2	|	$ 'rvv&'

 
 
(C77bff88rww	' 's   .C$$C-c                    t        |       }t        |g       }t        |t              sJ t	        |j
                        dkD  sJ dg}t        ||      }|j
                  |j
                  k(  sJ |j                  ddd      }t	        |      dkD  sJ t        ||      }|j
                  |j
                  k(  sJ |j                  j                  dk(  sJ ddg}|j                  |      }||k(  sJ t        ||      }|j
                  |j
                  k7  sJ ||j                  |j
                        k(  sJ |j                  |j
                        }||k(  sJ |j                  |      }|d   d   d	k(  sJ |j                  |      }|d
k(  sJ t        j                  t              5  |j                  d       d d d        y # 1 sw Y   y xY w)Nr   z--cat_a=plopr      )r9   cat_bz--a=4r   r   za=4g      ?)r/   r6   
isinstancer   lenr@   value_to_argvcfgrI   get_argv_from_sigr?   get_xp_historyget_namer<   r=   
ValueError)	rA   r&   r#   r4   rE   prexp3metricsnames	            r%   r	   r	   L   s   FD	dBBb"rvv;??D
tT
C66SWWB34Dt9q==
tT
C66SWW77==A7
#Cc"D3;;
tS/C66SWW4))#''2222


sww
'C#::!!#&G2;v#%%%==D5==	z	"  3     s   (GG)N)argparser   pathlibr   r+   r3   r<   r&   r   r#   r   r   r   add_argumentintr*   r/   r6   rF   r	    r'   r%   <module>r\      s    $  	 
     
	$   E  $   ER  0   O#r  :   I    IC  ( '
", " r'   