<s> from __future__ import print_function , absolute_import <EOL> import numpy as np <EOL> from numba import cuda <EOL> from numba . cuda . testing import unittest <EOL> class TestCudaComplex ( unittest . TestCase ) : <EOL> def test_cuda_complex_arg ( self ) : <EOL> @ cuda . jit ( '<STR_LIT>' ) <EOL> def foo ( a , b ) : <EOL> i = cuda . grid ( <NUM_LIT:1> ) <EOL> a [ i ] += b <EOL> a = np . arange ( <NUM_LIT:5> , dtype = np . complex128 ) <EOL> a0 = a . copy ( ) <EOL> foo [ <NUM_LIT:1> , a . shape ] ( a , <NUM_LIT> ) <EOL> self . assertTrue ( np . allclose ( a , a0 + <NUM_LIT> ) ) <EOL> if __name__ == '<STR_LIT:__main__>' : <EOL> unittest . main ( ) </s>
