
    mi                     p    d dl mZ d dlZd dlZddlmZ ddlmZ  G d d      Zdefd	Z	d
 Z
d Zd Zd Zy)    )PathN   )
DoraConfig)XPc                       e Zd Zy)_CfgN)__name__
__module____qualname__     L/var/www/stems/demucs_env/lib/python3.12/site-packages/dora/tests/test_xp.pyr   r      s    r   r   tmpdirc                 0    t        t        |       dg      S )Na)direxclude)r   r   )r   s    r   get_dorar      s    $v,66r   c                      t        d      } | j                  j                         sJ d| _        | j                  j                         sJ y )Noutputsplop)r   r   is_absolute)doras    r   test_dora_dir_absr      s?    ID88!!!DH88!!!r   c                 `   t        t        |             } t        |       }t        |t	               g ddg      }|j
                  J t        |t	               g ddg      }|j
                  |j
                  k(  sJ t        |t	               g ddg      }|j
                  |j
                  k7  sJ y )Nr      b   r   cfgargvdelta)r   r    )r      )r   strr   r   r   sig)r   r   xpxp2xp3s        r   test_sigr+      s    #f+FFD	468Y2G	HB66
$DFIy3I
JC66SWW
$DFIy3I
JC66SWWr   c                    t        t        |             } t        |       }t        |t	               g ddg      }|j
                  j                  |        |j                  j                  |        |j                  j                  |        |j                  j                  |        |j                  j                  |        y )Nr   r   r!   )r   r&   r   r   r   folderrelative_tosubmititrendezvous_filehistory_argv_cache)r   r   r(   s      r   test_propertiesr3   ,   s    #f+FFD	468Y2G	HBII&!KKF#""6*JJ6"NNv&r   c                    t        t        |             } t        |       }t        |t	               g ddg      }|j
                  j                  d       |j                  j                  ddi       t        |t	               g ddg      }|j                  j                  g k(  sJ |j                  j                          |j                  j                  ddigk(  sJ dt        d      t        j                  d	      d
g}|j                  j                  |       |j                  j                  dddgd	z  d
gk(  sJ t        j                  t               5  |j                  j                  ddi       d d d        t        j                  t               5  |j                  j                  dt#               ig       d d d        y # 1 sw Y   RxY w# 1 sw Y   y xY w)Nr   r   r!   T)parentsr   *   +   r   )plokoutmatg        )r   r&   r   r   r   r-   mkdirlinkpush_metricsr1   loadtorchzerosupdate_historypytestraises
ValueErrorobject)r   r   r(   vals       r   	test_linkrG   8   s}   #f+FFD	468Y2G	HBIIOODO!GG&"&	468Y2G	HB77??b   GGLLN77??|n,,,tF|EKKNC
DCGG377??62$(KLLLL	z	" -
|,- 
z	" 5
 2345 5- -5 5s   G'GGG)pathlibr   r?   rB   confr   r(   r   r   r   r   r+   r3   rG   r   r   r   <module>rJ      s@        	 	7T 7"
	'5r   