
    iY                         S SK r S SKrS SKJr  S SKJr  S SKJr  S SKJ	r	  \ " S S5      5       r
S\S	\
S
\R                  4S jrS\R                  S	\
S
\R                  4S jrS\R                  S	\
S
\4S jrg)    N)	dataclass)
ModuleType)Pattern)TestFlowc                   `    \ rS rSr% Sr\\   S-  \S'    \\   S-  \S'    \\   S-  \S'   Sr	g)
TestFilter   z$A set of filters for test discovery.Nbackendsflows
name_regex )
__name__
__module____qualname____firstlineno____doc__setstr__annotations__r   __static_attributes__r       g/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/backends/test/suite/discovery.pyr   r      s9    .#hoOs8d?Qt##Wr   r   root_moduletest_filterreturnc                     U R                   nUc  [        SU  S35      e[        R                  " 5       n[        R
                  R                  U5      nUR                  U5      n[        XQ5      $ )NzModule z has no __file__ attribute)	__file__RuntimeErrorunittest
TestLoaderospathdirnamediscover_filter_tests)r   r   module_fileloader
module_dirsuites         r   discover_testsr*   %   sc     &&KW[M1KLMM  "F-JOOJ'E,,r   r)   c                 j   [         R                  " 5       nU  H  n[        U[         R                  5      (       a  UR                  [	        X15      5        M>  [        U[         R
                  5      (       a%  [        X15      (       a  UR                  U5        M  M  [        S[        U5       35      e   U$ )NzUnexpected test type: )	r   	TestSuite
isinstanceaddTestr%   TestCase_is_test_enabledr   type)r)   r   filtered_suitechilds       r   r%   r%   6   s     '')NeX//00""=#DEx001133&&u- 4 !7U}EFF  r   	test_casec                    [        X R                  5      nS[        U5      ;   a  [        SU R                   S35        g[	        US5      (       d  [        SU R                   SU 35      eUR                  nUR                  b  UR                  UR                  ;  a  gUR                  b  UR                  UR                  ;  a  gUR                  b/  UR                  R                  U R                  5       5      (       d  gg)	NtestFailurezWarning: Test z failed to import.T_flowzTest missing flow:  F)getattr_testMethodNamer   printhasattrr   r7   r
   backendr   namer   searchid)r4   r   test_methodflows       r   r0   r0   H   s    )%>%>?K
 K((y8899KLM;((!)";";!<Ak]K
 	
 !&&D'DLL@T@T,T$+:K:K)K)+2H2H2O2O3 3 r   )r!   r   dataclassesr   typesr   typingr   #executorch.backends.test.suite.flowr   r   r,   r*   r%   r/   boolr0   r   r   r   <module>rH      s    
  !   8 
X 
X 
X--*4--",6$ 1 1 
 t r   