Function test_np_complex64_params

python/tests/kernel/test_kernel_complex.py:407–441

Test that we can pass complex lists to kernel functions.

()

def test_np_complex64_params():
    """Test that we can pass complex lists to kernel functions."""

    # Pass a list of complex as a parameter
    c = [
        np.complex64(.70710678 + 1j),
        np.complex64(0. + 2j),
        np.complex64(0.),
        np.complex64(0.70710678)
    ]

    @cudaq.kernel
    def complex_vec_param(vec: list[np.complex64], i: int) -> np.complex64:
        return vec[i]

    for i in range(len(c)):
        is_close(c[i].real, complex_vec_param(c, i).real)
    for i in range(len(c)):
        is_close(c[i].imag, complex_vec_param(c, i).imag)

    @cudaq.kernel
    def complex_vec_param_real(vec: list[np.complex64], i: int) -> np.float32:
        v = vec[i]
        return v.real

    for i in range(len(c)):
        assert is_close(c[i].real, complex_vec_param_real(c, i))

    @cudaq.kernel
    def complex_vec_param_imag(vec: list[np.complex64], i: int) -> np.float32:
        v = vec[i]
        return v.imag

    for i in range(len(c)):
        assert is_close(c[i].imag, complex_vec_param_imag(c, i))


def test_np_complex64_capture():