
    i                     n    S SK r S SKJr  S SKJs  Js  Jr  S SKrS SKJr   " S S\ R                  5      r	g)    N)parameterizedc                       \ rS rSr\R
                  " \R                  5      S\S\	R                  R                  SS4S j5       rSrg)TestCapture   
model_namemodelreturnNc                     UR                  5       nU" U6 n[        R                  " X#[        R                  " 5       5      nU R	                  [
        R                  " XE" U6 5      5        g )N)get_random_inputsexircaptureCaptureConfig
assertTruetorchallclose)selfr   r   inputsexpectedexported_programs         a/var/www/html/ai-image-ml/venv/lib/python3.13/site-packages/executorch/exir/tests/test_capture.pytest_module_callTestCapture.test_module_call   sR     ((*&><<t7I7I7KLx1A61JKL     )__name__
__module____qualname____firstlineno__r   expandmodelsMODELSstrr   nnModuler   __static_attributes__r   r   r   r   r      sE    &--(M3 Muxx M4 M )Mr   r   )
unittestexecutorch.exirr   executorch.exir.tests.modelstestsr    r   r   TestCaser   r   r   r   <module>r+      s-      - -  'M(## Mr   