B \ @sddlmZmZmZddlZddlZddlmZmZ ddl m Z m Z m Z Gddde e jZddZd d ZGd d d eZGd ddeZejddZdS))print_functionabsolute_importdivisionN)configunittest_support)captured_stdout SerialMixinredirect_c_stdoutc@seZdZddZdS) CUDATestCasecCsddlm}|dS)Nr)reset)Znumba.cuda.cudadrv.devicesr )selfr r 1lib/python3.7/site-packages/numba/cuda/testing.pytearDowns zCUDATestCase.tearDownN)__name__ __module__ __qualname__rr r r rr sr cCsttj|S)N)unittestZskipIfrENABLE_CUDASIM)reasonr r rskip_on_cudasimsrcCsttj|S)N)rZ skipUnlessrr)rr r rskip_unless_cudasimsrc@seZdZddZddZdS)CUDATextCapturecCs ||_dS)N)_stream)r streamr r r__init__szCUDATextCapture.__init__cCs |jS)N)rread)r r r rgetvalue"szCUDATextCapture.getvalueN)rrrrrr r r rrsrc@seZdZddZddZdS)PythonTextCapturecCs ||_dS)N)r)r rr r rr(szPythonTextCapture.__init__cCs |jS)N)rr)r r r rr+szPythonTextCapture.getvalueN)rrrrrr r r rr&src csbtjtjr.t}t|VWdQRXn0ddlm}t }t |V| WdQRXdS)zl Return a minimal stream-like object capturing the text output of either CUDA or the simulator. Nr)cuda) sysstdoutflushrrrrnumbarr rZ synchronize)rrr r rcaptured_cuda_stdout/s   r$)Z __future__rrr contextlibr r#rrrZnumba.tests.supportrrr ZTestCaser rrobjectrrcontextmanagerr$r r r rs