
    sh                         S SK r S SKrS SKrS SKJr  S SKJr  S SKJrJ	r	   " S S\5      r
\S:X  a  \R                  " 5         gg)    N)dedent)TemporaryDirectory)TestCaserun_in_subprocessc                       \ rS rSrSrS rSrg)TestChromeTraceModule
   z(
Test chrome tracing generated file(s).
c           	      2   Sn[        U5      n[        5        n[        R                  R	                  US5      n[        R
                  R                  5       nXES'   [        X%S9  [        U5       n[        R                  " U5      nU R                  U[        5        U HD  nU R                  U[        5        U R                  [        UR!                  5       5      1 Sk5        MF     S S S 5        S S S 5        g ! , (       d  f       N= f! , (       d  f       g = f)NaY  
            from numba import njit
            import numpy as np

            x = np.arange(100).reshape(10, 10)

            @njit
            def go_fast(a):
                trace = 0.0
                for i in range(a.shape[0]):
                    trace += np.tanh(a[i, i])
                return a + trace

            go_fast(x)
        ztest_trace.jsonNUMBA_CHROME_TRACE)env>   phtscatpidtidargsname)r   r   ospathjoinenvironcopyr   openjsonloadassertIsInstancelistdictassertEqualsetkeys)	selfcodesrctmpdirr   r   fileeventsevs	            q/Users/tiagomarins/Projetos/claudeai/copy_bank/venv/lib/python3.13/site-packages/numba/tests/test_chrome_trace.pytest_trace_output'TestChromeTraceModule.test_trace_output   s      Tl!V77<<(9:D**//#C(,$%c+dt4%%fd3 B))"d3$$BGGII !  "!
  "!s%   AD/A7C7&D7
D	D
D N)__name__
__module____qualname____firstlineno____doc__r*   __static_attributes__r,       r)   r   r   
   s     r3   r   __main__)r   r   unittesttextwrapr   tempfiler   numba.tests.supportr   r   r   r-   mainr,   r3   r)   <module>r:      s<    	    ' ;%H %P zMMO r3   