iddlZddlZddlZddlmZddlmZddlmZm Z GddeZ e dk(rejyy)N)dedent)TemporaryDirectory)TestCaserun_in_subprocessceZdZdZdZy)TestChromeTraceModulez0 Test chrome tracing generated file(s). c $d}t|}t5}tjj |d}tj j }||d<t||t|5}tj|}|j|t|D]C}|j|t|jt|j!hdE ddddddy#1swYxYw#1swYyxYw)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) ztest_trace.jsonNUMBA_CHROME_TRACE)env>phtscatpidtidargsname)rrospathjoinenvironcopyropenjsonloadassertIsInstancelistdict assertEqualsetkeys) selfcodesrctmpdirrr fileeventsevs r/Volumes/fast/ai/experiments/voice-extract-mac/.venv/lib/python3.12/site-packages/numba/tests/test_chrome_trace.pytest_trace_outputz'TestChromeTraceModule.test_trace_outputs  Tl  !V77<<(9:D**//#C(,C$ % cs +dt4%%fd3 B))"d3$$BGGII! " !  " !s%AD4A4C:)D:D ?DDN)__name__ __module__ __qualname____doc__r)r(rr s  r/r__main__) rrunittesttextwraprtempfilernumba.tests.supportrrrr*mainr.r/r(r6s> ';%H%P zHMMOr/