Function test_np_complex128_params

python/tests/kernel/test_kernel_complex.py:229–263

Test that we can pass complex lists to kernel functions.

()

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

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

    @cudaq.kernel
    def complex_vec_param(vec: list[np.complex128], i: int) -> complex:
        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.complex128], i: int) -> float:
        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.complex128], i: int) -> float:
        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_complex128_capture():